from customer_model import db, Customer
# tell logger where to save
LOG_FILE = datetime.datetime.now().strftime('%Y-%m-%d') + '.log'
FH = logging.FileHandler(LOG_FILE)
# tell logger how to save each row
LOG_FORMAT = "%(asctime)s %(filename)s:%(lineno)-3d %(levelname)s %(message)s"
FORMATTER = logging.Formatter(LOG_FORMAT)
FH.setFormatter(FORMATTER)
# get that f****r going
logging.basicConfig(level=logging.INFO)
LOGGER = logging.getLogger()
LOGGER.addHandler(FH)
Customer.drop_table()
Customer.create_table()
def add_customer(_name, _last_name, _home_address, _phone_number,
_email_address, _status, _poverty_score):
'''add new customer function.'''
try:
with db.transaction():
new_customer = Customer.create(name=_name,
last_name=_last_name,
home_address=_home_address,
phone_number=_phone_number,
email_address=_email_address,
status=_status,
poverty_score=_poverty_score)
new_customer.save()
def create_customer_table():
""" create Customer table if not exists """
with database:
Customer.create_table()
logger.info('Customer table create successful')
database.close()
