1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
from django.conf import settings
from django.contrib.auth import get_user_model
from django.core.files.storage import FileSystemStorage
from django.db import models
from django.utils.html import format_html
from django.utils.safestring import mark_safe
import uuid
# For handling file uploads to a permenant direcrory
file_storage = FileSystemStorage(
location=settings.FILE_STORAGE_ROOT, base_url=settings.FILE_STORAGE_URL)
media = FileSystemStorage(
location=settings.MEDIA_ROOT, base_url='.')
class BookCategory(models.Model):
id = models.AutoField(primary_key=True)
category_name = models.CharField(max_length=500)
def __str__(self):
return self.category_name
class Book(models.Model):
id = models.AutoField(primary_key=True)
book_name = models.CharField(max_length=500)
author_name = models.CharField(max_length=500)
category = models.ForeignKey(BookCategory, on_delete=models.CASCADE)
def __str__(self):
return self.book_name
class StateSave(models.Model):
id = models.AutoField(primary_key=True)
save_id = models.UUIDField(default=uuid.uuid4)
name = models.CharField(max_length=100, default="Untitled")
description = models.CharField(max_length=400, null=True)
shared = models.BooleanField(default=False)
create_time = models.DateTimeField(auto_now_add=True)
save_time = models.DateTimeField(auto_now=True, db_index=True)
owner = models.ForeignKey(get_user_model(), null=True, on_delete=models.CASCADE)
data_dump = models.TextField(null=False)
base64_image = models.ImageField(
upload_to='simulation_images', storage=file_storage, null=True)
script_dump = models.TextField(null=True)
def save(self, *args, **kwargs):
super(StateSave, self).save(*args, **kwargs)
def __str__(self):
return f"{self.save_id.hex} - {self.name}"
class Gallery(models.Model):
id = models.AutoField(primary_key=True)
save_id = models.CharField(unique=True, max_length=50, null=False)
name = models.CharField(max_length=100, default="Untitled")
description = models.CharField(max_length=400, null=True)
save_time = models.DateTimeField(auto_now=True)
book = models.ForeignKey(Book, related_name='examples', on_delete=models.CASCADE, null=True)
data_dump = models.TextField(null=False)
blocks = models.CharField(max_length=300, null=True)
media = models.CharField(max_length=100, null=True)
script_dump = models.TextField(null=True)
def lcname(self):
return self.name.lower()
def lcdescription(self):
return self.description.lower()
# For Django Admin Panel
def image_tag(self):
if self.media:
return format_html('<img src="{}" style="width: 45px; height:45px;" />',
mark_safe(self.media.url))
else:
return 'No Image Found'
image_tag.short_description = 'Image'
def __str__(self):
return f"{self.save_id.hex} - {self.name}"
|
