Changing from iterfzf to system installed fzf using a helper script
This commit is contained in:
parent
73a9528f0a
commit
29a4dd3c8d
@ -1,9 +1,13 @@
|
|||||||
import os, subprocess, sys, argparse
|
import os, subprocess, sys, argparse
|
||||||
from iterfzf import iterfzf
|
|
||||||
from time import sleep
|
|
||||||
|
|
||||||
# global variables
|
|
||||||
def main():
|
def main():
|
||||||
|
# exit function
|
||||||
|
def exit_adl():
|
||||||
|
fzf_file.close()
|
||||||
|
os.remove(fzf_file_path)
|
||||||
|
sys.exit()
|
||||||
|
|
||||||
|
# global variables
|
||||||
dn = os.path.dirname(os.path.realpath(__file__))
|
dn = os.path.dirname(os.path.realpath(__file__))
|
||||||
account = "0" # trackma account
|
account = "0" # trackma account
|
||||||
retrieve = True # retrieve new list
|
retrieve = True # retrieve new list
|
||||||
@ -12,6 +16,9 @@ def main():
|
|||||||
msg = "watching" # msg for the watch prompt
|
msg = "watching" # msg for the watch prompt
|
||||||
good_title = open(dn + "/good_title.txt").readlines() # the list of good titles
|
good_title = open(dn + "/good_title.txt").readlines() # the list of good titles
|
||||||
problematic_titles = open(dn + "/problem_title.txt").readlines() # list of problematic titles
|
problematic_titles = open(dn + "/problem_title.txt").readlines() # list of problematic titles
|
||||||
|
fzf_file = open(dn + "/fzf.txt", "w+")
|
||||||
|
fzf_file_path = dn +"/fzf.txt"
|
||||||
|
print_fzf_path = "python " + dn + "/print_fzf.py"
|
||||||
|
|
||||||
ap = argparse.ArgumentParser()
|
ap = argparse.ArgumentParser()
|
||||||
|
|
||||||
@ -38,7 +45,7 @@ def main():
|
|||||||
|
|
||||||
if args["version"]:
|
if args["version"]:
|
||||||
print("Py-adl version 0.1")
|
print("Py-adl version 0.1")
|
||||||
sys.exit()
|
exit_adl()
|
||||||
|
|
||||||
# colored print
|
# colored print
|
||||||
def color_print(text):
|
def color_print(text):
|
||||||
@ -52,12 +59,6 @@ def main():
|
|||||||
def color_prommpt(text):
|
def color_prommpt(text):
|
||||||
return input("\033[0;34m" + text + "\033[0m")
|
return input("\033[0;34m" + text + "\033[0m")
|
||||||
|
|
||||||
# iter the list
|
|
||||||
def iter_list(alist):
|
|
||||||
for anime in alist:
|
|
||||||
yield anime.strip()
|
|
||||||
sleep(0.01)
|
|
||||||
|
|
||||||
# retrieve new list
|
# retrieve new list
|
||||||
def retrieve_list():
|
def retrieve_list():
|
||||||
color_print ("Running trackma retrieve for account " + account + "...")
|
color_print ("Running trackma retrieve for account " + account + "...")
|
||||||
@ -77,15 +78,22 @@ def main():
|
|||||||
alist.pop()
|
alist.pop()
|
||||||
return alist
|
return alist
|
||||||
|
|
||||||
|
# write list to a file
|
||||||
|
def list2file(list, file):
|
||||||
|
for line in list:
|
||||||
|
file.write(line)
|
||||||
|
if not list.index(line) == len(list) - 1 :
|
||||||
|
file.write("\n")
|
||||||
|
|
||||||
# exit prompt
|
# exit prompt
|
||||||
def exit_ask():
|
def exit_ask():
|
||||||
while True:
|
while True:
|
||||||
os.system("cls")
|
os.system("cls")
|
||||||
choice = color_prommpt("Want to watch another anime? [Y/n]: ")
|
choice = color_prommpt("Want to watch another anime? [Y/n]: ")
|
||||||
if choice == "":
|
if choice == "":
|
||||||
sys.exit()
|
exit_adl()
|
||||||
elif choice == "N" or choice == "n":
|
elif choice == "N" or choice == "n":
|
||||||
sys.exit()
|
exit_adl()
|
||||||
elif choice == "Y" or choice == "y":
|
elif choice == "Y" or choice == "y":
|
||||||
return
|
return
|
||||||
|
|
||||||
@ -275,20 +283,24 @@ def main():
|
|||||||
|
|
||||||
# get the list of anime
|
# get the list of anime
|
||||||
alist = load_list()
|
alist = load_list()
|
||||||
# choose an anime from the list
|
|
||||||
choice = iterfzf(iter_list(alist))
|
# write list to file
|
||||||
|
list2file(alist, fzf_file)
|
||||||
|
|
||||||
|
# reload file (I Guess ??)
|
||||||
|
fzf_file.seek(0)
|
||||||
|
|
||||||
|
choice = subprocess.getoutput(print_fzf_path + ' | fzf --ansi --reverse --prompt "Choose anime to watch: "')
|
||||||
|
|
||||||
if choice:
|
if choice:
|
||||||
# get the whole choice into a string
|
|
||||||
full_choice = "".join(choice)
|
|
||||||
# get the title
|
# get the title
|
||||||
title = get_title(full_choice)
|
title = get_title(choice)
|
||||||
# get current episode
|
# get current episode
|
||||||
episode = get_episode(full_choice)
|
episode = get_episode(choice)
|
||||||
# get latest episode
|
# get latest episode
|
||||||
last_episode = get_all_episodes(full_choice)
|
last_episode = get_all_episodes(choice)
|
||||||
# get current score
|
# get current score
|
||||||
score = get_score(full_choice)
|
score = get_score(choice)
|
||||||
|
|
||||||
# the watch loop
|
# the watch loop
|
||||||
while True:
|
while True:
|
||||||
|
Loading…
Reference in New Issue
Block a user