from polymorphic.models import PolymorphicModel
from phonenumber_field.modelfields import PhoneNumberField

from django.db.models import OneToOneField, CASCADE, CharField, ImageField
from django.contrib.auth import get_user_model

from core.helpers.files import generate_storage_filename

class Profile(PolymorphicModel):
    user = OneToOneField(get_user_model(), CASCADE)
    mobile = PhoneNumberField(blank=True)

class AdminProfile(Profile):
    role = CharField(max_length=255)
    image = ImageField(null=True, blank=True, upload_to=generate_storage_filename)
    display_name = CharField("Internal Display Name", max_length=255, null=True, blank=True)

    @property
    def get_internal_name(self):
        return self.display_name if self.display_name else self.user.get_full_name