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)
109 lines
4.3 KiB
Python
109 lines
4.3 KiB
Python
from core.chat_blob import ChatBlob
|
|
from core.command_param_types import Any
|
|
from core.db import DB
|
|
from core.decorators import instance, command
|
|
from core.text import Text
|
|
|
|
|
|
@instance()
|
|
class WhompahController:
|
|
|
|
def inject(self, registry):
|
|
self.db: DB = registry.get_instance("db")
|
|
self.text: Text = registry.get_instance("text")
|
|
|
|
def pre_start(self):
|
|
self.db.load_sql_file(self.module_dir + "/sql/" + "whompah_cities.sql", pre_optimized=True)
|
|
self.db.create_view("whompah_cities")
|
|
self.db.create_view("whompah_cities_rel")
|
|
|
|
@command(command="whompah", params=[], access_level="member",
|
|
description="Show list of whompah cities")
|
|
def whompah_list_cmd(self, request):
|
|
cities = self.db.query("SELECT id, city_name, zone, faction, short_name FROM whompah_cities ORDER BY city_name")
|
|
|
|
blob = ""
|
|
for city in cities:
|
|
blob += f"{self.text.get_formatted_faction(city.faction.lower(), city.city_name)} " \
|
|
f"({self.text.make_tellcmd(city.short_name, f'whompah {city.short_name}')})\n"
|
|
|
|
return ChatBlob("Whompah Cities", blob)
|
|
|
|
@command(command="whompah", params=[Any("city1"), Any("city2")], access_level="member",
|
|
description="Show whompah route between two cities")
|
|
def whompah_travel_cmd(self, request, city_name1, city_name2):
|
|
city1 = self.get_whompah_city(city_name1)
|
|
city2 = self.get_whompah_city(city_name2)
|
|
|
|
if not city1:
|
|
return f"Could not find whompah city <highlight>{city_name1}</highlight>."
|
|
elif not city2:
|
|
return f"Could not find whompah city <highlight>{city_name2}</highlight>."
|
|
|
|
data = self.db.query("SELECT w1.*, w2.city2_id AS city_rel FROM whompah_cities w1 "
|
|
"JOIN whompah_cities_rel w2 ON w1.id = w2.city1_id")
|
|
|
|
cities = {}
|
|
for city in data:
|
|
rel = cities.get(city.id, {}).get("rel", [])
|
|
rel.append(city.city_rel)
|
|
cities[city.id] = city
|
|
cities[city.id]["rel"] = rel
|
|
|
|
path = self.format_path(self.find_path(cities, city1.id, city2.id))
|
|
return " -> ".join(path)
|
|
|
|
@command(command="whompah", params=[Any("city")], access_level="member",
|
|
description="Show whompah destinations for a city")
|
|
def whompah_city_cmd(self, request, city_name):
|
|
city = self.get_whompah_city(city_name)
|
|
|
|
if not city:
|
|
return f"Could not find whompah city <highlight>{city_name}</highlight>."
|
|
|
|
cities = self.db.query("SELECT w2.* FROM whompah_cities_rel w1 "
|
|
"JOIN whompah_cities w2 ON w1.city2_id = w2.id "
|
|
"WHERE w1.city1_id = ?", [city.id])
|
|
msg = f"From {city.city_name} you can get to: "
|
|
msg += ", ".join(map(lambda x: f"{self.text.get_formatted_faction(x.faction.lower(), x.city_name)} "
|
|
f"({x.short_name})", cities))
|
|
return msg
|
|
|
|
def get_whompah_city(self, city):
|
|
return self.db.query_single("SELECT id, city_name, zone, faction, short_name FROM whompah_cities "
|
|
"WHERE city_name LIKE ? "
|
|
"OR short_name LIKE ?", [city, city])
|
|
|
|
def find_path(self, cities, start_city_id, end_city_id):
|
|
def get_and_remove(key):
|
|
value = cities[key]
|
|
del cities[key]
|
|
return value
|
|
|
|
end_city = get_and_remove(end_city_id)
|
|
|
|
if start_city_id == end_city_id:
|
|
return end_city
|
|
|
|
# create stack and initialize
|
|
# start with ending city and traverse backwards so we don't have to reverse the result
|
|
stack = [end_city]
|
|
while stack:
|
|
root = stack.pop(0)
|
|
for rel in root["rel"]:
|
|
if rel in cities:
|
|
c = get_and_remove(rel)
|
|
c["parent"] = root
|
|
if c["id"] == start_city_id:
|
|
return c
|
|
stack.append(c)
|
|
return None
|
|
|
|
def format_path(self, path):
|
|
result = []
|
|
root = path
|
|
while root:
|
|
result.append(self.text.get_formatted_faction(root["faction"].lower(), root["city_name"]))
|
|
root = root.get("parent")
|
|
return result
|