def check_duplicate_email(self, email):
duplicate = False
with CustomerDatabaseUtils() as db:
for cust in db.getAllCustomers():
if cust[4] == email:
duplicate = True
return duplicate
def login(self, email, password):
find = False
with CustomerDatabaseUtils() as db:
for cust in db.getAllCustomers():
find = False
if cust[4] == email and self.verify_password(
cust[5], password):
find = True
break
return find
def insertCustomer(self, fname, lastname, dob, email, password, phone,
plan, expiry):
with CustomerDatabaseUtils() as db:
db.insertDummy(fname, lastname, dob, email,
self.hash_password(password), phone, plan, expiry)
def createCustomerTable(self):
with CustomerDatabaseUtils() as db:
db.deleteTable()
db.createCustomerTable()
def getCustomer(self, email):
with CustomerDatabaseUtils() as db:
for cust in db.getCustomer(email):
return cust
return None
def register(self, fname, lastname, dob, email, password, phone):
with CustomerDatabaseUtils() as db:
db.insertCustomer(fname, lastname, dob, email,
self.hash_password(password), phone, 0)
def set_premium_expiry(self, cust_id):
with CustomerDatabaseUtils() as db:
today = date.today()
from datetime import timedelta
today = today + timedelta(days=30)
return db.set_premium_expiry(today, cust_id)
def set_plan(self, plan, cust_id):
with CustomerDatabaseUtils() as db:
if plan == 0:
db.set_plan_premium(cust_id)
else:
db.set_plan_standard(cust_id)
def getAllCustomers(self):
with CustomerDatabaseUtils() as db:
return db.getAllCustomers()
def resetPassword(self, password, cust_id):
with CustomerDatabaseUtils() as db:
db.resetPassword(self.hash_password(password), cust_id)