def add_homework(id):
if id != 0:
db = db_session.create_session()
task = db.query(Homework).filter(
Homework.homework_id == int(id)).first()
if not task or task.user != current_user:
db.close()
return redirect('/homework/0')
all_data = {
'task': task.task,
'lesson': task.lesson,
'day_of_week': task.day_of_week,
'num_of_week': task.num_of_week,
'ready': task.ready
}
homework_form = HomeworkForm(data=all_data)
if homework_form.validate_on_submit():
task.task = homework_form.task.data
task.lesson = homework_form.lesson.data
task.day_of_week = homework_form.day_of_week.data
task.num_of_week = homework_form.num_of_week.data
task.ready = homework_form.ready.data
db.commit()
db.close()
return redirect('/homework/0')
db.close()
return render_template('add_homework.html',
form=homework_form,
deletion=True,
id=id)
else:
homework_form = HomeworkForm()
if homework_form.validate_on_submit():
db = db_session.create_session()
homework = Homework()
homework.task = homework_form.task.data
homework.lesson = homework_form.lesson.data
homework.day_of_week = homework_form.day_of_week.data
homework.num_of_week = homework_form.num_of_week.data
homework.ready = homework_form.ready.data
homework.file = 'nothing'
current_user.homework.append(homework)
db.merge(current_user)
db.commit()
db.close()
return redirect('/homework/0')
return render_template('add_homework.html',
form=homework_form,
deletion=False)
def add_timetable(id):
if id != 0:
db = db_session.create_session()
lesson = db.query(Timetable).filter(
Timetable.lesson_id == int(id)).first()
if not lesson or lesson.user != current_user:
db.close()
return redirect('/')
all_data = {
'lesson': lesson.lesson,
'room': lesson.room,
'day_of_week': lesson.day_of_week,
'num_of_week': lesson.num_of_week,
'time': lesson.time
}
timetable_form = TimetableForm(data=all_data)
if timetable_form.validate_on_submit():
lesson.lesson = timetable_form.lesson.data
lesson.room = timetable_form.room.data
lesson.day_of_week = timetable_form.day_of_week.data
lesson.num_of_week = timetable_form.num_of_week.data
lesson.time = timetable_form.time.data
db.commit()
db.close()
return redirect('/')
db.close()
return render_template('add_timetable.html',
form=timetable_form,
deletion=True,
id=id)
else:
timetable_form = TimetableForm()
if timetable_form.validate_on_submit():
db = db_session.create_session()
timetable = Timetable()
timetable.lesson = timetable_form.lesson.data
timetable.room = timetable_form.room.data
timetable.day_of_week = timetable_form.day_of_week.data
timetable.num_of_week = timetable_form.num_of_week.data
timetable.time = timetable_form.time.data
current_user.timetable.append(timetable)
db.merge(current_user)
db.commit()
db.close()
return redirect('/')
return render_template('add_timetable.html',
form=timetable_form,
deletion=False)
def change():
all_data = {
'login': current_user.login,
'name': current_user.name,
'email': current_user.email
}
change_form = ChangeForm(data=all_data)
if change_form.validate_on_submit():
db = db_session.create_session()
password = change_form.password.data
user_now = db.query(User).filter(User.id == current_user.id).first()
if user_now.check_password(password):
user_now.login = change_form.login.data
user_now.name = change_form.name.data
emails = db.query(User).filter(
User.email == change_form.email.data).all()
if len(emails) >= 2 and emails[0] != user_now.email:
db.close()
return render_template(
'change.html',
form=change_form,
message="Такая почта уже используется в другом аккаунте")
else:
user_now.email = change_form.email.data
db.commit()
db.close()
return redirect('/user_info')
else:
return render_template('change.html',
form=change_form,
message="Неправильный пароль")
return render_template('change.html', form=change_form)
def delete_homework(id):
db = db_session.create_session()
task = db.query(Homework).filter(Homework.user == current_user,
Homework.homework_id == id).first()
if task and task.user == current_user:
db.delete(task)
db.commit()
db.close()
return redirect('/homework/0')
def delete_timetable(id):
db = db_session.create_session()
lesson = db.query(Timetable).filter(Timetable.user == current_user,
Timetable.lesson_id == id).first()
if lesson and lesson.user == current_user:
db.delete(lesson)
db.commit()
db.close()
return redirect('/')
def reqister():
if current_user.is_authenticated:
return redirect('/')
form = RegisterForm()
if form.password.data and form.login.data and form.name.data and form.email.data:
if form.validate_on_submit():
if len(form.password.data) < 6:
return render_template(
'register.html',
form=form,
message="Пароль должен содержать не меньше 6 символов")
if form.password.data.isdigit() or form.password.data.isalpha():
return render_template(
'register.html',
form=form,
message="Пароль должен состоять не только из букв или цифр"
)
if not form.login.data or not form.name.data or not form.email.data:
return render_template(
'register.html',
form=form,
message='Проверьте правильность заполнения полей')
db = db_session.create_session()
if db.query(User).filter(User.email == form.email.data).first() or \
db.query(User).filter(User.login == form.login.data).first():
return render_template(
'register.html',
form=form,
message="Такой пользователь уже существует")
user = User(login=form.login.data,
name=form.name.data,
email=form.email.data,
theme='1')
user.set_password(form.password.data)
db.add(user)
db.commit()
login_user(user, remember=True)
db.close()
return redirect('/user_info')
return render_template('register.html', form=form)
def login():
if current_user.is_authenticated:
return redirect('/')
login_form = LoginForm()
if login_form.validate_on_submit():
db = db_session.create_session()
user = db.query(User).filter(
User.login == login_form.login.data).first()
if not user:
db.close()
return render_template(
'login.html',
form1=login_form,
message1="Такого пользователя не существует")
if user.check_password(login_form.password.data):
login_user(user, remember=True)
db.close()
return redirect('/user_info')
else:
db.close()
return render_template('login.html',
form1=login_form,
message1="Неправильный пароль")
return render_template('login.html', form1=login_form)
def index_start(num_of_week):
if current_user.is_anonymous:
return render_template('main.html', len_task=0)
if num_of_week != '0' and num_of_week != '1' and num_of_week != '2' and num_of_week != '-1' and num_of_week != '-2':
return redirect('/timetable/0')
db = db_session.create_session()
lessons = db.query(Timetable).filter(Timetable.id == current_user.id).all()
lessons1 = []
lessons2 = []
lessons3 = []
lessons4 = []
lessons5 = []
lessons6 = []
for lesson in lessons:
if lesson.day_of_week == '1' and lesson.num_of_week == num_of_week:
lessons1.append(lesson)
elif lesson.day_of_week == '2' and lesson.num_of_week == num_of_week:
lessons2.append(lesson)
elif lesson.day_of_week == '3' and lesson.num_of_week == num_of_week:
lessons3.append(lesson)
elif lesson.day_of_week == '4' and lesson.num_of_week == num_of_week:
lessons4.append(lesson)
elif lesson.day_of_week == '5' and lesson.num_of_week == num_of_week:
lessons5.append(lesson)
elif lesson.day_of_week == '6' and lesson.num_of_week == num_of_week:
lessons6.append(lesson)
next_num = str(int(num_of_week) + 1)
previous_num = str(int(num_of_week) - 1)
if num_of_week == '0':
message_of_week = 'На этой неделе'
elif num_of_week == '-1':
message_of_week = 'На предыдушей неделе'
elif num_of_week == '-2':
message_of_week = 'На позапрошлой неделе'
elif num_of_week == '1':
message_of_week = 'На следующей неделе'
elif num_of_week == '2':
message_of_week = 'На неделе после следующей'
if lessons:
db.close()
return render_template('main.html',
lessons=lessons,
len_lesson=len(lessons),
lessons1=lessons1,
lessons2=lessons2,
lessons3=lessons3,
lessons4=lessons4,
lessons5=lessons5,
lessons6=lessons6,
len_lesson1=len(lessons1),
len_lesson2=len(lessons2),
len_lesson3=len(lessons3),
len_lesson4=len(lessons4),
len_lesson5=len(lessons5),
len_lesson6=len(lessons6),
num_of_week=num_of_week,
next_num=next_num,
previous_num=previous_num,
message_of_week=message_of_week)
else:
db.close()
return render_template('main.html',
lessons=lessons,
len_task=0,
num_of_week=num_of_week,
next_num=next_num,
previous_num=previous_num,
message_of_week=message_of_week)
def load_user(user_id):
db = db_session.create_session()
user = db.query(User).get(user_id)
return user
def homework(num_of_week):
if current_user.is_anonymous:
return render_template('homework.html', len_task=0)
if num_of_week != '0' and num_of_week != '1' and num_of_week != '2' and num_of_week != '-1' and num_of_week != '-2':
return redirect('/homework/0')
db = db_session.create_session()
tasks = db.query(Homework).filter(Homework.id == current_user.id).all()
tasks1 = []
tasks2 = []
tasks3 = []
tasks4 = []
tasks5 = []
tasks6 = []
for task in tasks:
if task.day_of_week == '1' and task.num_of_week == num_of_week:
tasks1.append(task)
elif task.day_of_week == '2' and task.num_of_week == num_of_week:
tasks2.append(task)
elif task.day_of_week == '3' and task.num_of_week == num_of_week:
tasks3.append(task)
elif task.day_of_week == '4' and task.num_of_week == num_of_week:
tasks4.append(task)
elif task.day_of_week == '5' and task.num_of_week == num_of_week:
tasks5.append(task)
elif task.day_of_week == '6' and task.num_of_week == num_of_week:
tasks6.append(task)
next_num = str(int(num_of_week) + 1)
previous_num = str(int(num_of_week) - 1)
if num_of_week == '0':
message_of_week = 'На этой неделе'
elif num_of_week == '-1':
message_of_week = 'На предыдушей неделе'
elif num_of_week == '-2':
message_of_week = 'На позапрошлой неделе'
elif num_of_week == '1':
message_of_week = 'На следующей неделе'
elif num_of_week == '2':
message_of_week = 'На неделе после следующей'
if tasks:
db.close()
return render_template('homework.html',
tasks=tasks,
len_task=len(tasks),
tasks1=tasks1,
tasks2=tasks2,
tasks3=tasks3,
tasks4=tasks4,
tasks5=tasks5,
tasks6=tasks6,
len_task1=len(tasks1),
len_task2=len(tasks2),
len_task3=len(tasks3),
len_task4=len(tasks4),
len_task5=len(tasks5),
len_task6=len(tasks6),
num_of_week=num_of_week,
next_num=next_num,
previous_num=previous_num,
message_of_week=message_of_week)
else:
db.close()
return render_template('homework.html',
tasks=tasks,
len_task=0,
num_of_week=num_of_week,
next_num=next_num,
previous_num=previous_num,
message_of_week=message_of_week)