api/endpoints/rand_msg.py

import os
import random
from typing import LiteralString, Optional, Union
import aiosqlite as sqlite3
from fastapi import FastAPI
from fastapi.responses import JSONResponse
from .constructors import RandMsgRequest

class RandMsg(FastAPI):
    """
    Random Message Endpoint
    """   
    def __init__(self, app: FastAPI,
                 util, constants) -> None:
        self.app: FastAPI = app
        self.util = util
        self.constants = constants
        self.endpoint_name = "randmsg"

        app.add_api_route(f"/{self.endpoint_name}", self.randmsg_handler, methods=["POST"])
    
    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: Optional[bool] = False
        if isinstance(data, RandMsgRequest):
            short = data.short
            if short:
                db_rand_selected: int = 9
        db_rand_selected = random.choice([0, 1, 3])
        title_attr: str = "Unknown"

        match db_rand_selected:
            case 0:
                randmsg_db_path: Union[str, LiteralString] = os.path.join("/usr/local/share",
                                               "sqlite_dbs", "qajoke.db")  # For qajoke db
                db_query: str = "SELECT id, ('<b>Q:</b> ' || question || '<br/><b>A:</b> ' \
                    || answer) FROM jokes ORDER BY RANDOM() LIMIT 1" # For qajoke db
                title_attr = "QA Joke DB"
            case 1 | 9:
                randmsg_db_path = os.path.join("/usr/local/share",
                                               "sqlite_dbs",
                                               "randmsg.db") # For randmsg db
                db_query = "SELECT id, msg FROM msgs WHERE \
                    LENGTH(msg) <= 180 ORDER BY RANDOM() LIMIT 1" # For randmsg db
                if db_rand_selected == 9:
                    db_query = db_query.replace("<= 180", "<= 126")
                title_attr = "Random Msg DB"
            case 2:
                randmsg_db_path = os.path.join("/usr/local/share",
                                               "sqlite_dbs",
                                               "trump.db") # For Trump Tweet DB
                db_query = "SELECT id, content FROM tweets \
                    ORDER BY RANDOM() LIMIT 1" # For Trump Tweet DB
                title_attr = "Trump Tweet DB"
            case 3:
                randmsg_db_path = os.path.join("/usr/local/share",
                                               "sqlite_dbs",
                                               "philo.db") # For Philo DB
                db_query = "SELECT id, (content || '<br> - ' || speaker) FROM quotes \
                    ORDER BY RANDOM() LIMIT 1"  # For Philo DB
                title_attr = "Philosophical Quotes DB"
            case 4:
                randmsg_db_path = os.path.join("/usr/local/share",
                                               "sqlite_dbs",
                                               "hate.db") # For Hate DB
                db_query = """SELECT id, ("<font color='#FF0000'>" || comment) FROM hate_speech \
                    WHERE length(comment) <= 180 ORDER BY RANDOM() LIMIT 1"""
                title_attr = "Hate Speech DB"
            case 5:
                randmsg_db_path = os.path.join("/usr/local/share",
                                               "sqlite_dbs",
                                               "rjokes.db") # r/jokes DB
                db_query = """SELECT id, (title || "<br>" || body) FROM jokes \
                    WHERE score >= 10000 ORDER BY RANDOM() LIMIT 1"""
                title_attr = "r/jokes DB"
                                              
        async with sqlite3.connect(database=randmsg_db_path, timeout=1) as _db:
            async with await _db.execute(db_query) as _cursor:
                result: sqlite3.Row = await _cursor.fetchone()
                (result_id, result_msg) = result
                result_msg = result_msg.strip()
                return JSONResponse(content=
                                    {
                                        "id": result_id,
                                        "msg": result_msg,
                                        "title": title_attr,
                })