mirror of
https://github.com/dancojocaru2000/foxbank.git
synced 2025-02-23 12:29:35 +02:00
78 lines
1.5 KiB
Python
78 lines
1.5 KiB
Python
from http import HTTPStatus as _HTTPStatus
|
|
from typing import Any
|
|
|
|
|
|
def _make_error(http_status, code: str, message: str | None = None):
|
|
try:
|
|
http_status = http_status[0]
|
|
except Exception:
|
|
pass
|
|
|
|
payload = {
|
|
'status': 'error',
|
|
'code': code,
|
|
}
|
|
|
|
if message is not None:
|
|
payload['message'] = message
|
|
|
|
return payload, http_status
|
|
|
|
|
|
# General
|
|
|
|
INVALID_REQUEST = _make_error(
|
|
_HTTPStatus.BAD_REQUEST,
|
|
'general/invalid_request',
|
|
)
|
|
|
|
NOT_FOUND = _make_error(
|
|
_HTTPStatus.NOT_FOUND,
|
|
'general/not_found',
|
|
)
|
|
|
|
# Login
|
|
|
|
INVALID_DETAILS = _make_error(
|
|
_HTTPStatus.UNAUTHORIZED,
|
|
'login/invalid_details',
|
|
)
|
|
|
|
NO_AUTHORIZATION = _make_error(
|
|
_HTTPStatus.UNAUTHORIZED,
|
|
'login/no_authorization',
|
|
)
|
|
|
|
INVALID_AUTHORIZATION = _make_error(
|
|
_HTTPStatus.UNAUTHORIZED,
|
|
'login/invalid_authorization',
|
|
)
|
|
|
|
UNAUTHORIZED = _make_error(
|
|
_HTTPStatus.UNAUTHORIZED,
|
|
'login/unauthorized',
|
|
"You are logged in but the resource you're trying to access isn't available to you",
|
|
)
|
|
|
|
|
|
# Success
|
|
|
|
def success(http_status: Any = _HTTPStatus.OK, /, **kargs):
|
|
try:
|
|
http_status = http_status[0]
|
|
except Exception:
|
|
pass
|
|
|
|
return dict(kargs, status='success'), http_status
|
|
|
|
# Schemas
|
|
|
|
from marshmallow import Schema, fields
|
|
|
|
class ErrorSchema(Schema):
|
|
status = fields.Constant('error')
|
|
code = fields.Str()
|
|
message = fields.Str(required=False)
|
|
|
|
class SuccessSchema(Schema):
|
|
status = fields.Constant('success')
|