73 lines
1.4 KiB
Python
73 lines
1.4 KiB
Python
from fastapi import FastAPI, Header, Path
|
|
from pydantic import BaseModel
|
|
from typing import List
|
|
from random import randint
|
|
from starlette.templating import Jinja2Templates
|
|
|
|
|
|
templates = Jinja2Templates(directory='templates')
|
|
|
|
|
|
app = FastAPI(
|
|
title='Capacity API', description='API for returning metric data.',
|
|
version='1.0.0', redoc_url='/documentation'
|
|
)
|
|
|
|
|
|
class MetricResponse(BaseModel):
|
|
data: List[int]
|
|
time: List[int]
|
|
metric: str
|
|
|
|
|
|
@app.get("/metrics/{metric}", response_model=MetricResponse)
|
|
def get_metric_data(
|
|
metric: str = Path(
|
|
...,
|
|
title='Metric',
|
|
description='A data metric. Valid metrics are ["bets", '
|
|
'"deposits", "registrations"]',
|
|
regex=r'^(bets$)|(deposits$)|(registrations$)',
|
|
),
|
|
earliest: str = Header(None, title='Earliest time',),
|
|
):
|
|
"""
|
|
Will return metric data for a timespan.
|
|
"""
|
|
# do something
|
|
time = []
|
|
data = []
|
|
|
|
for i in range(0, 10):
|
|
time.append(randint(1582238840, 1582238890))
|
|
data.append(randint(1, 5000))
|
|
|
|
return {
|
|
'metric': metric,
|
|
'time': time,
|
|
'data': data,
|
|
}
|
|
|
|
|
|
def test(*, user_agent=Header(None)):
|
|
return 's'
|
|
|
|
|
|
def test1(request):
|
|
pass
|
|
|
|
|
|
app.add_api_route('/test', endpoint=test, include_in_schema=False)
|
|
|
|
|
|
class Item(BaseModel):
|
|
name: str
|
|
description: str = None
|
|
price: float
|
|
tax: float = None
|
|
|
|
|
|
@app.post("/items/")
|
|
async def create_item(item: Item):
|
|
return item
|