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

Better info snipping

Browse Source
This commit is contained in:
CronyAkatsuki 2022-02-03 18:48:28 +01:00
parent b86f419812
commit 32833aee89
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
adl.py
View File

@ -1,9 +1,10 @@
from base64 import encode
import os, subprocess, sys, argparse, signal import os, subprocess, sys, argparse, signal
# required files # required files
DN = os.path.dirname(os.path.realpath(__file__)) # get current directory of the script DN = os.path.dirname(os.path.realpath(__file__)) # get current directory of the script
GOOD_TITLES = open(DN + "/good_title.txt").readlines() # the list of good titles GOOD_TITLES = open(DN + "/good_title.txt", encoding='utf-8').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", encoding='utf-8').readlines() # list of problematic titles
FZF_FILE = open(DN + "/fzf.txt", "w+") # temp file for fzf FZF_FILE = open(DN + "/fzf.txt", "w+") # temp file for fzf
FZF_FILE_PATH = DN +"/fzf.txt" # path of the temp file FZF_FILE_PATH = DN +"/fzf.txt" # path of the temp file
PRINT_FZF_PATH = "python " + DN + "/print_fzf.py" # print the fzf file PRINT_FZF_PATH = "python " + DN + "/print_fzf.py" # print the fzf file
@ -71,25 +72,26 @@ def exit_ask():
# check for problematic title # check for problematic title
def check_title(title): def check_title(title):
for problem in PROBLEMATIC_TITLES: for problem in PROBLEMATIC_TITLES:
if problem.__contains__(title): if problem == title:
title = GOOD_TITLES[PROBLEMATIC_TITLES.index(problem)] title = GOOD_TITLES[PROBLEMATIC_TITLES.index(problem)].encode('utf-8')
title = title.decode('utf-8')
return title return title
# get your title # get your title
def get_title(choice): def get_title(choice):
choice = choice[9:96] choice = choice[9:-19]
choice = choice.rstrip(".") choice = choice.rstrip(".")
title = check_title(choice) title = check_title(choice)
return title return title
# get episode # get episode
def get_episode(choice): def get_episode(choice):
choice = choice[98:100] choice = choice[-19:-15]
return int(choice) return int(choice)
# get score # get score
def get_score(choice): def get_score(choice):
choice = choice[108:110] choice = choice[-10:-5]
return choice return choice
# watch animes # watch animes
@ -394,7 +396,7 @@ def main_loop(retrieve, account, msg, download,provider, download_location):
FZF_FILE.seek(0) FZF_FILE.seek(0)
# get choice from fzf # get choice from fzf
choice = subprocess.getoutput(f'{PRINT_FZF_PATH} | fzf --ansi --reverse --prompt "Choose anime to watch: "') choice = subprocess.getoutput(f'{PRINT_FZF_PATH} | fzf --ansi --unicode --reverse --prompt "Choose anime to watch: "')
if choice: if choice:
# get the title # get the title
@ -480,7 +482,7 @@ def main():
# setup env variables for better readability of outputs # setup env variables for better readability of outputs
os.environ['LINES'] = '25' os.environ['LINES'] = '25'
os.environ['COLUMNS'] = '120' os.environ['COLUMNS'] = '120'
os.environ['PYTHONIOENCODING'] = 'utf-8'
# get argument and config parameters # get argument and config parameters
(provider, show, episode, account, download, msg, download_location) = argument_and_config_parser() (provider, show, episode, account, download, msg, download_location) = argument_and_config_parser()