diff options
| -rw-r--r-- | python/Dawn/models.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/python/Dawn/models.py b/python/Dawn/models.py index dc8ff7d4..55625724 100644 --- a/python/Dawn/models.py +++ b/python/Dawn/models.py @@ -6,6 +6,7 @@ from sqlalchemy.types import TypeDecorator, LargeBinary from wtforms import SelectField from wtforms.validators import Length from wtforms_alchemy import model_form_factory +from sqlalchemy_utils import EmailType, PhoneNumberType import datetime db = SQLAlchemy() @@ -25,14 +26,16 @@ class Counterparties(db.Model): state = db.Column(db.String(2)) dtc_number = db.Column(db.Integer) sales_contact = db.Column(db.String) - sales_email = db.Column(db.String) - sales_phone = db.Column(db.String) + sales_email = db.Column(EmailType) + sales_phone = db.Column(PhoneNumberType()) valuation_contact1 = db.Column(db.String) - valuation_email1 = db.Column(db.String) + valuation_email1 = db.Column(EmailType) valuation_contact2 = db.Column(db.String) - valuation_email2 = db.Column(db.String) + valuation_email2 = db.Column(EmailType) valuation_contact3 = db.Column(db.String) - valuation_email3 = db.Column(db.String) + valuation_email3 = db.Column(EmailType) + valuation_contact4 = db.Column(db.String) + valuation_email4 = db.Column(EmailType) notes = db.Column(db.String) BOND_STRAT = ENUM('M_STR_MAV', 'M_STR_SMEZZ', 'CSO_TRANCH', |