tza/handy-scripts
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added ab!torrent command

Browse Source
This commit is contained in:
marios 2020-01-23 21:24:34 +02:00
parent c9de243847
commit ddbb5b106b
1 changed files with 31 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
Animebyter/animebyter.py
View File

@ -39,7 +39,7 @@ downloading = []
class Anime: class Anime:
def __init__(self,name,le,tl,res): def __init__(self,name,le,tl,res):
self.title = name self.title = name.replace("/","-")
self.last_episode = le self.last_episode = le
self.torrent_link = tl self.torrent_link = tl
self.resolution = res.strip() self.resolution = res.strip()
@ -74,21 +74,24 @@ async def get_airing():
continue continue
return r return r
async def add_torrent(anime): async def add_anime(anime):
print("Adding episode {} of {}".format(anime.last_episode,anime.title)) print("Adding episode {} of {}".format(anime.last_episode,anime.title))
try: try:
res = await web.post(QB_URL+'/command/download',data={'urls':anime.torrent_link,'savepath':join(path,anime.title),'category':'Anime'}) await add_torrent(anime.torrent_link, join(path,anime.title), 'Anime')
except Exception as e: except Exception as e:
print(str(e)) print("Failed to add episode {} of {} ({})".format(anime.last_episode,anime.title,e))
return else:
finally:
res.close()
if res.status==200:
msg = "Added episode {} of {}".format(anime.last_episode,anime.title) msg = "Added episode {} of {}".format(anime.last_episode,anime.title)
print(msg) print(msg)
return msg return msg
else:
print("Failed to add episode {} of {} ({})".format(anime.last_episode,anime.title,res.status))
async def add_torrent(url,path,category):
async with web.post(QB_URL+'/command/download',data={'urls':url,'savepath':path,'category':category}) as res:
if res.status==200:
return 1
else:
raise Exception(await res.text())
async def main(): async def main():
print("Starting new episode checker") print("Starting new episode checker")
@ -101,7 +104,7 @@ async def main():
if res: if res:
le = res[0]['last_episode'] le = res[0]['last_episode']
if le<i.last_episode and i.resolution in ("1080p"): if le<i.last_episode and i.resolution in ("1080p"):
msg = await add_torrent(i) msg = await add_anime(i)
if msg: if msg:
await client.get_channel(get_channel_id()).send(msg) await client.get_channel(get_channel_id()).send(msg)
db.update({'last_episode':i.last_episode},Query().title==i.title) db.update({'last_episode':i.last_episode},Query().title==i.title)
@ -211,7 +214,23 @@ async def down(ctx):
return await ctx.send(e) return await ctx.send(e)
if msg>=len(airing): if msg>=len(airing):
return await ctx.send("Invalid number") return await ctx.send("Invalid number")
await ctx.send(await add_torrent(airing[msg])) await ctx.send(await add_anime(airing[msg]))
@client.command(pass_context=True)
async def torrent(ctx,url,path=None,category=None):
if not path:
if 'jpopsuki' in url:
path = '/Music/'
else:
path = '/Downloads/'
if not category:
if 'jpopsuki' in url:
category = 'Music'
try:
await ctx.send(await add_torrent(url,path,category))
except Exception as e:
await ctx.send("Failed to add torrent: {}".format(e))
@client.command(pass_context=True) @client.command(pass_context=True)
async def setchannel(ctx): async def setchannel(ctx):