This commit is contained in:
Auric Vente 2024-02-22 14:45:50 -06:00
parent 9984090447
commit cf9f9eacc0
1 changed files with 7 additions and 4 deletions

11
main.py
View File

@ -35,7 +35,7 @@ def get_path(name):
return str(Path(HERE, name)) return str(Path(HERE, name))
def extract_range(string): def extract_range(string):
pattern = re.compile(r"(\d+)(?:-(\d+))?") pattern = re.compile(r"\s*(\d+)\s*(?:-\s*(\d+))?\s*")
match = pattern.search(string) match = pattern.search(string)
if not match: if not match:
@ -49,6 +49,9 @@ def extract_range(string):
else: else:
return [int(start), int(end)] return [int(start), int(end)]
def clean_list(lst):
return list(filter(lambda x: x != "", lst))
headers = { headers = {
"User-Agent": "gluebot", "User-Agent": "gluebot",
"Origin": "https://deek.chat", "Origin": "https://deek.chat",
@ -142,8 +145,8 @@ async def on_message(ws, message):
await gif_wins(args[0], room_id) await gif_wins(args[0], room_id)
elif cmd == "numbers" or cmd == "number" or cmd == "nums" or cmd == "num": elif cmd == "numbers" or cmd == "number" or cmd == "nums" or cmd == "num":
if len(args) >= 1: if len(args) > 0:
arg = args[0] arg = " ".join(clean_list(args))
else: else:
arg = None arg = None
@ -210,7 +213,7 @@ async def gif_numbers(arg, room_id):
if numbers[0] < numbers[1]: if numbers[0] < numbers[1]:
num = random_int(numbers[0], numbers[1]) num = random_int(numbers[0], numbers[1])
if num <= 0: if num < 0:
num = random_int(0, 999) num = random_int(0, 999)
input_path = get_path("numbers.png") input_path = get_path("numbers.png")