tabby-web/backend/tabby/app/models.py

import secrets
from datetime import date
from django.db import models
from django.contrib.auth.models import AbstractUser


class Config(models.Model):
    user = models.ForeignKey('app.User', related_name='configs', on_delete=models.CASCADE)
    name = models.CharField(max_length=255)
    content = models.TextField(default='{}')
    last_used_with_version = models.CharField(max_length=32, null=True)
    created_at = models.DateTimeField(auto_now_add=True)
    modified_at = models.DateTimeField(auto_now=True)

    def save(self, *args, **kwargs):
        if not self.name:
            self.name = f'Unnamed config ({date.today()})'
        super().save(*args, **kwargs)


class User(AbstractUser):
    active_config = models.ForeignKey(Config, null=True, on_delete=models.SET_NULL, related_name='+')
    active_version = models.CharField(max_length=32, null=True)
    custom_connection_gateway = models.CharField(max_length=255, null=True, blank=True)
    custom_connection_gateway_token = models.CharField(max_length=255, null=True, blank=True)
    config_sync_token = models.CharField(max_length=255)
    force_pro = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)
    modified_at = models.DateTimeField(auto_now=True)

    def save(self, *args, **kwargs):
        if not self.config_sync_token:
            self.config_sync_token = secrets.token_hex(64)
        super().save(*args, **kwargs)


class Gateway(models.Model):
    host = models.CharField(max_length=255)
    port = models.IntegerField(default=1234)
    admin_port = models.IntegerField(default=1235)
    enabled = models.BooleanField(default=True)
    secure = models.BooleanField(default=True)

    def __str__(self):
        return f'{self.host}:{self.port}'