54 lines
2.3 KiB
Python
54 lines
2.3 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
from users.models import Agency, AgencyGroup
|
|
# Create your models here.
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
|
|
def user_directory_path(instance, filename):
|
|
# file will be uploaded to MEDIA_ROOT/agency_<id>/files/<subdirs>/<filename>
|
|
return 'agencydata/agency_{0}/files/{1}'.format(instance.agency.pk, filename)
|
|
|
|
|
|
class Data(models.Model):
|
|
subdir = models.CharField(max_length=2000, default="", blank=True, null=True)
|
|
file = models.FileField(null=True, max_length=255, upload_to=user_directory_path)
|
|
date_created = models.DateTimeField(default = timezone.now)
|
|
date_last_modified = models.DateTimeField(default = timezone.now)
|
|
owner = models.ForeignKey(User, on_delete=models.PROTECT)
|
|
agency = models.ForeignKey(Agency, on_delete=models.CASCADE, default=None)
|
|
|
|
def __str__(self):
|
|
return str(self.file.name)
|
|
|
|
|
|
|
|
class DataDir(models.Model):
|
|
name = models.CharField(max_length=2000, default="", blank=True, null=True)
|
|
is_root = models.BooleanField(default=False)
|
|
dirs = models.ManyToManyField('self', blank=True, related_name='dirs_in_dirs', symmetrical = False)
|
|
visibleby = models.ManyToManyField(AgencyGroup, blank=True, related_name='visible_by_user')
|
|
date_created = models.DateTimeField(default = timezone.now)
|
|
date_last_modified = models.DateTimeField(default = timezone.now)
|
|
owner = models.ForeignKey(User, on_delete=models.PROTECT, blank=True, null=True)
|
|
agency = models.ForeignKey(Agency, on_delete=models.CASCADE)
|
|
parent = models.ForeignKey('DataDir', on_delete=models.CASCADE, blank=True, null=True, related_name='dir_in_dir')
|
|
|
|
def __str__(self):
|
|
return str(self.name)
|
|
|
|
|
|
class DataFile(models.Model):
|
|
name = models.CharField(max_length=2000, default="", blank=True, null=True)
|
|
file = models.FileField(null=True, max_length=255, upload_to=user_directory_path, blank=True)
|
|
date_created = models.DateTimeField(default = timezone.now)
|
|
date_last_modified = models.DateTimeField(default = timezone.now)
|
|
owner = models.ForeignKey(User, on_delete=models.PROTECT, default=None, blank=True, null=True)
|
|
agency = models.ForeignKey(Agency, on_delete=models.CASCADE, default=None, blank=True, null=True)
|
|
parent = models.ForeignKey(DataDir, on_delete=models.PROTECT, related_name='thisfileindir', blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return str(self.name)
|
|
|