Add Save Model in DeviceA24

This commit is contained in:
fjrodriguez
2023-05-04 15:17:04 -06:00
parent f0d67ae4eb
commit 1b07b3bce6
6 changed files with 27 additions and 53 deletions

View File

@@ -26,7 +26,7 @@ class Sistemas_por_cliente_A24(models.Model):
unique_together = ('id_sistema', 'cliente')
class DeviceA24(models.Model):
clienteA24 = models.ForeignKey(ClientesA24, related_name='cliA24',on_delete=models.CASCADE)
clienteA24 = models.ForeignKey(ClientesA24,on_delete=models.CASCADE)
deviceName = models.CharField(max_length=255)
deviceOS = models.CharField(max_length=255)
deviceIP = models.GenericIPAddressField()
@@ -35,14 +35,30 @@ class DeviceA24(models.Model):
sistema = models.ForeignKey(Sistema, on_delete=models.CASCADE)
MAC = models.CharField(max_length=30, blank=True, null=True)
dataBase = models.CharField(max_length=40, blank=True,null=True)
timestamp = models.DateTimeField(auto_now_add=True)
class Meta:
ordering = ('username',)
abstract =True
def generate_username(self):
username_ = f"A24_{self.clienteA24}_{self.deviceName}_{self.MAC}"
""""""
username_ = f"SCAII_{self.clienteA24}_{self.deviceName}_{self.MAC}"
username_ = re.sub(r'\W+', '', username_)
if User.objects.filter(username=username_).exists():
pass
return User.objects.get(username=username_)
return User.objects.create_user(username=username_)
def save(self,*args, **kwargs):
"""Overrides save method"""
if not self.pk:
obj = self.generate_username()
self.username=obj
token, created = Token.objects.get_or_create(user=obj)
self.token = token
super().save(*args, **kwargs)
def __str__(self):
return f'{self.username}'