api_v1.4/hub_api/city_commands.py

"""
HeatPump Service
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2023 Project Author Koa Wells kekoa.wells@concordia.ca
"""

import json
from flask import Response, request
from flask_restful import Resource
from imports.user_factory import UserFactory
from exports.user_factory import UserFactory as ExUserFactory
import os
from hub_logger import logger
from hub_api.helpers.session_helper import refresh_session
from hub_api.helpers.auth import generate_auth_token, role_required
from persistence.models import UserRoles

#Admin user commands
"""
SaveCity class
"""
class SaveCity(Resource):
    def __init__(self):
        print()

    @role_required([UserRoles.Admin.value])
    def put(self):
        session = refresh_session(request)
        if session is None:
            return Response(json.dumps({'error': 'invalid session'}), status=401)
        headers = session.headers


"""
UpdateCity class
"""
class UpdateCity(Resource):
    def __init__(self):
        print()

    @role_required([UserRoles.Admin.value])
    def put(self):
        session = refresh_session(request)
        if session is None:
            return Response(json.dumps({'error': 'invalid session'}), status=401)
        headers = session.headers
"""
DeleteCity class
"""
class DeleteCity(Resource):
    def __init__(self):
        print()

    @role_required([UserRoles.Admin.value])
    def delete(self):
        session = refresh_session(request)
        if session is None:
            return Response(json.dumps({'error': 'invalid session'}), status=401)
        headers = session.headers

#Standard user commands
"""
ListCities class
"""
class ListCities(Resource):
    def __init__(self):
        print()

    @role_required([UserRoles.Admin.value])
    def put(self):
        session = refresh_session(request)
        if session is None:
            return Response(json.dumps({'error': 'invalid session'}), status=401)
        headers = session.headers
"""
SearchCity class
"""
class SearchCity(Resource):
    def __init__(self):
        print()

    def get(self):
        session = refresh_session(request)
        if session is None:
            return Response(json.dumps({'error': 'invalid session'}), status=401)
        headers = session.headers
"""
UpdateCity class
"""
class UpdateCity(Resource):
    def __init__(self):
        print()

    def get(self):
        session = refresh_session(request)
        if session is None:
            return Response(json.dumps({'error': 'invalid session'}), status=401)
        headers = session.headers