from sqlalchemy import Table, MetaData, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Boolean, Sequence, Column, Integer, String, Date, Float, ForeignKey
from sqlalchemy.dialects.postgresql import ENUM, OID
from sqlalchemy import create_engine
from wtforms_alchemy import ModelForm
from wtforms import IntegerField
from jinja2 import Environment, FileSystemLoader

engine = create_engine('postgresql://dawn_user@debian/dawndb')
Base = declarative_base(engine)

class Counterparties(Base):
    __tablename__ = 'counterparties'
    code = Column(String(12), primary_key=True)
    name = Column(String)
    city = Column(String)
    state = Column(String(2))
    dtc_number = Column(Integer)
    sales_contact = Column(String)
    sales_email = Column(String)
    sales_phone = Column(String)
    valuation_contact1 = Column(String)
    valuation_email1 = Column(String)
    valuation_contact2 = Column(String)
    valuation_email2 = Column(String)
    valuation_contact3 = Column(String)
    valuation_email3 = Column(String)
    notes = Column(String)

BOND_STRAT = ENUM('M_STR_MAV', 'M_STR_SMEZZ', 'CSO_TRANCH',
                  'M_CLO_BB20', 'M_CLO_AAA', 'M_CLO_BBB', 'M_MTG_IO', 'M_MTG_THRU',
                  'M_MTG_GOOD', 'M_MTG_B4PR', 'M_MTG_RW', name='bond_strat', metadata=Base.metadata)
ASSET_CLASS = ENUM('CSO', 'Subprime', 'CLO', 'Tranches', 'Futures', 'Cash', 'FX', 'Cleared',
                   name='asset_class', metadata=Base.metadata)

class BondDeal(Base):
    __tablename__ = 'bonds'
    dealid = Column(String(28), primary_key=True)
    folder = Column(BOND_STRAT, nullable=False)
    custodian = Column(String(12), nullable=False)
    cashaccount = Column(String(10), nullable=False)
    counterparty = Column(String, ForeignKey("counterparties.code"), nullable=False)
    trade_date = Column(Date, nullable = False)
    settle_date = Column(Date, nullable = False)
    cusip = Column(String(9))
    isin = Column(String(12))
    description = Column(String(32))
    buysell = Column(Boolean, nullable = False)
    faceamount = Column(Float, nullable=False)
    price = Column(Float, nullable=False)
    account = Column(String(10))
    accrued = Column(Float)
    asset_class = Column(ASSET_CLASS)
    ticket = Column(OID, info={'form_field_class': IntegerField})

# Base.metadata.create_all(engine)

class BondForm(ModelForm):
    class Meta:
        model = BondDeal

form = BondForm()
jinja_env = Environment(loader = FileSystemLoader("."))
print(jinja_env.get_template("trade_entry.html").render(form = form))