crony/anidl
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Added cli arguments and downloading support

Browse Source
This commit is contained in:
CronyAkatsuki 2021-07-03 15:05:12 +02:00
parent 1ff4f58dcd
commit 8570384308
1 changed files with 90 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

130
adl.py
View File

@ -1,21 +1,48 @@
import os, subprocess, sys import os, subprocess, sys, argparse
from iterfzf import iterfzf from iterfzf import iterfzf
from time import sleep from time import sleep
# global variables # global variables
account = "0" # choose an account account = "0" # trackma account
episode = "" # specific episode
retrieve = True # retrieve new list retrieve = True # retrieve new list
player = "mpv" # specific player player = "mpv" # specific player
download = False # specify whether to download or not download = False # specify whether to download or not
msg = "watching"
ap = argparse.ArgumentParser()
ap.add_argument("-a", "--account", required=False,
help="By default trackma will use account 1. Use '-a 2' for example to change trackma account")
ap.add_argument("-p", "--player", required=False,
help="Define player used for streaming. Ex: \033[0;36mpyadl -p mpv\033[0m")
ap.add_argument("-d", "--download", required=False, type=bool, nargs='?', const=True, default=False,
help="Download instead of streaming")
ap.add_argument("-v", "--version", required=False,
help="Display version and exit")
args = vars(ap.parse_args())
if args['account']:
account = str(int(args["account"]) - 1)
if args["player"]:
player = str(args["player"])
if args["download"]:
download = True
msg = "downloading"
if args["version"]:
print("version: 0.1")
sys.exit()
# colored print # colored print
def color_print(text): def color_print(text):
print("\033[0;36m" + text + " \033[0m") print("\033[0;36m" + text + " \033[0m")
# colored watch primpt # colored watch primpt
def watch_prompt(title, episode): def watch_prompt(title, episode):
print("Now watching \033[0;34m" + title + "\033[0m, episode \033[0;34m" + str(episode) + " \033[0m") print("Now " + msg + " \033[0;34m" + title + "\033[0m, episode \033[0;34m" + str(episode) + " \033[0m")
# colored input # colored input
def color_prommpt(text): def color_prommpt(text):
@ -28,7 +55,7 @@ def iter_list(alist):
sleep(0.01) sleep(0.01)
# retrieve new list # retrieve new list
def retrieve_list(account): def retrieve_list():
color_print ("Running trackma retrieve for account " + account + "...") color_print ("Running trackma retrieve for account " + account + "...")
os.system("trackma -a " + account + " retrieve") os.system("trackma -a " + account + " retrieve")
os.system("cls") os.system("cls")
@ -40,7 +67,7 @@ def retrieve_list_update(account):
os.system("cls") os.system("cls")
# load list # load list
def load_list(account): def load_list():
alist = subprocess.getoutput("trackma -a " + account + " list").splitlines() alist = subprocess.getoutput("trackma -a " + account + " list").splitlines()
alist.pop(0) alist.pop(0)
alist.pop() alist.pop()
@ -80,53 +107,76 @@ def get_score(full_choice):
return full_choice return full_choice
# next episode # next episode
def next_episode(title,episode,player): def next_episode(title,episode):
watch_next = True if not download:
while watch_next: watch_next = True
while watch_next:
episode = episode + 1
watch_prompt(title, str(episode))
os.system('anime dl "' + title + '" --episodes ' + str(episode) + ' --play ' + player)
while True:
color_print("Current watched episode: " + str(episode))
yn = color_prommpt("Wanna watch next episode? [Y/n]: ")
if yn == "Y" or yn == "y":
break
elif yn == "N" or yn == "n":
watch_next = False
break
else:
episode = episode + 1 episode = episode + 1
watch_prompt(title, str(episode)) watch_prompt(title, str(episode))
os.system('anime dl "' + title + '" --episodes ' + str(episode) + ' --play ' + player) os.system('anime dl "' + title + '" --episodes ' + str(episode))
while True:
color_print("Current watched episode: " + str(episode))
yn = color_prommpt("Wanna watch next episode? [Y/n]: ")
if yn == "Y" or yn == "y":
break
elif yn == "N" or yn == "n":
watch_next = False
break
# all from last watched # all from last watched
def all_from_last(title,episode, player): def all_from_last(title,episode):
watch_prompt(title, str(episode) + " all left episodes") watch_prompt(title, str(episode) + " all left episodes")
os.system('anime dl "' + title + '" --episodes ' + str(episode + 1) + ': --play' + player) if not download:
os.system('anime dl "' + title + '" --episodes ' + str(episode + 1) + ': --play' + player)
else:
os.system('anime dl "' + title + '" --episodes ' + str(episode + 1) + ':')
# all episode # all episode
def all_episodes(title, player): def all_episodes(title):
watch_prompt(title, "all") watch_prompt(title, "all")
os.system('anime dl "' + title + '" --episodes 1: --play ' + player) if not download:
os.system('anime dl "' + title + '" --episodes 1: --play ' + player)
else:
os.system('anime dl "' + title + '" --episodes 1:')
# watch from custom range # watch from custom range
def custom_episode_range(title, player): def custom_episode_range(title):
begginig = color_prommpt("Beggining of interval?: ") begginig = color_prommpt("Beggining of interval?: ")
end = color_prommpt("End of interval?: ") end = color_prommpt("End of interval?: ")
watch_prompt(title, begginig + " to " + end) watch_prompt(title, begginig + " to " + end)
os.system('anime dl "' + title + '" --episodes ' + begginig + ':' + end +' --play ' + player) if not download:
os.system('anime dl "' + title + '" --episodes ' + begginig + ':' + end +' --play ' + player)
else:
os.system('anime dl "' + title + '" --episodes ' + begginig + ':' + end)
# add to last watched m # add to last watched m
def next_plus_n(title, episode, player, action): def next_plus_n(title, episode, action):
watch_prompt(title, str(episode + int(action))) watch_prompt(title, str(episode + int(action)))
os.system('anime dl "' + title + '" --episodes ' + str(episode + int(action)) + ' --play ' + player) if not download:
os.system('anime dl "' + title + '" --episodes ' + str(episode + int(action)) + ' --play ' + player)
else:
os.system('anime dl "' + title + '" --episodes ' + str(episode + int(action)))
# rewatch current episode # rewatch current episode
def rewatch_episode(title, episode, player): def rewatch_episode(title, episode):
watch_prompt(title, str(episode)) watch_prompt(title, str(episode))
os.system('anime dl "' + title + '" --episodes ' + str(episode) + ' --play ' + player) if not download:
os.system('anime dl "' + title + '" --episodes ' + str(episode) + ' --play ' + player)
else:
os.system('anime dl "' + title + '" --episodes ' + str(episode))
# watch custom episode # watch custom episode
def custom_episode(title, player): def custom_episode(title):
episode = color_prommpt("Enter custom episode: ") episode = color_prommpt("Enter custom episode: ")
watch_prompt(title, episode) watch_prompt(title, episode)
os.system('anime dl "' + title + '" --episodes ' + episode + ' --play ' + player) if not download:
os.system('anime dl "' + title + '" --episodes ' + episode + ' --play ' + player)
else:
os.system('anime dl "' + title + '" --episodes ' + episode)
# update title # update title
def update_title(title, episode): def update_title(title, episode):
@ -203,11 +253,11 @@ def choose_episode():
while True: while True:
# retrieving the list on start # retrieving the list on start
if retrieve: if retrieve:
retrieve_list(account) retrieve_list()
retrieve = False retrieve = False
# get the list of anime # get the list of anime
alist = load_list(account) alist = load_list()
# choose an anime from the list # choose an anime from the list
choice = iterfzf(iter_list(alist)) choice = iterfzf(iter_list(alist))
@ -229,27 +279,27 @@ while True:
action = choose_episode() action = choose_episode()
# watch next episode # watch next episode
if action == "": if action == "":
next_episode(title, episode, player) next_episode(title, episode)
wanna_update_title_after_watch(title, episode, score) wanna_update_title_after_watch(title, episode, score)
break break
# same # same
elif action == "n" or action == "N": elif action == "n" or action == "N":
next_episode(title, episode, player) next_episode(title, episode)
wanna_update_title_after_watch(title, episode, score) wanna_update_title_after_watch(title, episode, score)
break break
# watch all left episodes # watch all left episodes
elif action == "l" or action == "L": elif action == "l" or action == "L":
all_from_last(title, episode,last_episode, player) all_from_last(title, episode,last_episode)
wanna_update_title_after_watch(title, episode, score) wanna_update_title_after_watch(title, episode, score)
break break
# watch every episode available # watch every episode available
elif action == "a" or action == "A": elif action == "a" or action == "A":
all_episodes(title, player) all_episodes(title)
wanna_update_title_after_watch(title, episode, score) wanna_update_title_after_watch(title, episode, score)
break break
# custom range of episodes # custom range of episodes
elif action == "i" or action == "I": elif action == "i" or action == "I":
custom_episode_range(title, player) custom_episode_range(title)
if wanna_continu_watch(): if wanna_continu_watch():
continue continue
else: else:
@ -257,7 +307,7 @@ while True:
break break
# something? # something?
elif action == "1" or action == "2" or action == "3" or action == "4" or action == "5" or action == "6" or action == "7" or action == "8" or action == "9": elif action == "1" or action == "2" or action == "3" or action == "4" or action == "5" or action == "6" or action == "7" or action == "8" or action == "9":
next_plus_n(title, episode, player, action) next_plus_n(title, episode, action)
if wanna_continu_watch(): if wanna_continu_watch():
continue continue
else: else:
@ -265,7 +315,7 @@ while True:
break break
# rewatch current episode # rewatch current episode
elif action == "r" or action == "R": elif action == "r" or action == "R":
rewatch_episode(title, episode, player) rewatch_episode(title, episode)
if wanna_continu_watch(): if wanna_continu_watch():
continue continue
else: else:
@ -273,7 +323,7 @@ while True:
break break
# watch custom episode # watch custom episode
elif action == "c" or action == "C": elif action == "c" or action == "C":
custom_episode(title, player) custom_episode(title)
if wanna_continu_watch(): if wanna_continu_watch():
continue continue
else: else: