Minidodo
c04f76c0db
Fixed command & event threading Events are now threaded by event_type (i.e. all buddy_logon events get ran in the same one) Added default preferences Fixed recipe loading for multiple installs (i.e. on different machines)
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
import random
|
|
import time
|
|
|
|
from core.command_param_types import Const, Int, Any
|
|
from core.db import DB
|
|
from core.decorators import instance, command
|
|
from core.text import Text
|
|
|
|
|
|
@instance()
|
|
class QuoteController:
|
|
def inject(self, registry):
|
|
self.db: DB = registry.get_instance("db")
|
|
self.text: Text = registry.get_instance("text")
|
|
|
|
def start(self):
|
|
self.db.exec("CREATE TABLE IF NOT EXISTS quote ("
|
|
"id INT PRIMARY KEY AUTO_INCREMENT, "
|
|
"char_id INT NOT NULL, "
|
|
"created_at INT NOT NULL, "
|
|
"content VARCHAR(4096) NOT NULL)")
|
|
|
|
@command(command="quote", params=[], access_level="member",
|
|
description="Show a random quote")
|
|
def quote_command(self, request):
|
|
quote = self.get_quote_info()
|
|
|
|
if quote:
|
|
return quote
|
|
else:
|
|
return "There are no quotes to display."
|
|
|
|
@command(command="quote", params=[Int("quote_id")], access_level="member",
|
|
description="Show a specific quote")
|
|
def quote_view_command(self, _, quote_id):
|
|
quote = self.get_quote_info(quote_id)
|
|
|
|
if quote:
|
|
return quote
|
|
else:
|
|
return f"Could not find quote with ID <highlight>{quote_id:d}</highlight>."
|
|
|
|
@command(command="quote", params=[Const("add"), Any("quote")], access_level="member",
|
|
description="Show a specific quote")
|
|
def quote_add_command(self, request, _, quote):
|
|
if len(quote) > 4096:
|
|
return "Your quote must be less than 4096 characters."
|
|
|
|
self.db.exec("INSERT INTO quote (char_id, created_at, content) "
|
|
"VALUES (?, ?, ?)",
|
|
[request.sender.char_id, int(time.time()), quote])
|
|
|
|
return "Your quote has been added successfully."
|
|
|
|
def get_quote_info(self, quote_id=None):
|
|
stats = self.db.query_single("SELECT COUNT(1) AS count, MAX(id) AS max FROM quote")
|
|
|
|
if stats.count == 0:
|
|
return None
|
|
|
|
if not quote_id:
|
|
row = self.db.query_single("SELECT q.*, p.name FROM quote q "
|
|
"LEFT JOIN player p ON q.char_id = p.char_id "
|
|
"LIMIT 1 OFFSET ?",
|
|
[random.randint(0, stats.count - 1)])
|
|
else:
|
|
row = self.db.query_single("SELECT q.*, p.name FROM quote q "
|
|
"LEFT JOIN player p ON q.char_id = p.char_id "
|
|
"WHERE id = ?",
|
|
[quote_id])
|
|
|
|
return f"{row.id:d}. {row.content}"
|