You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import AbstractUser
|
|
from django.utils import timezone
|
|
|
|
|
|
class User(AbstractUser):
|
|
"""Custom User model with role and admin flag"""
|
|
ROLE_CHOICES = [
|
|
('employee', 'Employee'),
|
|
('viewer', 'Viewer'),
|
|
]
|
|
|
|
role = models.CharField(max_length=20, choices=ROLE_CHOICES, default='viewer')
|
|
is_admin = models.BooleanField(default=False)
|
|
|
|
class Meta:
|
|
db_table = 'api_user'
|
|
|
|
def __str__(self):
|
|
return f"{self.username} ({self.get_role_display()})"
|
|
|
|
|
|
class ComPortReading(models.Model):
|
|
"""Model to store serial port readings"""
|
|
port = models.CharField(max_length=20)
|
|
data = models.TextField()
|
|
timestamp = models.DateTimeField(auto_now_add=True)
|
|
source_ip = models.GenericIPAddressField(null=True, blank=True)
|
|
|
|
class Meta:
|
|
ordering = ['-timestamp']
|
|
indexes = [
|
|
models.Index(fields=['-timestamp']),
|
|
models.Index(fields=['port', '-timestamp']),
|
|
]
|
|
|
|
def __str__(self):
|
|
return f"{self.port} - {self.timestamp}"
|