60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
from fastapi import APIRouter, HTTPException, Depends, Header
|
|
import logging
|
|
import db
|
|
import auth.api_security as auth
|
|
from pydantic import BaseModel
|
|
|
|
|
|
|
|
def check_token(token:str = Header(...)):
|
|
if not auth.check_token(token):
|
|
raise HTTPException(status_code=401, detail="Invalid token")
|
|
return token
|
|
|
|
def check_admin_token(admin_token: str = Header(...)):
|
|
if not auth.check_admin_token(admin_token):
|
|
raise HTTPException(status_code=401, detail="No admin token")
|
|
return admin_token
|
|
|
|
|
|
#Logging
|
|
|
|
logging.basicConfig(
|
|
filename='api.log', # Log to a file
|
|
level=logging.DEBUG, # Log everything (DEBUG and above)
|
|
format='%(asctime)s %(levelname)s %(message)s'
|
|
)
|
|
|
|
logging.info('Application started')
|
|
try:
|
|
logging.debug('Debug info here')
|
|
except Exception as e:
|
|
logging.exception('An error occurred')
|
|
|
|
router = APIRouter(prefix="/tokens", tags=["tokens"])
|
|
|
|
|
|
class Item_duration(BaseModel):
|
|
duration: str | int
|
|
|
|
class Item_token(BaseModel):
|
|
token: str
|
|
|
|
@router.get("/help")
|
|
def help():
|
|
return{"duration": "string or int"}
|
|
|
|
|
|
@router.post("/create")
|
|
def create_token(duration: Item_duration, admin_token: str = Depends(check_admin_token)):
|
|
new_token = auth.generate_token(duration.duration)
|
|
return {"new token": new_token}
|
|
|
|
@router.delete("/delete")
|
|
def delete(token: Item_token, admin_token: str = Depends(check_admin_token)):
|
|
auth.delete_token(token.token)
|
|
return {"messages": "token deleted"}
|
|
|
|
@router.get("/test")
|
|
def test(token: str = Depends(check_token)):
|
|
return {"message": "you can authenticate", "status": "ok"} |