gpp/port/models.py

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() 
    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):
    num = 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 = 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):
    num = 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):
    num = 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)

    # 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))