98 lines
2.5 KiB
Python
98 lines
2.5 KiB
Python
"""
|
|
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 |