def update(self, admin=None, employee_number= None, username=None, password=None, email=None):
try:
if email:
email_is_valid(email)
if username:
username_is_valid(username)
if password:
password_is_valid(password)
if employee_number:
employee_number_is_valid(employee_number)
except UserErrors.InvalidEmail as e:
return e
except UserErrors.InvalidUsername as e:
return e
except UserErrors.InvalidPassword as e:
return e
except UserErrors.InvalidEmployeeNumber as e:
return e
self.username = self.username if username is None else username
self.password = self.password if password is None else password
self.email = self.email if email is None else email
self.employee_number = self.employee_number if employee_number is None else employee_number
self.admin = self.admin if admin is None else admin
db.session.commit()
def register(self):
if not email_is_valid(self.email):
raise UserErrors.InvalidEmail("Wrong email format.")
if User.query.filter_by(email=self.email).first() is not None:
raise UserErrors.EmailExists("Email already exists")
if User.query.filter_by(employee_number=self.employee_number).first() is not None:
raise UserErrors.EmployeeNumberExists("User number:{} is already used.".format(self.employee_number))
self.save_to_database()
