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