Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import datetime
from typing import Dict, Union
from fastapi import FastAPI, Request, Response
from fastapi.exceptions import RequestValidationError
from starlette.responses import RedirectResponse
from . import data, elements
ERROR_CODE_MESSAGES = {
204: "No Data",
400: "Bad Request",
404: "Not Found",
409: "Conflict",
500: "Internal Server Error",
501: "Not Implemented",
503: "Service Unavailable",
}
VERSION = "version"
app = FastAPI(docs_url="/docs", openapi_prefix="/ws")
app.include_router(data.router)
app.include_router(elements.router)
@app.get("/", include_in_schema=False)
async def redirect_to_docs():
return RedirectResponse("/ws/docs")
@app.exception_handler(ValueError)
async def validation_exception_handler(request: Request, exc: RequestValidationError):
"""Value errors are user errors.
"""
if "format" in request.query_params:
data_format = str(request.query_params["format"])
return format_error(400, str(exc), data_format, request)
@app.exception_handler(Exception)
async def validation_exception_handler(request: Request, exc: RequestValidationError):
"""Other exceptions are server errors.
"""
if "format" in request.query_params:
data_format = str(request.query_params["format"])
return format_error(500, str(exc), data_format, request)
def format_error(
status_code: int, exception: str, format: str, request: Request
) -> Response:
"""Assign error_body value based on error format."""
if format == "json":
return json_error(status_code, exception, request.url)
else:
return Response(
text_error(status_code, exception, request.url), media_type="text/plain"
)
def json_error(code: int, error: Exception, url: str) -> Dict:
"""Format json error message.
Returns
-------
error_body : str
body of json error message.
"""
return {
"type": "Error",
"metadata": {
"title": ERROR_CODE_MESSAGES[code],
"status": code,
"error": str(error),
"generated": datetime.datetime.utcnow(),
"url": str(url),
},
}
def text_error(code: int, error: Exception, url: str) -> str:
"""Format error message as plain text
Returns
-------
error message formatted as plain text.
"""
return f"""Error {code}: {ERROR_CODE_MESSAGES[code]}
{error}
Usage details are available from
Request:
{url}
Request Submitted:
{datetime.datetime.utcnow().isoformat()}
Service Version:
{VERSION}
"""