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))
|
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")
|
||||||
|
|
Loading…
Reference in New Issue