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 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 is_exsist(user_id):
"""
Функция для провекри существования каких-либо сущностей
(пользователей и т.д)
:param kwargs:
:return:
"""
if session.query(Users).filter(Users.id == int(user_id)).first():
return True
else:
return False
def get_random_task():
tasks = session.query(Users_tasks).filter(
Users_tasks.is_checked == False).all()
if tasks:
# length = tasks.count()
if len(tasks) > 1:
task = tasks[random.randint(0, len(tasks))]
return task.id, task.file_id
else:
return tasks[0].id, tasks[0].file_id
else:
return None, None
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 get_user_fio(user_id):
if is_exsist(user_id):
user = session.query(Users).filter(Users.id == int(user_id)).first()
return user.first_name, user.second_name
def get_all_user_task(user_id):
users_tasks = session.query(Users_tasks).filter(
Users_tasks.user_id == user_id).all()
return users_tasks
def get_all_users():
users = session.query(Users).all()
return users
def get_user_task(user_id):
state = session.query(User_State).filter(User_State.id == user_id).first()
if state:
return state.task_id
else:
return 'none'
def get_user_id_by_name(first_name, second_name):
user = session.query(Users).filter(Users.first_name == first_name).filter(
Users.second_name == second_name).first()
if user:
return user.id
def get_mark(task_id):
mark = session.query(Marks).filter(Marks.task_id == task_id).first()
if mark:
return mark
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_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()