from werkzeug.security import generate_password_hash, check_password_hash
from datetime import date
import json
class User:
def __init__(self, name, partysize, password, dob):
self._name = name # variables with self prefix become part of the object,
self._partysize = partysize
self.set_password(password)
self._dob = dob
@property
def name(self):
return self._name
# a setter function, allows name to be updated after initial object creation
@name.setter
def name(self, name):
self._name = name
# a getter method, extracts email from object
@property
def partysize(self):
return self._partysize
# a setter function, allows name to be updated after initial object creation
@partysize.setter
def partysize(self, partysize):
self._partysize = partysize
# check if uid parameter matches user id in object, return boolean
def is_partysize(self, partysize):
return self._partysize == partysize
# dob property is returned as string, to avoid unfriendly outcomes
@property
def dob(self):
dob_string = self._dob.strftime('%m-%d-%Y')
return dob_string
# dob should be have verification for type date
@dob.setter
def dob(self, dob):
self._dob = dob
# age is calculated and returned each time it is accessed
@property
def age(self):
today = date.today()
return today.year - self._dob.year - ((today.month, today.day) < (self._dob.month, self._dob.day))
# dictionary is customized, removing password for security purposes
@property
def dictionary(self):
dict = {
"name" : self.name,
"partysize" : self.partysize,
"dob" : self.dob,
"age" : self.age
}
return dict
# update password, this is conventional setter
def set_password(self, password):
"""Create a hashed password."""
self._password = generate_password_hash(password, method='sha256')
# check password parameter versus stored/encrypted password
def is_password(self, password):
"""Check against hashed password."""
result = check_password_hash(self._password, password)
return result
# output content using json dumps, this is ready for API response
def __str__(self):
return json.dumps(self.dictionary)
# output command to recreate the object, uses attribute directly
def __repr__(self):
return f'User(name={self._name}, partysize={self._partysize}, password={self._password},dob={self._dob})'
if __name__ == "__main__":
u1 = User(name='Jacky Joe', partysize='4', password='JJJ44#', dob=date(2000, 4, 19))
print("User Information:\n", u1, "\n")