From 14f735695763b1ac021d55d9b6a70ced874ce0fa Mon Sep 17 00:00:00 2001 From: iou1name Date: Thu, 10 May 2018 03:11:59 -0400 Subject: [PATCH] improved url grabbing and filename --- oppaiBot.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/oppaiBot.py b/oppaiBot.py index c7c42a9..7a182c9 100755 --- a/oppaiBot.py +++ b/oppaiBot.py @@ -37,6 +37,7 @@ class OppaiBot(irc.IRCClient): res.raise_for_status() fname = re.search(r'filename="(.+)"',res.headers['content-disposition']) fname = fname.group(1) + fname = fname.replace(u"â\x80\x93", "–") print("Saving torrent:", fname) with open(os.path.join(directory, fname), "wb") as file: for chunk in res.iter_content(100000): @@ -51,7 +52,7 @@ class OppaiBot(irc.IRCClient): # More advanced logic is left as an exercise to the reader. tags_in_msg = [tag in message for tag in self.tags] if any(tags_in_msg): - url = message.split(" - ")[1].split(" / ")[1] + url = re.findall(r"(http.+?) ", message)[1] self.save_torrent(url, self.watch_dir)