make user and group management more robust
- remove TaskResultInline and subclasses - add custom perform_delete_selected action to UserAdmin and GroupAdmin - properly clean asynchronous tasks in rabbitmq - wrap user operations in transactions
This commit is contained in:
parent
7bce88e18a
commit
e877087127
2 changed files with 91 additions and 43 deletions
|
@ -1,3 +1,5 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
from datetime import date
|
||||
import os
|
||||
|
||||
|
@ -42,18 +44,17 @@ class TaskResult(TimeStampedModel, models.Model):
|
|||
abstract = True
|
||||
|
||||
def _set_result_fields(self, asyncresult):
|
||||
if asyncresult.ready():
|
||||
if not self.is_finished:
|
||||
result = asyncresult.get(no_ack=False)
|
||||
self.is_finished = True
|
||||
self.is_success = asyncresult.state == 'SUCCESS'
|
||||
self.result_body = str(asyncresult.result)
|
||||
self.state = asyncresult.state
|
||||
asyncresult.get(no_ack=False)
|
||||
self.result_body = str(result)
|
||||
|
||||
def update_taskstatus(self):
|
||||
if not self.is_finished:
|
||||
asyncresult = AsyncResult(self.task_uuid)
|
||||
self._set_result_fields(asyncresult)
|
||||
self.save()
|
||||
asyncresult = AsyncResult(self.task_uuid)
|
||||
self._set_result_fields(asyncresult)
|
||||
self.save()
|
||||
|
||||
|
||||
class GroupManager(models.Manager):
|
||||
|
@ -84,6 +85,7 @@ class Group(TimeStampedModel, models.Model):
|
|||
def __str__(self):
|
||||
return '{0} ({1})'.format(self.groupname, self.gid)
|
||||
|
||||
@transaction.atomic
|
||||
def save(self, *args, **kwargs):
|
||||
super(Group, self).save(*args, **kwargs)
|
||||
GroupTaskResult.objects.create_grouptaskresult(
|
||||
|
@ -93,6 +95,7 @@ class Group(TimeStampedModel, models.Model):
|
|||
)
|
||||
return self
|
||||
|
||||
@transaction.atomic
|
||||
def delete(self, *args, **kwargs):
|
||||
DeleteTaskResult.objects.create_deletetaskresult(
|
||||
'group', self.groupname,
|
||||
|
@ -125,6 +128,7 @@ class DeleteTaskResultManager(TaskResultManager):
|
|||
return taskresult
|
||||
|
||||
|
||||
@python_2_unicode_compatible
|
||||
class DeleteTaskResult(TaskResult):
|
||||
|
||||
modeltype = models.CharField(max_length=20, db_index=True)
|
||||
|
@ -132,6 +136,11 @@ class DeleteTaskResult(TaskResult):
|
|||
|
||||
objects = DeleteTaskResultManager()
|
||||
|
||||
def __str__(self):
|
||||
return "{task_uuid} {task_name} {modeltype} {modelname}".format(
|
||||
task_uuid=self.task_uuid, task_name=self.task_name,
|
||||
modeltype=self.modeltype, modelname=self.modelname)
|
||||
|
||||
|
||||
class GroupTaskResultManager(TaskResultManager):
|
||||
|
||||
|
@ -145,12 +154,18 @@ class GroupTaskResultManager(TaskResultManager):
|
|||
return taskresult
|
||||
|
||||
|
||||
@python_2_unicode_compatible
|
||||
class GroupTaskResult(TaskResult):
|
||||
|
||||
group = models.ForeignKey(Group)
|
||||
|
||||
objects = GroupTaskResultManager()
|
||||
|
||||
def __str__(self):
|
||||
return "{task_uuid} {task_name} {group}".format(
|
||||
task_uuid=self.task_uuid, task_name=self.task_name,
|
||||
group=self.group)
|
||||
|
||||
|
||||
class UserManager(models.Manager):
|
||||
|
||||
|
@ -215,6 +230,7 @@ class User(TimeStampedModel, models.Model):
|
|||
def __str__(self):
|
||||
return '{0} ({1})'.format(self.username, self.uid)
|
||||
|
||||
@transaction.atomic
|
||||
def set_password(self, password):
|
||||
if hasattr(self, 'shadow'):
|
||||
self.shadow.set_password(password)
|
||||
|
@ -232,6 +248,7 @@ class User(TimeStampedModel, models.Model):
|
|||
commit=True
|
||||
)
|
||||
|
||||
@transaction.atomic
|
||||
def save(self, *args, **kwargs):
|
||||
UserTaskResult.objects.create_usertaskresult(
|
||||
self,
|
||||
|
@ -243,6 +260,7 @@ class User(TimeStampedModel, models.Model):
|
|||
)
|
||||
return super(User, self).save(*args, **kwargs)
|
||||
|
||||
@transaction.atomic
|
||||
def delete(self, *args, **kwargs):
|
||||
for group in [
|
||||
ag.group for ag in AdditionalGroup.objects.filter(user=self)
|
||||
|
@ -274,15 +292,21 @@ class UserTaskResultManager(TaskResultManager):
|
|||
return taskresult
|
||||
|
||||
|
||||
@python_2_unicode_compatible
|
||||
class UserTaskResult(TaskResult):
|
||||
|
||||
user = models.ForeignKey(User)
|
||||
|
||||
objects = UserTaskResultManager()
|
||||
|
||||
def __str__(self):
|
||||
return "{task_uuid} {task_name} {user}".format(
|
||||
task_uuid=self.task_uuid, task_name=self.task_name, user=self.user)
|
||||
|
||||
|
||||
class ShadowManager(models.Manager):
|
||||
|
||||
@transaction.atomic
|
||||
def create_shadow(self, user, password):
|
||||
changedays = (timezone.now().date() - date(1970, 1, 1)).days
|
||||
shadow = self.create(
|
||||
|
@ -356,6 +380,7 @@ class AdditionalGroup(TimeStampedModel, models.Model):
|
|||
if self.user.group == self.group:
|
||||
raise ValidationError(CANNOT_USE_PRIMARY_GROUP_AS_ADDITIONAL)
|
||||
|
||||
@transaction.atomic
|
||||
def save(self, *args, **kwargs):
|
||||
GroupTaskResult.objects.create_grouptaskresult(
|
||||
self.group,
|
||||
|
@ -365,6 +390,7 @@ class AdditionalGroup(TimeStampedModel, models.Model):
|
|||
)
|
||||
super(AdditionalGroup, self).save(*args, **kwargs)
|
||||
|
||||
@transaction.atomic
|
||||
def delete(self, *args, **kwargs):
|
||||
DeleteTaskResult.objects.create_deletetaskresult(
|
||||
'usergroup',
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue