reimbursinator/back/backend/models.py

55 lines
2.2 KiB
Python

from django.db import models
from django.conf import settings
import datetime
class Report(models.Model):
user_id = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
title = models.CharField(max_length=128)
date_created = models.DateTimeField('date created', default=datetime.date.today)
date_submitted = models.DateTimeField('date submitted', default=datetime.date.today)
submitted = models.BooleanField(default=False)
def __str__(self):
return self.title
class Section(models.Model):
report_id = models.ForeignKey(Report, on_delete=models.CASCADE)
auto_submit = models.BooleanField()
required = models.BooleanField()
completed = models.BooleanField()
title = models.CharField(max_length=256)
html_description = models.TextField()
number = models.IntegerField()
def __str__(self):
return "{0}(#{1})".format(self.title, self.number)
class Field(models.Model):
section_id = models.ForeignKey(Section, on_delete=models.CASCADE)
label = models.CharField(max_length=512)
number = models.IntegerField()
type = models.CharField(max_length=128)
completed = models.BooleanField(default=False)
data_bool = models.BooleanField(default=False)
data_decimal = models.DecimalField(max_digits=9,decimal_places=2, null=True, blank=True)
data_date = models.DateField(default=datetime.date.today)
data_file = models.FileField(upload_to='uploads/%Y/%m/%d/', max_length=512, null=True, blank=True)
data_string = models.TextField(default='', blank=True)
data_integer = models.IntegerField(default=0, blank=True)
def __str__(self):
if self.type == "boolean":
if self.data_bool:
return "True"
else:
return "False"
elif self.type == "decimal":
return "{}".format(self.data_decimal)
elif self.type == "date":
return "{}".format(self.data_date)
elif self.type == "file":
return "{}".format(self.data_file)
elif self.type == "string":
return "{}".format(self.data_string)
elif self.type == "integer":
return "{}".format(self.data_integer)