1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.contrib import admin
from asynchronous_send_mail import send_mail
from django.conf import settings
# Register your models here.
from .serializer import UserMachineSerializer
from .models import UserMachine, UserProfile
class UserProfileAdmin(admin.ModelAdmin):
list_display = ('id', 'username', 'email', 'is_active', 'last_login', )
list_filter = ('is_active',)
admin.site.register(UserProfile, UserProfileAdmin)
class UserMachineAdmin(admin.ModelAdmin):
list_display = ('id', 'alias', 'state', 'machine_sn', 'machine_secret', )
list_filter = ('state',)
actions = ['approve_machine','only_approve']
def approve_machine(self, request, queryset):
total = 0
error = 0
success = 0
for machine in queryset:
ret = machine.approve_machine()
# ret = {"is_success": True, "alias": self.alias.name, "secret": self.machine_secret, "system": system, "compiler":compiler,"email":user_email}
if ret['is_success']:
success += 1
# send email to notice user
content = "Greetings,\n\
\
The machine you have applied for has been approved.\n\
Here is the information about it: \n \
\n \
alias: %s\n \
secret: %s\n \
system: %s\n \
compiler: %s\n \
\n \
Regards,\n \
PG PERF FARM" % (ret['alias'], ret['secret'], ret['system'], ret['compiler'])
send_mail('[PG PERF FARM<test mail>]Machine Approval Notice', content, settings.EMAIL_HOST_USER, [ret['email']],
fail_silently=False)
else:
error += 1
total += 1
# rows_updated = queryset.update(state=1)
# message_bit = "%s machine(s)" % rows_updated
self.message_user(request,
"Total: %s ,Success: %s ,Error: %s. Please make sure there are enough unused aliases." % (
total, success, error))
approve_machine.short_description = u'Approve Machine(Modify the state to active, generate machine_sn, machine_secret and assign an alias)'
def only_approve(self, request, queryset):
total = 0
error = 0
success = 0
for machine in queryset:
ret = machine.approve_machine()
# ret = {"is_success": True, "alias": self.alias.name, "secret": self.machine_secret, "system": system, "compiler":compiler,"email":user_email}
if ret['is_success']:
success += 1
else:
error += 1
total += 1
# rows_updated = queryset.update(state=1)
# message_bit = "%s machine(s)" % rows_updated
self.message_user(request,
"Total: %s ,Success: %s ,Error: %s. Please make sure there are enough unused aliases." % (
total, success, error))
only_approve.short_description = u'Approve Machine(with send emails)'
admin.site.register(UserMachine, UserMachineAdmin)
|
