diff --git a/endpoints/rand_msg.py b/endpoints/rand_msg.py index acc8779..854b704 100644 --- a/endpoints/rand_msg.py +++ b/endpoints/rand_msg.py @@ -2,7 +2,7 @@ import os import random -from typing import Union, LiteralString +from typing import LiteralString, Optional, Union import aiosqlite as sqlite3 from fastapi import FastAPI from fastapi.responses import JSONResponse @@ -21,18 +21,17 @@ class RandMsg(FastAPI): app.add_api_route(f"/{self.endpoint_name}", self.randmsg_handler, methods=["POST"]) - async def randmsg_handler(self, data: RandMsgRequest) -> JSONResponse: + async def randmsg_handler(self, data: Optional[RandMsgRequest] = None) -> JSONResponse: """ Get a randomly generated message - **short**: Optional, if True, will limit length of returned random messages to <=126 characters (Discord restriction related) """ random.seed() - short: bool = data.short if data.short else False - - if not short: - db_rand_selected: int = random.choice([0, 1, 3]) - else: - db_rand_selected = 9 + short: Optional[bool] = False + if isinstance(data, RandMsgRequest): + short = data.short + db_rand_selected: int = 9 + db_rand_selected = random.choice([0, 1, 3]) title_attr: str = "Unknown" match db_rand_selected: