resident_residency_and_location_tracker/app/database.py

"""Database module"""

from app import session
from app.models import State, Region, Player, StateRegion


def get_state_regions(state_id):
    """Get regions from state"""
    state = session.query(State).get(state_id)
    regions = state.regions.filter(StateRegion.until_date_time == None).all()
    return regions

# def get_latest_professor(state_id, department_type):
#     """Get latest professor from database"""
#     department = get_department(state_id, department_type)
#     professor = department.department_stats.order_by(DepartmentStat.date_time.desc()).first()
#     return professor
#
# def get_player(player_id, player_name):
#     """Get player from database"""
#     player = session.query(Player).get(player_id)
#     if player is None:
#         player = Player()
#         player.id = player_id
#         player.name = player_name
#         session.add(player)
#         session.commit()
#     return player
#
# def get_department(state_id, department_type):
#     """Get department from database"""
#     department = session.query(Department).filter(
#         Department.state_id == state_id
#     ).filter(
#         Department.department_type == department_type
#     ).first()
#     if department is None:
#         department = Department()
#         department.state_id = state_id
#         department.department_type = department_type
#         session.add(department)
#         session.commit()
#     return department
#
# def save_professors(state_id, department_type, professors):
#     """Save professors to database"""
#     department = get_department(state_id, department_type)
#
#     for professor in professors:
#         player = get_player(professor['id'], professor['name'])
#         department_stat = DepartmentStat()
#         department_stat.department_id = department.id
#         department_stat.date_time = professor['date_time']
#         department_stat.points = professor['points']
#         department_stat.player_id = player.id
#         session.add(department_stat)
#     session.commit()