40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from django.conf import settings
|
|
from django.db import models
|
|
|
|
from core.models.base import BaseModel
|
|
|
|
|
|
class DemoEnvironment(BaseModel):
|
|
class Status(models.TextChoices):
|
|
ACTIVE = "active", "Active"
|
|
EXPIRED = "expired", "Expired"
|
|
CLEANED = "cleaned", "Cleaned"
|
|
|
|
owner_user = models.OneToOneField(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE,
|
|
related_name="demo_environment",
|
|
)
|
|
workspace = models.OneToOneField(
|
|
"workspaces.Workspace",
|
|
on_delete=models.CASCADE,
|
|
related_name="demo_environment",
|
|
)
|
|
expires_at = models.DateTimeField()
|
|
status = models.CharField(max_length=16, choices=Status.choices, default=Status.ACTIVE)
|
|
seed_version = models.CharField(max_length=32, default="v1")
|
|
cleaned_at = models.DateTimeField(blank=True, null=True)
|
|
cleanup_error = models.TextField(blank=True)
|
|
|
|
class Meta:
|
|
db_table = "demo_environment"
|
|
ordering = ("-created_at",)
|
|
indexes = [
|
|
models.Index(fields=["status", "expires_at"], name="demo_status_expires_idx"),
|
|
models.Index(fields=["owner_user"], name="demo_owner_user_idx"),
|
|
models.Index(fields=["workspace"], name="demo_workspace_idx"),
|
|
]
|
|
|
|
def __str__(self):
|
|
return f"Demo {self.workspace_id} for {self.owner_user_id}"
|