Attendance/app.py

332 lines
13 KiB
Python

from flask import *
import pyrebase
from datetime import datetime
import time
import pytz
from sendgrid import SendGridAPIClient
from sendgrid.helpers.mail import Mail
import csv
import os
from dotenv import load_dotenv
from pprint import pprint
load_dotenv()
app = Flask(__name__)
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY')
config = {
"apiKey": os.environ.get('apiKey'),
"authDomain": os.environ.get('authDomain'),
"databaseURL": os.environ.get('databaseURL'),
"storageBucket": os.environ.get('storageBucket'),
"serviceAccount": os.environ.get('serviceAccount'),
"messagingSenderId": os.environ.get('messagingSenderId'),
"appId": os.environ.get('appId'),
"measurementId": os.environ.get('measurementId'),
}
firebase = pyrebase.initialize_app(config)
db = firebase.database()
auth = firebase.auth()
tz = pytz.timezone('Asia/Taipei')
def check_login_status():
return ('is_logged_in' not in session or
session['is_logged_in'] == False or
(datetime.now(tz) - session['loginTime']).total_seconds() > 3600)
def manageProcess(fCommand, fData):
# this is to fix a bug where pyrebase doesnt load the first request
db.child("Users").child(
session['uid']).child("permission").get().val()
# end bug fix
pl = db.child("Users").child(
session['uid']).child("permission").get().val()
print(pl)
print(db.child("Users").child(
'DRZqqSSpg3OkPSCuWkv417dv0vh1').child("permission").get().val())
print(fCommand, fData, session['uid'], pl)
if pl == 'admin':
return pl
elif pl == 'group':
classes = db.child("Users").child(
session['uid']).child("class").get().val()
cclass = {}
for i in classes:
cclass = {
"name": db.child("Classes").child(i).child(
"Class").child(classes[i]).child("name").get().val(),
"category": i,
"class_id": classes[i]
}
print("got class")
students = db.child("Classes").child(cclass['category']).child(
"Class").child(cclass['class_id']).child("Students").get().val()
print(students['9']['11'])
all_stud_list = {}
for grade in students:
print(grade)
all_stud_list[grade] = {}
print(type(students[grade]))
for homeroom in students[grade]:
print(homeroom)
roomData = db.child("Homerooms").child(
grade).child(homeroom).get().val()
all_stud_list[grade][homeroom] = {}
if type(students[grade][homeroom]) == list:
i = 0
for student in students[grade][homeroom]:
if student == 0:
all_stud_list[grade][homeroom][i] = {
"name": roomData[str(i)]["name"],
"eng_name": roomData[str(i)]["eng_name"]
}
i += 1
else:
for student in students[grade][homeroom]:
print(student)
all_stud_list[grade][homeroom][student] = {
"name": roomData[student]["name"],
"eng_name": roomData[student]["eng_name"]
}
print("got students")
dates = db.child("Classes").child(
cclass['category']).child("Dates").get().val()
status = 0
attendance = {}
if fCommand == 'date':
currDate = fData
if cclass['class_id'] in dates[currDate]:
status = 1
for grade in dates[currDate]['Absent']:
attendance[grade] = {}
for homeroom in dates[currDate]['Absent'][grade]:
attendance[grade][homeroom] = {}
for student in dates[currDate]['Absent'][grade][homeroom]:
attendance[grade][homeroom][student] = 0
else:
for i in dates:
if i >= datetime.now(tz).strftime("%Y-%m-%d"):
currDate = i
if cclass['class_id'] in dates[currDate]:
status = 1
for grade in dates[currDate]['Absent']:
attendance[grade] = {}
for homeroom in dates[currDate]['Absent'][grade]:
attendance[grade][homeroom] = {}
for student in dates[currDate]['Absent'][grade][homeroom]:
attendance[grade][homeroom][student] = 0
break
dates[i].pop('placeholder')
print("got dates")
return render_template('group_teach.html', cclass=cclass, all_stud_list=all_stud_list, dates=dates, currDate=currDate, status=status, attendance=attendance)
elif pl == 'homeroom':
homeroom = db.child("Users").child(
session['uid']).child("homeroom").get().val()
homeroomCode = homeroom.split('^')
homeroom = db.child("Homerooms").child(
homeroomCode[0]).child(homeroomCode[1]).get().val()
categories = homeroom['Categories'].split('^')
currCategory = categories[0]
currDate = ""
gpClasses = db.child("Classes").child(currCategory).get().val()
dates = gpClasses['Dates']
confirmedClasses = []
status = 0
if fCommand == 'date':
currDate = fData
tmp1 = 0
tmp2 = 0
for k in gpClasses['Class']:
if k in dates[currDate]:
confirmedClasses.append(k)
tmp2 += 1
tmp1 += 1
if tmp1 == tmp2:
status = 1
else:
for i in dates:
if i >= datetime.now(tz).strftime("%Y-%m-%d"):
currDate = i
tmp1 = 0
tmp2 = 0
for k in gpClasses['Class']:
if k in dates[currDate]:
confirmedClasses.append(k)
tmp2 += 1
tmp1 += 1
if tmp1 == tmp2:
status = 1
break
print("got dates")
db.child("Classes").child(currCategory).child("Dates")
homeroom.pop('Categories')
all_stud_list = {}
for i in homeroom:
all_stud_list[i] = {}
all_stud_list[i]['name'] = homeroom[i]['name']
all_stud_list[i]['eng_name'] = homeroom[i]['eng_name']
all_stud_list[i]['gpClass'] = homeroom[i]['Classes'][currCategory]
if all_stud_list[i]['gpClass'] in confirmedClasses:
if (homeroomCode[0] in gpClasses['Dates'][currDate]['Absent'] and
homeroomCode[1] in gpClasses['Dates'][currDate]['Absent'][homeroomCode[0]] and
i in gpClasses['Dates'][currDate]['Absent'][homeroomCode[0]][homeroomCode[1]]):
# confirmed by teacher and absent
all_stud_list[i]['status'] = 2
else:
# confirmed by teacher and not absent
all_stud_list[i]['status'] = 1
else:
all_stud_list[i]['status'] = 0 # not yet confirmed by teacher
return render_template('homeroom.html', all_stud_list=all_stud_list, currDate=currDate, dates=dates, gpClasses=gpClasses, confirmedClasses=confirmedClasses,
currCategory=currCategory, categories=categories, homeroomCode=homeroomCode, status=status)
else:
return redirect('/')
@ app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'GET':
if check_login_status():
return render_template('login.html', error=False)
return redirect('/manage')
elif request.method == 'POST':
if check_login_status():
try:
user = auth.sign_in_with_email_and_password(
request.form['username'] + "@group-attendence.fhjh.tp.edu.tw", request.form['password'])
session['is_logged_in'] = True
session['email'] = user['email']
session['uid'] = user['localId']
session['token'] = user['idToken']
session['refreshToken'] = user['refreshToken']
session['loginTime'] = datetime.now(tz)
return redirect('/manage')
except Exception as e:
return render_template('login.html', error=True)
else:
return redirect('/manage')
@app.route('/manage', methods=['GET'])
def manage():
return manageProcess("", "")
@app.route('/manage/date', methods=['POST'])
def manage_date():
return manageProcess("date", request.form['date'])
@app.route('/manage/group_teach_publish', methods=['POST'])
def group_teach_publish():
classes = db.child("Users").child(
session['uid']).child("class").get().val()
cclass = {}
for i in classes:
cclass = {
"name": db.child("Classes").child(i).child(
"Class").child(classes[i]).child("name").get().val(),
"category": i,
"class_id": classes[i]
}
print("got class")
cDate = ""
for key in request.form.keys():
print(type(key), key)
if key == 'date':
print('here')
cDate = request.form[key]
db.child("Classes").child(cclass['category']).child(
"Dates").child(request.form[key]).update({'confirmed': 0})
db.child("Classes").child(cclass['category']).child(
"Dates").child(request.form[key]).update({cclass['class_id']: request.form['signatureData']})
elif key == 'signatureData':
pass
else:
# spilt string
id = key.split('^')
print(id)
db.child("Classes").child(cclass['category']).child("Dates").child(
cDate).child("Absent").child(id[0]).child(id[1]).update({id[2]: 1})
return "Success!"
@ app.route('/upload/homeroom', methods=['GET', 'POST'])
def upload_homeroom():
if request.method == 'GET':
return render_template('uploadcsv.html', title="Homeroom List", url="/upload/homeroom")
elif request.method == 'POST':
try:
# get csv
gradec = request.form['gradeCode']
classc = request.form['classcode']
csv_file = request.files['csv']
filepath = os.path.join('./temp', csv_file.filename)
csv_file.save(filepath)
with open(filepath) as file:
csv_dict = csv.DictReader(file)
for row in csv_dict:
db.child("Homerooms").child(gradec).child(
classc).child(row['number']).set(row)
# row['class'] row['number'] row['name'] row['eng_name']
os.remove(filepath)
except Exception as e:
os.remove(filepath)
return "Error. Please try again\n("+str(e)+")"
return "Successfully uploaded " + gradec + "-" + classc
@ app.route('/upload/stud_in_group', methods=['GET', 'POST'])
def upload_stud_in_group():
if request.method == 'GET':
return render_template('uploadcsv.html', title="Student in Group List", url="/upload/stud_in_group")
elif request.method == 'POST':
try:
gradec = request.form['gradeCode']
classc = request.form['classcode']
csv_file = request.files['csv']
filepath = os.path.join('./temp', csv_file.filename)
csv_file.save(filepath)
with open(filepath) as file:
csv_dict = csv.DictReader(file)
headers = csv_dict.fieldnames
headers = headers[1:]
for row in csv_dict:
for h in headers:
db.child("Homerooms").child(gradec).child(classc).child(
row['number']).child("Classes").child(h).set(row[h])
db.child("Classes").child(h).child("Class").child(row[h]).child(
"Students").child(gradec).child(classc).update({str(row['number']): 0})
os.remove(filepath)
except Exception as e:
os.remove(filepath)
return "Error. Please try again\n("+str(e)+")"
return "Successfully uploaded " + gradec + "-" + classc
# @ app.route('/upload/rm_all_data_of_class', methods=['GET', "POST"])
# def rm_all_data_of_class():
# if request.method == 'GET':
# return render_template('uploadcsv.html', title="Remove all data of class", url="/upload/rm_all_data_of_class")
# elif request.method == 'POST':
# try:
# classc = request.form['classcode']
# db.child("Homerooms").child(classc).remove()
# except Exception as e:
# return "Error. Please try again\n("+str(e)+")"
# return "Successfully removed " + classc
@app.route('/logout', methods=['GET'])
def logout():
session.clear()
return redirect('/')
if __name__ == '__main__':
app.run(debug=True)