system-pri/backend/app/coordinator/routes/groups.py

121 lines
4.0 KiB
Python
Raw Normal View History

from flask import abort, current_app
from apiflask import APIBlueprint
from flask_sqlalchemy import get_debug_queries
2022-06-12 17:45:45 +02:00
from ...students.models import Group, Student
from ...project_supervisor.models import ProjectSupervisor
from ..schemas import GroupSchema, GroupEditSchema, GroupsPaginationSchema, \
GroupCreateSchema, MessageSchema, GroupQuerySchema
from ...dependencies import db
2022-06-12 17:45:45 +02:00
from ...base.utils import paginate_models
bp = APIBlueprint("groups", __name__, url_prefix="/groups")
@bp.route("/", methods=["GET"])
2022-06-12 17:45:45 +02:00
@bp.input(GroupQuerySchema, location='query')
@bp.output(GroupsPaginationSchema)
def list_groups(query: dict) -> dict:
search_name = query.get('name')
page = query.get('page')
per_page = query.get('per_page')
2022-06-12 17:45:45 +02:00
groups_query = Group.search_by_name(search_name)
data = paginate_models(page, groups_query, per_page)
2022-06-12 17:45:45 +02:00
return {
"groups": data['items'],
"max_pages": data['max_pages']
}
2022-06-12 17:45:45 +02:00
@bp.route("/", methods=["POST"])
@bp.input(GroupCreateSchema)
@bp.output(MessageSchema)
def create_group(data: dict) -> dict:
name = data['name']
students_indexes = data['students']
2022-06-12 17:45:45 +02:00
project_supervisor_id = data['project_supervisor_id']
project_supervisor = ProjectSupervisor.query.filter_by(id=project_supervisor_id).first()
limit_student_per_group = current_app.config.get('LIMIT_STUDENTS_PER_GROUP')
if project_supervisor is None:
abort(400, f"Project Supervisor with id {project_supervisor_id} doesnt exist")
elif limit_student_per_group is not None and limit_student_per_group < len(students_indexes):
abort(400, f"Too much students you want add to group, "
f"The group can have only {limit_student_per_group} students")
limit = db.session.query(ProjectSupervisor.limit_group - db.func.count(ProjectSupervisor.id)).join(Group).filter(
ProjectSupervisor.id == project_supervisor_id).group_by(ProjectSupervisor.id).scalar()
if limit is not None and limit <= 0:
abort(400, "Can't create new group, project supervisor achieved a limit of groups")
group = Group(name=name, project_supervisor_id=project_supervisor_id)
students_without_groups = db.session.query(Student).join(Group, isouter=True) \
.filter(Group.id.is_(None)).filter(Student.index.in_(students_indexes)).count()
2022-06-12 17:45:45 +02:00
if students_without_groups != len(students_indexes):
abort(400, "One or more students have already belonged to group!")
db.session.add(group)
db.session.commit()
students = db.session.query(Student).filter(Student.index.in_(students_indexes)).all()
project_supervisor.count_groups += 1
2022-06-12 17:45:45 +02:00
for student in students:
student.group_id = group.id
2022-06-12 17:45:45 +02:00
db.session.commit()
return {"message": "Project Supervisor was created!"}
@bp.route("/<int:id>/", methods=["GET"])
2022-06-12 17:45:45 +02:00
@bp.output(GroupSchema)
def detail_group(id: int) -> Group:
group = Group.query.filter_by(id=id).first()
if group is None:
abort(400, f"Group with id {id} doesn't exist!")
return group
@bp.route("/<int:id>/", methods=["DELETE"])
2022-06-12 17:45:45 +02:00
@bp.output(MessageSchema, status_code=202)
def delete_group(id: int) -> dict:
group = Group.query.filter_by(id=id).first()
if group is None:
abort(400, f"Group with id {id} doesn't exist!")
project_supervisor = ProjectSupervisor.query.filter_by(id=group.project_supervisor_id).first()
project_supervisor.count_groups -= 1
students = db.session.query(Student).filter_by(group_id=id).all()
for student in students:
student.group_id = None
2022-06-12 17:45:45 +02:00
db.session.delete(group)
db.session.commit()
return {"message": "Group was deleted!"}
@bp.route("/<int:id>", methods=["PUT"])
@bp.input(GroupEditSchema)
@bp.output(MessageSchema)
def edit_group(id: int, data: dict) -> dict:
if not data:
abort(400, 'You have passed empty data!')
2022-06-12 17:45:45 +02:00
group_query = Group.query.filter_by(id=id)
group = group_query.first()
if group is None:
abort(400, f"Group with id {id} doesn't exist!")
group_query.update(data)
2022-06-12 17:45:45 +02:00
db.session.commit()
return {"message": "Group was updated!"}