iou1name/OppaiBot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

improved url grabbing and filename

Browse Source
This commit is contained in:
iou1name 2018-05-10 03:11:59 -04:00
parent 52113f6ad0
commit 14f7356957
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
oppaiBot.py
View File

@ -37,6 +37,7 @@ class OppaiBot(irc.IRCClient):
res.raise_for_status() res.raise_for_status()
fname = re.search(r'filename="(.+)"',res.headers['content-disposition']) fname = re.search(r'filename="(.+)"',res.headers['content-disposition'])
fname = fname.group(1) fname = fname.group(1)
fname = fname.replace(u"â\x80\x93", "–")
print("Saving torrent:", fname) print("Saving torrent:", fname)
with open(os.path.join(directory, fname), "wb") as file: with open(os.path.join(directory, fname), "wb") as file:
for chunk in res.iter_content(100000): 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. # More advanced logic is left as an exercise to the reader.
tags_in_msg = [tag in message for tag in self.tags] tags_in_msg = [tag in message for tag in self.tags]
if any(tags_in_msg): 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) self.save_torrent(url, self.watch_dir)