1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
import json
import random
from collections.abc import Iterable
import discord
from discord.ext.commands import Context, Converter
from pydis_core.utils.logging import get_logger
from rapidfuzz import fuzz
from bot.exts.fun.snakes._utils import SNAKE_RESOURCES
from bot.utils import disambiguate
log = get_logger(__name__)
class Snake(Converter):
"""Snake converter for the Snakes Cog."""
snakes = None
special_cases = None
async def convert(self, ctx: Context, name: str) -> str:
"""Convert the input snake name to the closest matching Snake object."""
await self.build_list()
name = name.lower()
if name == "python":
return "Python (programming language)"
def get_potential(iterable: Iterable, *, threshold: int = 80) -> list[str]:
nonlocal name
potential = []
for item in iterable:
original, item = item, item.lower()
if name == item:
return [original]
a, b = fuzz.ratio(name, item), fuzz.partial_ratio(name, item)
if a >= threshold or b >= threshold:
potential.append(original)
return potential
# Handle special cases
if name.lower() in self.special_cases:
return self.special_cases.get(name.lower(), name.lower())
names = {snake["name"]: snake["scientific"] for snake in self.snakes}
all_names = names.keys() | names.values()
timeout = len(all_names) * (3 / 4)
embed = discord.Embed(
title="Found multiple choices. Please choose the correct one.", colour=0x59982F)
embed.set_author(name=ctx.author.display_name, icon_url=ctx.author.display_avatar.url)
name = await disambiguate(ctx, get_potential(all_names), timeout=timeout, embed=embed)
return names.get(name, name)
@classmethod
async def build_list(cls) -> None:
"""Build list of snakes from the static snake resources."""
# Get all the snakes
if cls.snakes is None:
cls.snakes = json.loads((SNAKE_RESOURCES / "snake_names.json").read_text("utf8"))
# Get the special cases
if cls.special_cases is None:
special_cases = json.loads((SNAKE_RESOURCES / "special_snakes.json").read_text("utf8"))
cls.special_cases = {snake["name"].lower(): snake for snake in special_cases}
@classmethod
async def random(cls) -> str:
"""
Get a random Snake from the loaded resources.
This is stupid. We should find a way to somehow get the global session into a global context,
so I can get it from here.
"""
await cls.build_list()
names = [snake["scientific"] for snake in cls.snakes]
return random.choice(names)
|
