-
Notifications
You must be signed in to change notification settings - Fork 0
/
webapp.py
211 lines (178 loc) · 8.46 KB
/
webapp.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
from flask import Flask, render_template, request, flash, redirect, url_for
from flask_login import LoginManager, login_required, login_user, logout_user, current_user
from flask_sqlalchemy import SQLAlchemy
import src.calendarHelper as calendarHelper
import src.tables as tables
import src.shiftHelper as shiftHelper
import datetime
LM = LoginManager()
db = SQLAlchemy()
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.database'
app.secret_key = 'a secret shhhh'
LM.init_app(app)
db.init_app(app)
@LM.user_loader
def load_user(id):
return tables.getUserByID(id)
@app.route('/')
def start():
try:
current_user.id
return redirect(url_for('home'))
except:
return redirect(url_for('login'))
@app.route('/login', methods=['GET', 'POST'])
def login():
error = None
if request.method == 'POST':
user = tables.checkLogin(request.form['username'], request.form['password'])
if user == None:
error = 'Invalid Credentials. Please try again.'
else:
login_user(user)
return redirect(url_for('home'))
return render_template("login.html", error=error)
@app.route('/home', methods=['GET', 'POST'])
@login_required
def home():
thisYear = datetime.datetime.today().year
thisMonth = datetime.datetime.today().month
shifts = shiftHelper.organizeShifts(tables.getUserShifts(current_user.id), thisYear, thisMonth)
if request.method == 'POST':
if "next" in request.form:
if datetime.datetime.today().month == 12:
shifts = shiftHelper.organizeShifts(tables.getUserShifts(current_user.id, year=thisYear+1, month=1), thisYear + 1, 1)
else:
shifts = shiftHelper.organizeShifts(tables.getUserShifts(current_user.id, year=thisYear, month=thisMonth+1), thisYear, thisMonth+1)
return render_template("Home.html", month=calendarHelper.getNextMonth(), user=current_user, shifts=shifts)
else:
return render_template("Home.html", month=calendarHelper.getCurMonth(), user=current_user, shifts=shifts)
return render_template("Home.html", month=calendarHelper.getCurMonth(), user=current_user, shifts=shifts)
@app.route('/home/<int:errorcode>', methods=['GET', 'POST'])
@login_required
def homeError(errorcode):
error = None
if errorcode == 401:
error = 'You are not authorized to access this page'
elif errorcode == 412:
error = 'You are trying to clock in too early'
elif errorcode == 413:
error = 'You are trying to clock out too late'
elif errorcode == 414:
error = 'You have no shift to clock in/clock out to'
thisYear = datetime.datetime.today().year
thisMonth = datetime.datetime.today().month
shifts = shiftHelper.organizeShifts(tables.getUserShifts(current_user.id), thisYear, thisMonth)
if request.method == 'POST':
if "next" in request.form:
if datetime.datetime.today().month == 12:
shifts = shiftHelper.organizeShifts(tables.getUserShifts(current_user.id, year=thisYear+1, month=1), thisYear + 1, 1)
else:
shifts = shiftHelper.organizeShifts(tables.getUserShifts(current_user.id, year=thisYear, month=thisMonth+1), thisYear, thisMonth+1)
return render_template("Home.html", month=calendarHelper.getNextMonth(), user=current_user, shifts=shifts)
else:
return render_template("Home.html", month=calendarHelper.getCurMonth(), user=current_user, shifts=shifts)
return render_template("Home.html", month=calendarHelper.getCurMonth(), user=current_user, shifts=shifts, error=error)
@app.route("/logout")
@login_required
def logout():
logout_user()
return redirect(url_for('login'))
@app.route("/request-time", methods=['GET', 'POST'])
@login_required
def request_time():
if request.method == 'POST':
return redirect(url_for('home'))
return render_template("request-time.html")
@app.route('/check-hours')
@login_required
def checkHours():
shifts = shiftHelper.removeShiftsNotInMonth(tables.getUserShifts(current_user.id))
month = calendarHelper.getMonthAndYear()
hours = shiftHelper.getTotalHours(shifts)
return render_template("check-hours.html", shifts=shifts, month=month, hours=hours)
@app.route('/add-employee', methods=['GET', 'POST'])
@login_required
def addEmployee():
if not current_user.manager:
return redirect(url_for('homeError', errorcode=401))
if request.method == 'POST':
manager = False
if request.form['manager'] == 'True':
manager = True
tables.addUser(request.form['username'], request.form['name'], request.form['password'], manager)
return redirect(url_for('allEmployees'))
return render_template('add-employee.html')
@app.route('/all-employees')
@login_required
def allEmployees():
if not current_user.manager:
return redirect(url_for('homeError', errorcode=401))
return render_template('all-employees.html', users=tables.getAllUsers())
@app.route('/scheduler', methods=['GET', 'POST'])
@login_required
def scheduler():
if not current_user.manager:
return redirect(url_for('home', errorcode=401))
if request.method == 'POST':
if "nextweek" in request.form:
week = calendarHelper.getNextWeek(shiftHelper.convertToDateTime(request.form['max']))
return render_template('scheduler.html', users=tables.getAllUsers(), week=week, shifts=shiftHelper.weekShiftList(tables.getWeekShifts(week['min']), week))
elif "lastweek" in request.form:
week = calendarHelper.getLastWeek(shiftHelper.convertToDateTime(request.form['min']))
return render_template('scheduler.html', users=tables.getAllUsers(), week=week, shifts=shiftHelper.weekShiftList(tables.getWeekShifts(week['min']), week))
elif "shiftID" in request.form:
tables.removeShift(request.form['shiftID'])
week = calendarHelper.getCurWeekFromDay(shiftHelper.convertToDateTime(request.form['min']))
return render_template('scheduler.html', users=tables.getAllUsers(), week=week, shifts=shiftHelper.weekShiftList(tables.getWeekShifts(week['min']), week))
else: #create shift
start = shiftHelper.convertToDateTime(request.form['start'])
end = shiftHelper.convertToDateTime(request.form['end'])
message = shiftHelper.isValid(start, end)
week = calendarHelper.getCurWeekFromDay(shiftHelper.convertToDateTime(request.form['start']))
if message != 'All Good!':
return render_template('scheduler.html', users=tables.getAllUsers(), week=week, shifts=shiftHelper.weekShiftList(tables.getWeekShifts(week['min']), week), error=message)
tables.addShift(request.form['user_id'], start, end)
return render_template('scheduler.html', users=tables.getAllUsers(), week=week, shifts=shiftHelper.weekShiftList(tables.getWeekShifts(week['min']), week))
week = calendarHelper.getCurWeek()
return render_template('scheduler.html', users=tables.getAllUsers(), week=week, shifts=shiftHelper.weekShiftList(tables.getWeekShifts(week['min']), week))
@app.route('/remove-employee')
@login_required
def removeEmployee():
if not current_user.manager:
return redirect(url_for('homeError', errorcode=401))
return render_template('remove-employee.html', users=tables.getAllUsers())
@app.route('/remove/<id>')
@login_required
def remove(id):
if not current_user.manager:
return redirect(url_for('homeError', errorcode=401))
tables.removeUser(id)
return redirect(url_for('removeEmployee'))
@app.route('/clock-in')
@login_required
def clockIn():
shift = tables.getTodayShift(current_user.id)
if shift is None:
return redirect(url_for('homeError', errorcode=414))
if shiftHelper.validClockIn(shift):
tables.clockIn(shift.id)
else:
return redirect(url_for('homeError', errorcode=414))
return redirect(url_for('home'))
@app.route('/clock-out')
@login_required
def clockOut():
shift = tables.getTodayShift(current_user.id)
if shift is None:
return redirect(url_for('homeError', errorcode=414))
print(shift)
if shiftHelper.canUserClockOut(shift):
tables.clockOut(shift.id)
else:
return redirect(url_for('homeError', errorcode=414))
return redirect(url_for('home'))
if __name__ == "__main__":
calendarHelper.getNextMonth()
app.run(host="127.0.0.1", port=8080, debug=True)