from django.db import models

PARTNER_TYPE_CHOICES = (
    ('venue', 'Venue'),
    ('promoter', 'Promoter'),
    ('sponsor', 'Sponsor'),
    ('vendor', 'Vendor'),
    ('affiliate', 'Affiliate'),
    ('other', 'Other'),
)

KYC_DOCUMENT_TYPE_CHOICES = (
    ('aadhaar', 'Aadhaar'),
    ('pan', 'PAN'),
    ('driving_license', 'Driving License'),
    ('voter_id', 'Voter ID'),
    ('passport', 'Passport'),
    ('other', 'Other'),
)

KYC_COMPLIANCE_STATUS_CHOICES = (
    ('pending', 'Pending'),
    ('approved', 'Approved'),
    ('rejected', 'Rejected'),
    ('high_risk', 'High Risk'),
    ('low_risk', 'Low Risk'),
    ('medium_risk', 'Medium Risk'),
)

STATUS_CHOICES = (
    ('active', 'Active'),
    ('inactive', 'Inactive'),
    ('pending', 'Pending'),
    ('suspended', 'Suspended'),
    ('deleted', 'Deleted'),
    ('archived', 'Archived'),
)


class Partner(models.Model):
    name = models.CharField(max_length=250)
    partner_type = models.CharField(max_length=250, choices=PARTNER_TYPE_CHOICES)

    primary_contact_person_name = models.CharField(max_length=250)
    primary_contact_person_email = models.EmailField()
    primary_contact_person_phone = models.CharField(max_length=15)

    status = models.CharField(max_length=250, choices=STATUS_CHOICES, default='active', db_index=True)

    address = models.TextField(blank=True, null=True)
    city = models.CharField(max_length=250, blank=True, null=True)
    state = models.CharField(max_length=250, blank=True, null=True)
    country = models.CharField(max_length=250, blank=True, null=True)
    website_url = models.URLField(blank=True, null=True)

    pincode = models.CharField(max_length=10, blank=True, null=True)
    latitude = models.DecimalField(max_digits=9, decimal_places=6, blank=True, null=True)
    longitude = models.DecimalField(max_digits=9, decimal_places=6, blank=True, null=True)

    is_kyc_compliant = models.BooleanField(default=False)
    kyc_compliance_status = models.CharField(max_length=250, choices=KYC_COMPLIANCE_STATUS_CHOICES, default='pending', db_index=True)
    kyc_compliance_reason = models.TextField(blank=True, null=True)
    kyc_compliance_document_type = models.CharField(max_length=250, choices=KYC_DOCUMENT_TYPE_CHOICES, blank=True, null=True)
    kyc_compliance_document_other_type = models.CharField(max_length=250, blank=True, null=True)
    kyc_compliance_document_file = models.FileField(upload_to='kyc_documents/', blank=True, null=True)
    kyc_compliance_document_number = models.CharField(max_length=250, blank=True, null=True)

    def __str__(self):
        return self.name