Delete modules/pirates_list/pirates_list.py
This commit is contained in:
parent
e599fde099
commit
5b2aa79b9f
@ -1,256 +0,0 @@
|
|||||||
# modules/pirates_list/pirates_list.py
|
|
||||||
import asyncio
|
|
||||||
import discord
|
|
||||||
from discord.ext import commands
|
|
||||||
from mod_perms import require_mod_ctx # ctx-aware mod gate
|
|
||||||
|
|
||||||
|
|
||||||
class PiratesListCog(commands.Cog):
|
|
||||||
"""
|
|
||||||
Maintains a compact, alphabetized pirates list in the configured channel.
|
|
||||||
|
|
||||||
Entry format:
|
|
||||||
- **{Character}** (*{Account}*) [{Threat}%]
|
|
||||||
- In group: {never/rarely/...}. Destructive: {never/rarely/...}. Encounters: N. Last: <t:UNIX:R>
|
|
||||||
|
|
||||||
Posts are chunked to stay <2000 chars and previous posts are deleted on refresh.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, bot):
|
|
||||||
self.bot = bot
|
|
||||||
cfg = bot.config["DEFAULT"]
|
|
||||||
self.list_channel_id = int(cfg["pirates_list_channel_id"])
|
|
||||||
try:
|
|
||||||
self.group_threshold = int(cfg.get("threat_group_threshold", "3"))
|
|
||||||
except Exception:
|
|
||||||
self.group_threshold = 3
|
|
||||||
try:
|
|
||||||
self.min_samples = int(cfg.get("threat_min_samples_for_stats", "3"))
|
|
||||||
except Exception:
|
|
||||||
self.min_samples = 3
|
|
||||||
|
|
||||||
# serialize refreshes per guild
|
|
||||||
self._locks = {}
|
|
||||||
self._no_mentions = discord.AllowedMentions.none()
|
|
||||||
|
|
||||||
def _lock_for(self, guild_id: int):
|
|
||||||
import asyncio
|
|
||||||
self._locks.setdefault(guild_id, asyncio.Lock())
|
|
||||||
return self._locks[guild_id]
|
|
||||||
|
|
||||||
# send settings: never ping on posted content
|
|
||||||
self._no_mentions = discord.AllowedMentions.none()
|
|
||||||
|
|
||||||
# ----------------- utils -----------------
|
|
||||||
|
|
||||||
def _lock_for(self, guild_id: int) -> asyncio.Lock:
|
|
||||||
self._locks.setdefault(guild_id, asyncio.Lock())
|
|
||||||
return self._locks[guild_id]
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def _esc(s: str) -> str:
|
|
||||||
"""Escape markdown and neuter @mentions in user-provided strings."""
|
|
||||||
safe = discord.utils.escape_markdown(str(s))
|
|
||||||
return safe.replace("@", "@\u200b")
|
|
||||||
|
|
||||||
def _encounters_for(self, pirate: dict):
|
|
||||||
"""Return encounters matching either char name or account (case-insensitive), defensively."""
|
|
||||||
acct_l = str(pirate.get("account_name", "")).lower()
|
|
||||||
char_l = str(pirate.get("character_name", "")).lower()
|
|
||||||
out = []
|
|
||||||
for e in self.bot.data_manager.get("encounters"):
|
|
||||||
try:
|
|
||||||
ident = str(e.get("identifier", "")).lower()
|
|
||||||
if ident in (acct_l, char_l):
|
|
||||||
out.append(e)
|
|
||||||
except Exception:
|
|
||||||
continue
|
|
||||||
return out
|
|
||||||
|
|
||||||
def _bucket(self, rate: float, samples: int) -> str:
|
|
||||||
if samples < self.min_samples:
|
|
||||||
return "unknown"
|
|
||||||
if rate <= 0:
|
|
||||||
return "never"
|
|
||||||
if rate <= 0.25:
|
|
||||||
return "rarely"
|
|
||||||
if rate <= 0.60:
|
|
||||||
return "sometimes"
|
|
||||||
if rate <= 0.85:
|
|
||||||
return "often"
|
|
||||||
return "always"
|
|
||||||
|
|
||||||
def _format_entry(self, p: dict) -> str:
|
|
||||||
encs = self._encounters_for(p)
|
|
||||||
total = len(encs)
|
|
||||||
|
|
||||||
def _safe_int(v, default=0):
|
|
||||||
try:
|
|
||||||
return int(v)
|
|
||||||
except Exception:
|
|
||||||
return default
|
|
||||||
|
|
||||||
group_rate = (
|
|
||||||
sum(1 for e in encs if _safe_int(e.get("group_size"), 0) >= self.group_threshold) / total
|
|
||||||
if total else 0.0
|
|
||||||
)
|
|
||||||
destroy_rate = (
|
|
||||||
sum(1 for e in encs if bool(e.get("destruction"))) / total
|
|
||||||
if total else 0.0
|
|
||||||
)
|
|
||||||
groups_str = self._bucket(group_rate, total)
|
|
||||||
destr_str = self._bucket(destroy_rate, total)
|
|
||||||
|
|
||||||
last_ts = None
|
|
||||||
if total:
|
|
||||||
try:
|
|
||||||
last_ts = max(int(float(e.get("timestamp", 0))) for e in encs)
|
|
||||||
except Exception:
|
|
||||||
last_ts = None
|
|
||||||
|
|
||||||
char = self._esc(p.get("character_name", "Unknown"))
|
|
||||||
acct = self._esc(p.get("account_name", "Unknown#00000"))
|
|
||||||
threat = int(p.get("threat_level", 0))
|
|
||||||
last_str = f"<t:{last_ts}:R>" if last_ts else "—"
|
|
||||||
|
|
||||||
line1 = f"- **{char}** (*{acct}*) [{threat}%]"
|
|
||||||
line2 = f" - In group: {groups_str}. Destructive: {destr_str}. Encounters: {total}. Last: {last_str}"
|
|
||||||
return f"{line1}\n{line2}"
|
|
||||||
|
|
||||||
async def refresh_list(self, guild: discord.Guild):
|
|
||||||
"""Edit list messages in place; only send extra messages when we need more chunks (new pirates)."""
|
|
||||||
# ---- serialize per guild ----
|
|
||||||
lock = getattr(self, "_locks", {}).get(guild.id)
|
|
||||||
if lock is None:
|
|
||||||
# tiny fallback if you didn't add _lock_for()
|
|
||||||
import asyncio as _asyncio
|
|
||||||
if not hasattr(self, "_locks"):
|
|
||||||
self._locks = {}
|
|
||||||
self._locks[guild.id] = _asyncio.Lock()
|
|
||||||
lock = self._locks[guild.id]
|
|
||||||
|
|
||||||
async with lock:
|
|
||||||
channel = guild.get_channel(self.list_channel_id)
|
|
||||||
if not channel:
|
|
||||||
print("[pirates_list] list channel not found:", self.list_channel_id)
|
|
||||||
return
|
|
||||||
|
|
||||||
dm = self.bot.data_manager
|
|
||||||
allow = getattr(self, "_no_mentions", discord.AllowedMentions.none())
|
|
||||||
|
|
||||||
# ---- load & prune existing posts for this guild/channel ----
|
|
||||||
records = [r for r in dm.get("pirates_list_posts")
|
|
||||||
if r.get("guild_id") == guild.id and r.get("channel_id") == self.list_channel_id]
|
|
||||||
|
|
||||||
# fetch messages (drop any that vanished)
|
|
||||||
msgs, kept_records = [], []
|
|
||||||
for r in records:
|
|
||||||
try:
|
|
||||||
m = await channel.fetch_message(int(r["message_id"]))
|
|
||||||
msgs.append(m)
|
|
||||||
kept_records.append(r)
|
|
||||||
except Exception:
|
|
||||||
# prune dead record
|
|
||||||
dm.remove("pirates_list_posts", lambda x, mid=r["message_id"]: x.get("message_id") == mid)
|
|
||||||
|
|
||||||
records = kept_records # only live ones, in stored order
|
|
||||||
|
|
||||||
# ---- build fresh, sorted contents ----
|
|
||||||
pirates = sorted(
|
|
||||||
dm.get("pirates"),
|
|
||||||
key=lambda x: (
|
|
||||||
str(x.get("character_name", "")).lower(),
|
|
||||||
str(x.get("account_name", "")).lower()
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
# Empty state
|
|
||||||
if not pirates:
|
|
||||||
placeholder = "_No verified pirates yet._"
|
|
||||||
if msgs:
|
|
||||||
# edit first, delete the rest
|
|
||||||
if msgs[0].content != placeholder:
|
|
||||||
try:
|
|
||||||
await msgs[0].edit(content=placeholder, allowed_mentions=allow)
|
|
||||||
except Exception as e:
|
|
||||||
print("[pirates_list] edit placeholder failed:", repr(e))
|
|
||||||
# remove extra posts/records
|
|
||||||
for extra in msgs[1:]:
|
|
||||||
try:
|
|
||||||
await extra.delete()
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
dm.remove("pirates_list_posts", lambda r, mid=extra.id: r.get("message_id") == mid)
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
m = await channel.send(placeholder, allowed_mentions=allow)
|
|
||||||
dm.add("pirates_list_posts", {
|
|
||||||
"guild_id": guild.id,
|
|
||||||
"channel_id": self.list_channel_id,
|
|
||||||
"message_id": m.id
|
|
||||||
})
|
|
||||||
except Exception as e:
|
|
||||||
print("[pirates_list] send placeholder failed:", repr(e))
|
|
||||||
return
|
|
||||||
|
|
||||||
# Chunk into <2000 char blocks (~1900 for margin)
|
|
||||||
chunks, current = [], ""
|
|
||||||
for p in pirates:
|
|
||||||
entry = self._format_entry(p)
|
|
||||||
if len(current) + len(entry) + 1 > 1900:
|
|
||||||
chunks.append(current.rstrip())
|
|
||||||
current = entry + "\n"
|
|
||||||
else:
|
|
||||||
current += entry + "\n"
|
|
||||||
if current.strip():
|
|
||||||
chunks.append(current.rstrip())
|
|
||||||
|
|
||||||
# ---- edit existing messages with new content ----
|
|
||||||
common = min(len(msgs), len(chunks))
|
|
||||||
for i in range(common):
|
|
||||||
if msgs[i].content != chunks[i]:
|
|
||||||
try:
|
|
||||||
await msgs[i].edit(content=chunks[i], allowed_mentions=allow)
|
|
||||||
except Exception as e:
|
|
||||||
print("[pirates_list] edit block failed:", repr(e))
|
|
||||||
|
|
||||||
# ---- if we need *more* messages (usually after adding a pirate), send them ----
|
|
||||||
if len(chunks) > len(msgs):
|
|
||||||
for i in range(len(msgs), len(chunks)):
|
|
||||||
try:
|
|
||||||
m = await channel.send(chunks[i], allowed_mentions=allow)
|
|
||||||
dm.add("pirates_list_posts", {
|
|
||||||
"guild_id": guild.id,
|
|
||||||
"channel_id": self.list_channel_id,
|
|
||||||
"message_id": m.id
|
|
||||||
})
|
|
||||||
except Exception as e:
|
|
||||||
print("[pirates_list] send block failed:", repr(e))
|
|
||||||
|
|
||||||
# ---- if we need fewer messages (e.g., pirate removed), delete extras ----
|
|
||||||
elif len(chunks) < len(msgs):
|
|
||||||
extras = msgs[len(chunks):]
|
|
||||||
for m in extras:
|
|
||||||
try:
|
|
||||||
await m.delete()
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
dm.remove("pirates_list_posts", lambda r, mid=m.id: r.get("message_id") == mid)
|
|
||||||
|
|
||||||
# Manual refresh command (hybrid: works as /pirates_list_refresh and !pirates_list_refresh)
|
|
||||||
@commands.hybrid_command(name="pirates_list_refresh", description="Rebuild the compact pirates list")
|
|
||||||
@commands.cooldown(1, 10, commands.BucketType.guild) # tiny anti-spam
|
|
||||||
async def pirates_list_refresh(self, ctx: commands.Context):
|
|
||||||
if not await require_mod_ctx(ctx, "This command is restricted to moderators."):
|
|
||||||
return
|
|
||||||
if not ctx.guild:
|
|
||||||
return await ctx.reply("Use this in a server.", ephemeral=True)
|
|
||||||
|
|
||||||
await self.refresh_list(ctx.guild)
|
|
||||||
|
|
||||||
is_slash = hasattr(ctx, "interaction") and ctx.interaction is not None
|
|
||||||
await ctx.reply("Pirates list refreshed.", ephemeral=is_slash)
|
|
||||||
|
|
||||||
|
|
||||||
async def setup(bot):
|
|
||||||
await bot.add_cog(PiratesListCog(bot))
|
|
||||||
Loading…
Reference in New Issue
Block a user