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()