def addNewDevice():
if(request.method == 'POST'):
dvcName = request.form['dvcName']
dvcType = request.form['dvcType']
dvcPurDate = request.form['dvcPurDate']
dvcWarPeriod = request.form['dvcWarPeriod']
# Convert date time string to ISODate.
dvcPurDate = dateutil.parser.parse(dvcPurDate)
session_id = request.cookies['session']
userid = session.getSessionUserInfo(session_id)
appuserdevice = AppUserDeviceDB(userid, dvcName, dvcType, dvcPurDate, dvcWarPeriod)
appuserdevice.addNewUserDevice(db)
return(redirect(url_for('home'), code=302))
def home():
session_id = request.cookies['session']
user = session.getSessionUserInfo(session_id)
if user is None:
template = 'index.html'
else:
# Now if there is a user before rendering the template we would like to populate the table
table = populate_user_device_table(user, db)
template = 'user_screen.html'
if table is not None:
in_warranty_table = table[0]
out_warranty_table = table[1]
else:
in_warranty_table = out_warranty_table = 'No Devices found'
return(render_template(template, in_warranty_table=in_warranty_table,\
out_warranty_table=out_warranty_table))
def addNewDevice():
if (request.method == 'POST'):
dvcName = request.form['dvcName']
dvcType = request.form['dvcType']
dvcPurDate = request.form['dvcPurDate']
dvcWarPeriod = request.form['dvcWarPeriod']
# Convert date time string to ISODate.
dvcPurDate = dateutil.parser.parse(dvcPurDate)
session_id = request.cookies['session']
userid = session.getSessionUserInfo(session_id)
appuserdevice = AppUserDeviceDB(userid, dvcName, dvcType, dvcPurDate,
dvcWarPeriod)
appuserdevice.addNewUserDevice(db)
return (redirect(url_for('home'), code=302))
def home():
session_id = request.cookies['session']
user = session.getSessionUserInfo(session_id)
if user is None:
template = 'index.html'
else:
# Now if there is a user before rendering the template we would like to populate the table
table = populate_user_device_table(user, db)
template = 'user_screen.html'
if table is not None:
in_warranty_table = table[0]
out_warranty_table = table[1]
else:
in_warranty_table = out_warranty_table = 'No Devices found'
return(render_template(template, in_warranty_table=in_warranty_table,\
out_warranty_table=out_warranty_table))
def updateUsrPassword():
if request.method == 'POST':
cur_passwd = request.form['oldPasswd']
new_passwd = request.form['newPasswd']
conf_new_passwd = request.form['confPasswd']
session_id = request.cookies['session']
userid = session.getSessionUserInfo(session_id)
# We can use validate_user() to know whether current password is correct
cur_user = AppUserDB(userid, cur_passwd)
is_correct_passwd = cur_user.validate_user(userid, cur_passwd, db)
if not is_correct_passwd:
return (redirect(url_for('home'), code=302))
if new_passwd == conf_new_passwd:
cur_user.update_user_pwd(userid, new_passwd, db)
return (redirect(url_for('home'), code=302))
def updateUsrPassword():
if request.method == 'POST':
cur_passwd = request.form['oldPasswd']
new_passwd = request.form['newPasswd']
conf_new_passwd = request.form['confPasswd']
session_id = request.cookies['session']
userid = session.getSessionUserInfo(session_id)
# We can use validate_user() to know whether current password is correct
cur_user = AppUserDB(userid, cur_passwd)
is_correct_passwd = cur_user.validate_user(userid, cur_passwd, db)
if not is_correct_passwd:
return(redirect(url_for('home'), code=302))
if new_passwd == conf_new_passwd:
cur_user.update_user_pwd(userid, new_passwd, db)
return(redirect(url_for('home'), code=302))
