84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
from django.db import models
|
|
from django.utils.text import slugify
|
|
from django.utils import timezone
|
|
|
|
import random
|
|
|
|
import boto3
|
|
|
|
from datetime import datetime
|
|
|
|
import os.path
|
|
|
|
s3client = boto3.client('s3')
|
|
|
|
def generate_code(length=9):
|
|
return "".join([ random.choice('0123456789') for _ in range(length) ])
|
|
|
|
class Ensemble(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
code = models.CharField(max_length=9, default=generate_code)
|
|
passphrase = models.CharField(max_length=100)
|
|
bucket = models.CharField(max_length=255)
|
|
|
|
def active_projects(self):
|
|
return self.projects.filter(active=True)
|
|
|
|
def ensemble_code(self):
|
|
code = str(self.code)
|
|
return "{}-{}-{}".format(code[:3], code[3:6], code[6:])
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Project(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
ensemble = models.ForeignKey(Ensemble, related_name='projects', on_delete=models.CASCADE, null=True)
|
|
active = models.BooleanField(default=True)
|
|
deadline =models.DateField(null=True, blank=True)
|
|
|
|
def submissions(self):
|
|
return self.all_submissions.filter(complete=True).order_by('-pk')
|
|
|
|
def presigned_post(self, object_name, fields=None, conditions=None, expires=3600):
|
|
key = os.path.join(slugify(self.name), object_name)
|
|
return s3client.generate_presigned_post(self.ensemble.bucket, key, Fields=fields or {}, Conditions=conditions or [], ExpiresIn=expires)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Resource(models.Model):
|
|
project = models.ForeignKey(Project, related_name='resources', on_delete=models.CASCADE)
|
|
name = models.CharField(max_length=100)
|
|
uri = models.CharField(max_length=255)
|
|
|
|
class WikiPage(models.Model):
|
|
project = models.ForeignKey(Project, related_name='wiki_pages', on_delete=models.CASCADE)
|
|
title = models.CharField(max_length=255)
|
|
markdown = models.TextField()
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
class Submission(models.Model):
|
|
project = models.ForeignKey(Project, related_name='all_submissions', on_delete=models.CASCADE)
|
|
date = models.DateTimeField(auto_now_add=True, )
|
|
name = models.CharField(max_length=255)
|
|
instrument = models.CharField(max_length=100)
|
|
notes = models.TextField(blank=True)
|
|
complete = models.BooleanField(default=False)
|
|
key = models.CharField(max_length=512, blank=True)
|
|
|
|
def presigned_url(self):
|
|
params = {'Bucket': self.project.ensemble.bucket, 'Key': self.key}
|
|
return s3client.generate_presigned_url('get_object', Params=params, ExpiresIn=3600)
|
|
|
|
def key_template(self):
|
|
return "{}_{}_{}_${{filename}}".format(
|
|
timezone.localtime(self.date).isoformat(timespec='seconds').replace(':', ''),
|
|
slugify(self.name),
|
|
slugify(self.instrument)
|
|
)
|
|
|
|
def __str__(self):
|
|
return f"{self.name}: {self.date}" |