sarlink-portal-api/api/models.py

"""
This is the models module for api.
"""

from django.contrib.auth.models import AbstractUser
from django.db import models
from .managers import CustomUserManager
from django.utils import timezone


class User(AbstractUser):
    email = models.EmailField(unique=True, blank=True, null=True)
    address = models.CharField(max_length=255, blank=True)
    mobile = models.CharField(max_length=255, blank=True)
    designation = models.CharField(max_length=255, blank=True)
    acc_no = models.CharField(max_length=255, blank=True)
    id_card = models.CharField(max_length=255, blank=True)
    verified = models.BooleanField(default=False)
    dob = models.DateField(blank=True, null=True)
    terms_accepted = models.BooleanField(default=False)
    policy_accepted = models.BooleanField(default=False)
    wallet_balance = models.FloatField(default=0.0)
    ninja_user_id = models.CharField(max_length=255, blank=True)
    atoll = models.ForeignKey(
        "Atoll", on_delete=models.SET_NULL, null=True, blank=True, related_name="users"
    )
    island = models.ForeignKey(
        "Island", on_delete=models.SET_NULL, null=True, blank=True, related_name="users"
    )

    def get_all_fields(self, instance):
        return [field.name for field in instance.get_fields()]

    objects = CustomUserManager()


class Atoll(models.Model):
    name = models.CharField(max_length=255, unique=True)
    created_at = models.DateTimeField(default=timezone.now)
    updated_at = models.DateTimeField(auto_now=True)

    def __str__(self):
        return self.name


class Island(models.Model):
    atoll = models.ForeignKey(Atoll, on_delete=models.CASCADE, related_name="islands")
    name = models.CharField(max_length=255, unique=True)
    created_at = models.DateTimeField(default=timezone.now)
    updated_at = models.DateTimeField(auto_now=True)

    def __str__(self):
        return self.name