dumuzid/datastore/models/crew.py

71 lines
2.7 KiB
Python

from django.db import models
from .places import Nationality, ZipPlace, Country, Airport, PhoneCode
from .competences import Competence, Role
from ..helpers.uploads import get_upload_path
class Address(models.Model):
address = models.TextField()
zipplace = models.ForeignKey(ZipPlace, models.PROTECT)
class CrewMember(models.Model):
pin = models.IntegerField(primary_key=True)
first_name = models.CharField(max_length=128)
middle_name = models.CharField(max_length=128, null=True, blank=True)
last_name = models.CharField(max_length=128)
calling_name = models.CharField(max_length=128, null=True, blank=True)
suffix_name = models.CharField(max_length=128, null=True, blank=True)
title = models.CharField(max_length=128, null=True, blank=True)
initials = models.CharField(max_length=16, null=True, blank=True)
profile_picture = models.ImageField(
upload_to=get_upload_path, null=True, blank=True)
nationality = models.ForeignKey(
Nationality, models.PROTECT, null=True, blank=True)
place_of_birth = models.CharField(max_length=128, null=True, blank=True)
country_of_birth = models.ForeignKey(
Country, models.PROTECT, null=True, blank=True)
first_address = models.ForeignKey(
Address, models.PROTECT, related_name="crew_first", null=True, blank=True)
second_address = models.ForeignKey(
Address, models.PROTECT, null=True, related_name="crew_second", blank=True)
closest_airport = models.ForeignKey(
Airport, models.PROTECT, related_name="crew_first", null=True, blank=True)
second_airport = models.ForeignKey(
Airport, models.PROTECT, null=True, related_name="crew_second", blank=True)
roles = models.ManyToManyField(Role, null=True, blank=True)
competences = models.ManyToManyField(Competence, null=True, blank=True)
@property
def all_competences(self):
return self.competences.all().union(*[role.competences.all() for role in self.roles.all()])
def __str__(self):
return f"{self.pin}: {self.first_name} {self.last_name}"
class Phone(models.Model):
crew = models.ForeignKey(CrewMember, models.CASCADE)
idc = models.ForeignKey(PhoneCode, models.PROTECT, null=True)
number = models.CharField(max_length=128)
confirmed = models.BooleanField(default=True)
priority = models.IntegerField(default=0)
def __str__(self):
return f"{self.crew.pin}: {self.number}"
class Email(models.Model):
crew = models.ForeignKey(CrewMember, models.CASCADE)
email = models.EmailField()
confirmed = models.BooleanField(default=True)
priority = models.IntegerField(default=0)
def __str__(self):
return f"{self.crew.pin}: {self.email}"