def insert_file(user_id, file_id, file_name):
if not session.query(Users_tasks).filter(
Users_tasks.file_name == file_name).first():
user_task = Users_tasks(user_id, file_id, file_name)
user_task.is_checked = False
session.add(user_task)
session.commit()
else:
print('Error insert_file')
def set_user_state(user_id, state, task_id=None):
if session.query(User_State).filter(User_State.id == int(user_id)).first():
user_state = session.query(User_State).filter(
User_State.id == int(user_id)).first()
user_state.state = state
else:
user_state = User_State(user_id, state)
if task_id: user_state.task_id = task_id
session.add(user_state)
session.commit()
def add_user(user_id, first_name, second_name):
user = Users(user_id, first_name, second_name)
session.add(user)
session.commit()
def delete_user_state(user_id):
if session.query(User_State).filter(User_State.id == int(user_id)).first():
user_state = session.query(User_State).filter(
User_State.id == int(user_id)).first()
session.delete(user_state)
session.commit()
def set_task_comment(task_id, comment):
mark = session.query(Marks).filter(Marks.task_id == task_id).first()
mark.comment = comment
session.add(mark)
session.commit()
def set_task_mark(task_id, mark):
mark = Marks(task_id, mark)
session.add(mark)
session.commit()
def set_task_check(task_id, status):
task = session.query(Users_tasks).filter(Users_tasks.id == task_id).first()
if task:
task.is_checked = status
session.add(task)
session.commit()