from django.db import models

class UserProfile(models.Model):
    id = models.CharField(max_length=8, primary_key=True)
    username_up = models.CharField(max_length=16)
    email = models.EmailField()
    password_up = models.CharField(max_length=16)
    access_level = models.IntegerField(default=0)  # Access level: 0 - Regular user, 1 - Admin
    public_key = models.CharField(max_length=128, blank=True, null=True)
    private_key = models.CharField(max_length=128, blank=True, null=True)

    def __str__(self):
        return self.username_up

class Document(models.Model):
    document_id = models.AutoField(primary_key=True)
    document_owner = models.CharField(max_length=255)
    document_name = models.CharField(max_length=255)
    issuing_office = models.CharField(max_length=255)
    issue_date = models.DateTimeField()
    security_level = models.CharField(max_length=50)
    cc_office = models.CharField(max_length=255)
    file_type = models.CharField(max_length=50)
    modifier = models.CharField(max_length=50)
    modified_date = models.DateTimeField()
    file_address = models.CharField(max_length=255)
    is_sent = models.IntegerField(default=0)

    def __str__(self):
        return self.document_name

class AuditLog(models.Model):
    user_number = models.CharField(max_length=8)
    login_successful = models.BooleanField()
    login_time = models.DateTimeField()
    logout_time = models.DateTimeField()
    document = models.CharField(max_length=2000)
    document_status = models.IntegerField()  # 可能需要 choices 参数定义选择项
    document_creation_time = models.DateTimeField()
    document_deletion_time = models.DateTimeField()
    document_send_time = models.DateTimeField()
    document_view_time = models.DateTimeField()

    def __str__(self):
        return f"Log - User: {self.user_number}, Document: {self.document}"