Compare commits
4 Commits
16d1c9853b
...
9ef9c8454b
Author | SHA1 | Date | |
---|---|---|---|
9ef9c8454b | |||
8f2a575010 | |||
654f2982de | |||
f0230bbbc3 |
@ -13,6 +13,6 @@
|
|||||||
"TESTKEY"
|
"TESTKEY"
|
||||||
],
|
],
|
||||||
"max-data": "10GB",
|
"max-data": "10GB",
|
||||||
"database": "databse.json"
|
"database": "database.json"
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
@ -4,7 +4,7 @@ import argparse
|
|||||||
import json
|
import json
|
||||||
from os.path import exists
|
from os.path import exists
|
||||||
|
|
||||||
from fastapi import FastAPI, Request, HTTPException
|
from fastapi import FastAPI, Request, HTTPException, Query
|
||||||
from fastapi.responses import StreamingResponse
|
from fastapi.responses import StreamingResponse
|
||||||
from fastapi import status
|
from fastapi import status
|
||||||
from hypercorn.config import Config
|
from hypercorn.config import Config
|
||||||
@ -32,7 +32,7 @@ if not exists(DATABASE):
|
|||||||
save_database(DATABASE, {'data-used': 0})
|
save_database(DATABASE, {'data-used': 0})
|
||||||
|
|
||||||
|
|
||||||
api = FastAPI()
|
api = FastAPI(docs_url=None, redoc_url=None)
|
||||||
|
|
||||||
|
|
||||||
class MaxSizePerRequestError(Exception):
|
class MaxSizePerRequestError(Exception):
|
||||||
@ -44,7 +44,7 @@ class MinSizePerRequestError(Exception):
|
|||||||
|
|
||||||
|
|
||||||
@api.get('/')
|
@api.get('/')
|
||||||
def test_data(api_key: str, size: str, request: Request) -> StreamingResponse:
|
def test_data(request: Request, size: str, api_key: str = Query(..., alias="api-key")) -> StreamingResponse:
|
||||||
try:
|
try:
|
||||||
if api_key not in AUTHORIZED_KEYS:
|
if api_key not in AUTHORIZED_KEYS:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
|
Loading…
Reference in New Issue
Block a user