import uuid
from django.db import models
from datetime import datetime
from django.utils.translation import ugettext_lazy as _

APISource = (
    (1, 'Tripjack'),
    (2, 'TBO'),
    (3, 'goflysmart'),
    (4, 'Air IQ'),
    (5, 'Travel Z Gear (Cheap Fix Departure)'),
    (6, 'Fare Boutique'),
    (7, 'Nextra'),
    (8, 'Book N Travel'),
    (9, 'Travel O Pedia')
)

PlanType = (
    (1, 'Flight'),
    (2, 'Hotel')
)

APIMode = (
    (0, 'Development'),
    (1, 'Prod')
)


class TbAgencyApiKey(models.Model):
    id = models.BigAutoField(db_column='APISysId', primary_key=True)
    SecurityKey = models.CharField(max_length=200, verbose_name=_('Security Key'))
    api_source_sys_id = models.IntegerField(db_column='APISourceSysId', choices=APISource)
    plan_type = models.IntegerField(db_column='PlanType', choices=PlanType)
    api_mode = models.IntegerField(db_column='APIMode', choices=APIMode, default=0)
    ClientCode = models.CharField(max_length=200, verbose_name=_('ClientCode'), blank=True, null=True)
    UserName = models.CharField(max_length=200, verbose_name=_('UserName'), blank=True, null=True)
    SecretKey = models.CharField(max_length=200, verbose_name=_('Password'), blank=True, null=True)
    created_date = models.DateTimeField(db_column='CreatedDate', default=datetime.now, blank=True, editable=False)
    is_active = models.BooleanField(db_column='IsActive', default=True)
    is_mark_for_del = models.BooleanField(db_column='IsMarkForDel', default=False, editable=False)

    class Meta:
        managed = True
        db_table = 'TB_Agency_API'
        verbose_name = "Agency API"
        verbose_name_plural = "Agency API"

    def __str__(self):
        return str(self.id)


class TbAgencyApiToken(models.Model):
    id = models.BigAutoField(db_column='APITokenSysId', primary_key=True)
    TbAgencyApiKey = models.ForeignKey(TbAgencyApiKey, on_delete=models.CASCADE, verbose_name=_('Agency Api Key'))
    SecurityKey = models.CharField(max_length=200, verbose_name=_('Security Key'))
    Token = models.CharField(max_length=2000, verbose_name=_('Token'), blank=True, null=True)
    ExpireTime = models.CharField(max_length=200, verbose_name=_('ExpireTime'), blank=True, null=True)
    created_date = models.DateTimeField(db_column='CreatedDate', default=datetime.now, blank=True, editable=False)
    is_active = models.BooleanField(db_column='IsActive', default=True)
    is_mark_for_del = models.BooleanField(db_column='IsMarkForDel', default=False, editable=False)

    class Meta:
        managed = True
        db_table = 'TB_Agency_API_Token'
        verbose_name = "Agency API Token"
        verbose_name_plural = "Agency Token"

    def __str__(self):
        return str(self.id)