def displayMyAccount():
"""Fetches the info of the logged in user and allow him or her to modify it"""
if current_user.firsttimelogin == True:
flash('First you need to update your password.')
return redirect(url_for('reset'))
title = current_user.fname + ' ' + current_user.lname + ' Account'
formEmployee = EmployeeForm()
formLogout = LogoutForm()
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
if formEmployee.submit.data and formEmployee.is_submitted():
fn = formEmployee.first_name.data
ln = formEmployee.last_name.data
pn = formEmployee.phone_number.data
em = formEmployee.email.data
row = Employee.query.filter_by(id = current_user.id).first()
row.fname = fn
row.lname = ln
row.phone_number = pn
row.email = em
db.session.commit()
flash('Your information has just been updated!')
return redirect(url_for('chooseToDo'))
firstname = current_user.fname
lastname = current_user.lname
email = current_user.email
phonenumber = current_user.phone_number
return render_template("account.html", title=title, formEmployee=formEmployee, formLogout=formLogout,
firstname=firstname, lastname=lastname, email=email, phonenumber=phonenumber)
def reset():
"""Allows the user to reset their password. Once when they want to reset their password
and once when they are forced to"""
if current_user.is_authenticated:
title = 'First Login password change'
email = current_user.email
else:
email = ''
title = 'Reset Your Password'
resetform = ResetPasswordForm()
formLogout = LogoutForm()
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
if resetform.submit.data and resetform.is_submitted:
if (current_user.is_authenticated):
current_user.setfirstlogin(False)
current_user.setQuestion(resetform.question1.data,
resetform.question2.data)
current_user.setAnswer(resetform.answer1.data,
resetform.answer2.data)
current_user.set_password(resetform.newPassword.data)
db.session.commit()
flash('New password has been set for this account.')
return redirect(url_for('login'))
else:
user = Employee.query.filter_by(email=resetform.email.data).first()
if user is None:
flash('Invalid Email. Try Again')
return redirect(url_for('reset'))
else:
answer1 = resetform.answer1.data
answer2 = resetform.answer2.data
question1 = resetform.question1.data
question2 = resetform.question2.data
user1 = Employee.query.filter_by(answer1=answer1,
answer2=answer2,
question1=question1,
question2=question2).first()
if user1 is None:
flash('Invalid answers or wrong choice of questions')
return redirect(url_for('reset'))
else:
user1.set_password(resetform.newPassword.data)
user1.setfirstlogin(False)
db.session.commit()
flash('New password has been set successfully!')
return redirect(url_for('login'))
return redirect(url_for('login'))
#resetform.question1.choices = [(Employee.id) for question1 in question1.query.filter_by(question1='Whichcity').all()]
return render_template('reset.html',
title=title,
resetform=resetform,
formLogout=formLogout,
email=email)
def about():
if (current_user.is_authenticated):
title = "About Shifter By User"
else:
title = "About Shifter"
formLogout = LogoutForm()
if formLogout.Logout.data and formLogout.is_submitted():
flash('Logged out')
return redirect(url_for('logout'))
return render_template("About.html", title=title, formLogout=formLogout)
def managerview():
"""Lets the manager setup the schedule and add Shifts to employees"""
title = "Employee Schedules"
formLogout = LogoutForm()
tableForm = scheduleTableForm()
scheduleform = managerhomepageForm()
datechosen = date.today()
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
if tableForm.goTo.data and request.method == 'POST':
datechosen = request.form['datebox']
if scheduleform.Submit.data and scheduleform.is_submitted():
dateToWork = datetime.date.fromisoformat(request.form['startdate'])
st = datetime.time.fromisoformat(scheduleform.starttime.data)
et = datetime.time.fromisoformat(scheduleform.endtime.data)
empId = int(scheduleform.employees.data)
s = Schedule( thedates = dateToWork, starttime = st, endtime=et, emp_id = empId, org_id = current_user.organization_id)
db.session.add(s)
db.session.commit()
datechosen = request.form['startdate']
if current_user.manager==False:
return redirect(url_for('emphomepage'))
todaysdate = date.today()
tabledict = {}
schedules = Schedule.query.filter_by(org_id=current_user.organization_id).filter_by(thedates = datechosen).order_by(Schedule.starttime).all()
employees = Employee.query.filter_by(organization_id=current_user.organization_id).all()
employee_list = []
for employee in employees:
for schedule in schedules:
if schedule.emp_id == employee.id:
tabledict[employee.fname + ' ' + employee.lname] = [schedule.starttime.strftime(format='%H:%M'), schedule.endtime.strftime(format='%H:%M')]
employee_list.append((employee.id, employee.fname +' '+ employee.lname))
scheduleform.employees.choices = employee_list
#ts = time(12,30)
#te = time(16,30)
#d = date(2019,12,2)
#S = Schedule(id = 200,thedates = d, starttime = ts, endtime = te, emp_id = 2, org_id = 1 )
#db.session.add(S)
#db.session.commit()
return render_template("managerviewsch.html",
title=title,
formLogout=formLogout,
tableForm = tableForm,
scheduleform = scheduleform,
todaysdate = todaysdate,
datechosen = datechosen,
tabledict = tabledict)
def about():
"""
Tells you about Shifter Team
No parameters taken here just a simple CSS page
"""
if (current_user.is_authenticated):
title = "About Shifter By User"
else:
title = "About Shifter"
formLogout = LogoutForm()
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
return render_template("About.html", title=title, formLogout=formLogout)
def contact():
"""
Allow the customers to contact our team for support and help
Here is the contact us function
This uses the parameters in the class app.forms.ContacttForm
"""
if (current_user.is_authenticated):
title = "Contact Us By User"
else:
title = "Contact Us"
formLogout = LogoutForm()
formContact = ContactForm()
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
"""
This first if statement takes in the fromContact all the information filled
out by the user and organizes it in the message show below and send the the information
to the a support email.
"""
if request.method == 'POST':
if formContact.validate() == False:
flash('All fields are required.')
return render_template('Contact.html',
formContact=formContact,
title=title)
else:
msg = Message(formContact.subject.data,
sender='*****@*****.**',
recipients=['*****@*****.**'])
msg.body = f"""
From: {formContact.name.data}
Email: {formContact.email.data}
Phone Number: {formContact.phone_number.data}
Subject: {formContact.subject.data}
Message: {formContact.message.data}
"""
mail.send(msg)
return render_template('Contact.html', title=title, success=True)
elif request.method == 'GET':
return render_template("Contact.html",
title=title,
formContact=formContact,
formLogout=formLogout)
"""
def emphomepage():
"""Employee Home Page. Displays the Schedule of logged in employee"""
if current_user.firsttimelogin == True:
flash('First you need to update your password.')
return redirect(url_for('reset'))
formLogout = LogoutForm()
formchangeweek = ChangeWeekForm()
title = current_user.fname + ' ' + current_user.lname + ' Homepage'
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
startdate = date.today().replace(day=1)
monthlength = monthrange(startdate.year, startdate.month)
enddate = startdate + timedelta(days=monthlength[1] - 1)
if formchangeweek.nextMonth.data and formchangeweek.is_submitted:
startdate += timedelta(days=monthlength[1])
monthlength = monthrange(startdate.year, startdate.month)
enddate = startdate + timedelta(days=monthlength[1] - 1)
if formchangeweek.previous.data and formchangeweek.is_submitted:
startdate -= timedelta(days=monthlength[1] - 1)
monthlength = monthrange(startdate.year, startdate.month)
enddate = startdate + timedelta(days=monthlength[1] - 1)
array = getWeeklySchedule(startdate, enddate)
dates = monthlength[1] + 1
schedulestarts = {}
schedulesends = {}
for i in range(1, monthlength[1] + 1):
schedulestarts[i] = []
schedulesends[i] = []
for j in range(1, monthlength[1] + 1):
for i in range(0, len(array)):
if j == array[i].thedates.day:
schedulestarts[j].append(
(array[i].starttime).strftime('%H:%M '))
schedulesends[j].append((array[i].endtime).strftime('%H:%M'))
return render_template('emphomepage.html',
title=title,
formLogout=formLogout,
formchangeweek=formchangeweek,
startdate=startdate,
enddate=enddate,
dates=dates,
starttimes=schedulestarts,
endtimes=schedulesends,
len=dates)
def chooseToDo():
title = "ChooseToDo"
formLogout = LogoutForm()
formEditView = EditViewForm()
if formLogout.Logout.data and formLogout.is_submitted():
flash('Logged out')
return redirect(url_for('logout'))
elif formEditView.AddEmpl.data and formEditView.is_submitted():
return redirect(url_for('addemployee'))
return render_template("choose.html",
title=title,
formLogout=formLogout,
formEditView=formEditView)
def addemployee():
"""Adds a new Employee to the data base under the same organization id as the logged in user who is adding the
new user.
Also allows the new user to be selected as manager or employee"""
title = "Add employee to Shifter"
formEmployee = EmployeeForm()
formLogout = LogoutForm()
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
if (formEmployee.validate_on_submit()):
user = Employee.query.filter_by(email=formEmployee.email.data).first()
user1 = Employee.query.filter_by(
phone_number=formEmployee.phone_number.data).first()
if user is not None:
flash(
'There is already an account registered under this email address.'
)
return redirect(url_for('addemployee'))
elif user1 is not None:
flash(
'There is already an account registered under this phone number.'
)
return redirect(url_for('addemployee'))
else:
firstn = formEmployee.first_name.data
employee = Employee(fname=formEmployee.first_name.data,
lname=formEmployee.last_name.data,
email=formEmployee.email.data,
phone_number=formEmployee.phone_number.data,
organization_id=current_user.organization_id,
firsttimelogin=True)
employee.setManager(formEmployee.manager.data)
employee.set_password(formEmployee.phone_number.data)
db.session.add(employee)
db.session.commit()
flash(firstn + ' has been added to the company')
return redirect(url_for('login'))
return render_template("addemployee.html",
title=title,
formEmployee=formEmployee,
formLogout=formLogout)
def chooseToDo():
"""Provides options for the manager who is logged in."""
if current_user.firsttimelogin == True:
flash('First you need to update your password.')
return redirect(url_for('reset'))
title = "ChooseToDo"
formLogout = LogoutForm()
formEditView = EditViewForm()
if current_user.manager==False:
return redirect(url_for('emphomepage'))
if formLogout.Logout.data and formLogout.is_submitted():
return redirect(url_for('logout'))
elif formEditView.AddEmpl.data and formEditView.is_submitted():
return redirect(url_for('addemployee'))
elif formEditView.View.data and formEditView.is_submitted:
return redirect(url_for('managerview'))
elif formEditView.ViewOwn.data and formEditView.is_submitted:
return redirect(url_for('emphomepage'))
return render_template("choose.html", title = title, formLogout = formLogout, formEditView = formEditView)
def addemployee():
title = "Add employee to Shifter"
formEmployee = EmployeeForm()
formLogout = LogoutForm()
if formLogout.Logout.data and formLogout.is_submitted():
flash('Logged out')
return redirect(url_for('logout'))
if (formEmployee.validate_on_submit()):
user = Employee.query.filter_by(email=formEmployee.email.data).first()
user1 = Employee.query.filter_by(
phone_number=formEmployee.phone_number.data).first()
if user is not None:
flash(
'There is already an account registered under this email address.'
)
return redirect(url_for('addemployee'))
elif user1 is not None:
flash(
'There is already an account registered under this phone number.'
)
return redirect(url_for('addemployee'))
else:
employee = Employee(fname=formEmployee.first_name.data,
lname=formEmployee.last_name.data,
email=formEmployee.email.data,
phone_number=formEmployee.phone_number.data,
organization_id=current_user.organization_id,
firsttimelogin=True)
employee.setManager(formEmployee.manager.data)
employee.set_password(formEmployee.phone_number.data)
db.session.add(employee)
db.session.commit()
return redirect(url_for('login'))
return render_template("addemployee.html",
title=title,
formEmployee=formEmployee,
formLogout=formLogout)
def contact():
if (current_user.is_authenticated):
title = "Contact Us By User"
else:
title = "Contact Us"
formLogout = LogoutForm()
formContact = ContactForm()
if formLogout.Logout.data and formLogout.is_submitted():
flash('Logged out')
return redirect(url_for('logout'))
if request.method == 'POST':
if formContact.validate() == False:
flash('All fields are required.')
return render_template('Contact.html',
formContact=formContact,
title=title)
else:
msg = Message(formContact.subject.data,
sender='*****@*****.**',
recipients=['*****@*****.**'])
msg.body = f"""
From: {formContact.name.data}
Email: {formContact.email.data}
Phone Number: {formContact.phone_number.data}
Subject: {formContact.subject.data}
Message: {formContact.message.data}
"""
mail.send(msg)
return render_template('Contact.html', title=title, success=True)
elif request.method == 'GET':
return render_template("Contact.html",
title=title,
formContact=formContact,
formLogout=formLogout)