gpp/port/models.py

227 lines
7.1 KiB
Python

from django.db import models
from django.db.models import Model, CharField, EmailField, \
DateTimeField, DecimalField, ForeignKey, ManyToManyField, \
IntegerField, ImageField, BooleanField, \
CASCADE, SET_NULL, PROTECT
from django_countries.fields import CountryField
from phonenumber_field.modelfields import PhoneNumberField
class Address(Model):
address = CharField(max_length=200)
zip_code = CharField(max_length=10)
city = CharField(max_length=200)
country = CountryField()
def __str__(self):
return '{}, {} - {}, {}'.format(self.address, self.zip_code,
self.city, self.country)
class Person(Model):
name = CharField(max_length=50,null=True)
surname = CharField(max_length=50,null=True)
nationality = CountryField(null=True)
email = EmailField(max_length=254,null=True)
phone = PhoneNumberField(null=True)
# Foreign keys
address = ForeignKey(Address,on_delete=SET_NULL,null=True)
def __str__(self):
return '{} {}'.format(self.name, self.surname)
class Company(Model):
name = CharField(max_length=50)
email = EmailField(max_length=200,null=True)
phone = PhoneNumberField(null=True)
registration_num = CharField(max_length=50,null=True)
# Foreign keys
address = ForeignKey(Address,on_delete=SET_NULL,null=True)
def __str__(self):
return '{}'.format(self.name)
class Insurance(Company):
pass
class Boat(Model):
name = CharField(max_length=50)
registration_num = CharField(max_length=20,null=True)
length = DecimalField(max_digits=7, decimal_places=2, null=True)
beam = DecimalField(max_digits=7, decimal_places=2, null=True)
water_draught = DecimalField(max_digits=7, decimal_places=2, null=True)
tonnage = DecimalField(max_digits=7, decimal_places=2, null=True)
water_tank = DecimalField(max_digits=7, decimal_places=2, null=True)
model = CharField(max_length=50,null=True)
heating = CharField(max_length=50,null=True)
passenger_capacity = IntegerField(null=True)
picture = ImageField(upload_to='uploads/', height_field=None,
width_field=None, max_length=100, null=True)
# Foreign keys
company = ForeignKey(Company,on_delete=SET_NULL,null=True,related_name='+')
boat_insurance = ManyToManyField(
Insurance,
through='BoatInsurance',
through_fields=('boat', 'insurance')
)
persons = ManyToManyField(
Person,
through='SailsOn',
through_fields=('boat', 'person')
)
def __str__(self):
return '{}'.format(self.name)
class SailsOn(Model):
boat = ForeignKey(Boat,on_delete=PROTECT)
person = ForeignKey(Person,on_delete=PROTECT)
is_captain = BooleanField()
is_crew = BooleanField()
is_owner = BooleanField()
is_guest = BooleanField()
is_pet = BooleanField()
class BoatInsurance(Model):
contract = IntegerField(null=True)
date = DateTimeField(auto_now_add=True)
# Foreign keys
insurance = ForeignKey(Insurance,on_delete=PROTECT)
boat = ForeignKey(Boat,on_delete=PROTECT)
class Port(Model):
name = CharField(max_length=50)
# Foreign keys
address = ForeignKey(Address,on_delete=PROTECT)
company = ForeignKey(Company,on_delete=PROTECT)
employees = ManyToManyField(
Person,
through='Employee',
through_fields=('port', 'person')
)
def __str__(self):
return '{}'.format(self.name)
class Employee(Model):
position = CharField('Job', max_length=20, null=True)
# Foreign keys
port = ForeignKey(Port,on_delete=PROTECT)
person = ForeignKey(Person,on_delete=PROTECT)
def __str__(self):
return '{}, {} at {}'.format(str(self.person), self.position,
str(self.port))
class Dock(Model):
name = CharField(max_length=10)
length = DecimalField(max_digits=7, decimal_places=2, null=True)
width = DecimalField(max_digits=7, decimal_places=2, null=True)
depth_min = DecimalField(max_digits=7, decimal_places=2, null=True)
depth_max = DecimalField(max_digits=7, decimal_places=2, null=True)
# Foreign keys
port = ForeignKey(Port,on_delete=PROTECT)
def __str__(self):
return '{} : {}'.format(str(self.port), self.num)
class Plug(Model):
name = CharField(max_length=10)
amperage = DecimalField(max_digits=7, decimal_places=2)
voltage = DecimalField(max_digits=7, decimal_places=2)
# Foreign keys
port = ForeignKey(Port,on_delete=PROTECT)
def __str__(self):
return '{} : {}'.format(str(self.port), self.num)
class Tap(Model):
name = CharField(max_length=10)
# Foreign keys
port = ForeignKey(Port,on_delete=PROTECT)
def __str__(self):
return '{} : {}'.format(str(self.port), self.num)
class Payment(Model):
num = CharField(max_length=40)
date = DateTimeField()
amount = DecimalField(max_digits=7, decimal_places=2)
pay_type = CharField(
max_length=3,
choices=[
('CSH', 'Cash'),
('CHK', 'Check'),
('TSF', 'Transfer'),
('CRD', 'Credit Card')
],
default='CSH'
)
def __str__(self):
return '{} at {}'.format(self.num, str(self.date))
class Service(Model):
name = CharField(max_length=50)
def __str__(self):
return '{}'.format(self.name)
class Bill(Model):
num = IntegerField()
date = DateTimeField()
total = DecimalField(max_digits=7, decimal_places=2)
# Foreign keys
payments = ManyToManyField(
Payment,
through='BillPayment',
through_fields=('bill', 'payment')
)
def __str__(self):
return '{} at {}'.format(self.num, str(self.date))
class BillLine(Model):
quantity = DecimalField(max_digits=7, decimal_places=2)
value = DecimalField(max_digits=7, decimal_places=2, default=0.)
# Foreign keyks
service = ForeignKey(Service,on_delete=PROTECT)
bill = ForeignKey(Bill,on_delete=PROTECT)
def __str__(self):
return '{}x {} - {}'.format(self.quantity, str(self.service),
str(self.value))
class BillPayment(Model):
amount = DecimalField(max_digits=7, decimal_places=2)
# Foreign keys
bill = ForeignKey(Bill,on_delete=PROTECT)
payment = ForeignKey(Payment,on_delete=PROTECT)
class Stay(Model):
arrival = DateTimeField()
departure = DateTimeField(null=True)
coming_from = CharField(max_length=200,null=True)
going_to = CharField(max_length=200,null=True)
# Foreign keys
boat = ForeignKey(Boat,on_delete=PROTECT)
bill = ForeignKey(Bill,on_delete=PROTECT,null=True)
def __str__(self):
return '{} - {} : {}'.format(self.arrival, self.departure, str(self.boat))
class Mooring(Model):
date = DateTimeField()
# Foreign keys
stay = ForeignKey(Stay,on_delete=CASCADE)
dock = ForeignKey(Dock,on_delete=PROTECT)
tap = ForeignKey(Tap,on_delete=PROTECT,null=True)
plug = ForeignKey(Plug,on_delete=PROTECT,null=True)
# Methods
def __str__(self):
return '{} - {} at {}'.format(str(self.date), str(self.stay.boat), str(self.dock))