1
0
Fork 0
mirror of https://github.com/retspen/webvirtcloud synced 2025-07-31 12:41:08 +00:00

Cleaned up UserAttributes model

This commit is contained in:
Real-Gecko 2020-06-16 11:26:58 +06:00
parent 20d4aaac30
commit 2951d0b035
2 changed files with 15 additions and 23 deletions

View file

@ -55,28 +55,6 @@ class UserAttributes(models.Model):
validators=[MinValueValidator(-1)],
)
@staticmethod
def create_missing_userattributes(user):
try:
userattributes = user.userattributes
except UserAttributes.DoesNotExist:
userattributes = UserAttributes(user=user)
userattributes.save()
@staticmethod
def add_default_instances(user):
existing_instances = UserInstance.objects.filter(user=user)
if not existing_instances:
for instance_name in settings.NEW_USER_DEFAULT_INSTANCES:
instance = Instance.objects.get(name=instance_name)
user_instance = UserInstance(user=user, instance=instance)
user_instance.save()
@staticmethod
def configure_user(user):
UserAttributes.create_missing_userattributes(user)
UserAttributes.add_default_instances(user)
def __str__(self):
return self.user.username