class TestRecordAdmin(admin.ModelAdmin):
list_display = ('id', 'branch', 'uuid', 'test_machine')
list_filter = ('branch',)
- actions = ['approve_machine']
admin.site.register(TestRecord, TestRecordAdmin)
class UserMachineAdmin(admin.ModelAdmin):
list_display = ('id', 'alias', 'state', 'machine_sn', 'machine_secret', )
list_filter = ('state',)
- actions = ['approve_machine']
+ actions = ['approve_machine','only_approve']
def approve_machine(self, request, queryset):
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)