def myshifts(request): username = str(request.user) air_crew = AirCrew() my_shifts = air_crew.ShowMyShifts(username) future_shifts = air_crew.ShowFutureShifts(username) args = {'my_shifts' : my_shifts, 'future_shifts': future_shifts} return render(request, 'base/myshifts1.html', args)
def Profile(request): if request.user.is_staff: shift_manager = ShiftManager() usernamequery = (str(request.user), ) name = shift_manager.getname(usernamequery) totalgoal = shift_manager.totalgoal() totaldid = shift_manager.totaldid() remain = float(totalgoal) - float(totaldid) args = { 'name': name, 'totalgoal': totalgoal, 'totaldid': totaldid, 'remain': remain } return render(request, 'base/managerpage.html', args) else: username = str(request.user) usernamequery = (str(request.user), ) mydb = mysql.connector.connect(host="localhost", user="******", password='******', database='flight_manager', auth_plugin='mysql_native_password') mycursor = mydb.cursor() query = """ SELECT goal FROM flight_manager.air_crew WHERE personal_number = %s """ mycursor.execute(query, usernamequery) results = mycursor.fetchall() mydb.commit() for row in results: shift = str(row[0]) new_shift = Shift() all_shifts = new_shift.ShowAllShifts() if len(shift) == 0: shift = '0' air_crew = AirCrew() future_shifts = air_crew.ShowFutureShifts(username) name = air_crew.getname(usernamequery) howmanydid = air_crew.howmanydid(username) howmanytot = air_crew.howmanytot(usernamequery) name = air_crew.getname(usernamequery) try: left = int(shift) - int(howmanydid[0]) except: left = str('0') finally: args = { 'user': name, 'shifts': shift, 'all_shifts': all_shifts, 'howmanydid': howmanydid[0], 'left': left, 'shift': shift } return render(request, 'base/profile.html', args)