#!/usr/bin/env python3.12
import os
import random
from typing import Optional
import aiosqlite as sqlite3
from fastapi import FastAPI
from pydantic import BaseModel
class RandMsgRequest(BaseModel):
"""
- **short**: Short randmsg?
"""
short: Optional[bool] = False
class RandMsg(FastAPI):
"""Random Message Endpoint"""
def __init__(self, app: FastAPI, util, constants, glob_state): # pylint: disable=super-init-not-called
self.app = app
self.util = util
self.constants = constants
self.glob_state = glob_state
self.endpoint_name = "randmsg"
app.add_api_route(f"/{self.endpoint_name}/", self.randmsg_handler, methods=["POST"])
async def randmsg_handler(self, data: RandMsgRequest = None):
"""
Get a randomly generated message
"""
random.seed()
short = data.short if data else False
if not short:
db_rand_selected = random.choice([0, 1, 3])
else:
db_rand_selected = 9
title_attr = "Unknown"
match db_rand_selected:
case 0:
randmsg_db_path = os.path.join("/",
"var",
"lib",
"singerdbs",
"qajoke.db") # For qajoke db
db_query = "SELECT id, ('Q: ' || question || '
A: ' \
|| 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("/",
"var",
"lib",
"singerdbs",
"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("/",
"var",
"lib",
"singerdbs",
"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("/",
"var",
"lib",
"singerdbs",
"philo.db") # For Philo DB
db_query = "SELECT id, (content || '
- ' || speaker) FROM quotes \
ORDER BY RANDOM() LIMIT 1" # For Philo DB
title_attr = "Philosophical Quotes DB"
case 4:
randmsg_db_path = os.path.join("/",
"var",
"lib",
"singerdbs",
"hate.db") # For Hate DB
db_query = """SELECT id, ("" || 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("/",
"var",
"lib",
"singerdbs",
"rjokes.db") # r/jokes DB
db_query = """SELECT id, (title || "
" || body) FROM jokes \
WHERE score >= 10000 ORDER BY RANDOM() LIMIT 1"""
title_attr = "r/jokes DB"
case 6:
randmsg_db_path = os.path.join("/",
"var",
"lib",
"singerdbs",
"donnies.db") # Donnies DB
random.seed()
twilight_or_mice = random.choice(["twilight", "mice"])
db_query = f"SELECT id, text FROM {twilight_or_mice} ORDER BY RANDOM() LIMIT 1"
title_attr = "Donnies DB"
async with sqlite3.connect(database=randmsg_db_path, timeout=1) as _db:
async with _db.execute(db_query) as _cursor:
result = await _cursor.fetchone()
(result_id, result_msg) = result
result_msg = result_msg.strip()
return {
"id": result_id,
"msg": result_msg,
'title': title_attr
}