Add Save Model in DeviceA24
This commit is contained in:
@@ -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}'
|
||||
Reference in New Issue
Block a user