def get_user_by_email(self, email, partnercode):
try:
return User.select().where((User.email == email)
& (User.partnercode == partnercode)
& (User.deleted == 0)).get()
except User.DoesNotExist:
return None
class UserController:
def __init__(self):
self.model = User()
self.view = UserView()
self.menu()
def menu(self):
while True:
# Render User Menu and get command value
command = self.view.menu()
if command == 0:
return
elif command == 1:
self.addUser()
elif command == 4:
self.deleteUser()
elif command == 5:
self.listUser()
else:
pass
def listUser(self):
res = self.model.select()
self.view.listUser(res)
def addUser(self):
data = self.view.addUser()
data['password'] = bcrypt.hashpw(data['password'].encode('utf-8'),
bcrypt.gensalt()).decode('utf-8')
if self.model.insert(data):
print("OK")
else:
print("Failed")
def deleteUser(self):
data = self.view.deleteUser()
if data == 'cancelled':
return
else:
if self.model.delete(data):
print("Deleted")
else:
print("Failed")
