from django.db import models

from common.base_models import BaseModel



class VendorType(BaseModel):
    name = models.CharField(max_length=255)

    class Meta:
        db_table = 'vendor_types'


class Vendor(BaseModel):
    name = models.CharField(max_length=255)
    type = models.ForeignKey(VendorType, on_delete=models.CASCADE)
    phone_no = models.CharField(max_length=30)
    email = models.EmailField(null=True)
    note = models.TextField(null=True)

    def get_full_name(self):
        if self.created_by:
            return f"{self.created_by.first_name} {self.created_by.last_name}"
        return ""

    class Meta:
        db_table = 'vendors'