import os.path import json import datetime import dateutil import re import requests import urllib.parse from utils import APIConnector from utils import clear_downloads from utils import clear_screen from utils import create_file_name from authentication import check_timestamp from authentication import checkStatus from authentication import refreshToken class Series: def __init__(self, name, folder_name, id, url): self.name = name self.folder_name = folder_name self.id = str(id) self.url = url def search(): try: with open("login.json") as json_data: login = json.load(json_data) # json_data.close() if login["TIMESTAMP"] == "": print("There was an error checking your login. Try logging in again with 'Login/Change login'.") return None else: save_time = dateutil.parser.parse(login["TIMESTAMP"]) cur_time = datetime.datetime.now().replace(tzinfo=None) # TODO use UTC time? if check_timestamp(save_time, cur_time) == False: # Token is expired print("Your token has expired. Attempting to get a new one...") refreshToken() except Exception as ex: # print(ex) print("There was an error checking your login. Try logging in again with 'Login/Change login'.") return None api_con = APIConnector() # authHeaders = { # "Content-Type": "application/json", # "Accept": "application/json", # "Authorization": "Bearer " + login["TOKEN"] # } keyword = input("Enter series name to search: ") # Getting the search name and fixing s_keyword = urllib.parse.quote(keyword) # the url parse mistakes s_keyword = s_keyword.replace("%21", "!") # TODO find a better way of doing this s_keyword = s_keyword.replace("%2A", "*") s_keyword = s_keyword.replace("%28", "(") s_keyword = s_keyword.replace("%29", ")") s_keyword = s_keyword.replace("%27", "'") s_keyword = s_keyword.replace("/", "%2F") s_keyword = s_keyword.replace("%7E", "~") # search_url = "https://api.thetvdb.com/search/series?name={}".format(s_keyword) # response = requests.get(search_url, headers=authHeaders) res = api_con.send_http_req( "https://api.thetvdb.com/search/series?name={}".format(s_keyword)) # if (checkStatus(response, True) == False): # return None search_results = json.loads(res.content) title = -1 print() clear_screen() while title < 0 or title > len(search_results["data"]) - 1: # Looping until the user chooses print("Results:") # a series from the printed list count = 1 # or they input '0' to cancel for result in search_results["data"]: print("\n{})\nSeries Name: {}".format(str(count), str(result["seriesName"]))) print() desc = result["overview"] desc = str(desc).replace("\r\n\r\n", " ") # Removing format characters print("Description: \n{}".format(desc)) print() count = count + 1 print() #TODO this can crash with non integer inputs #TODO they should also be able to ctrl-c to cancel search title = int(input("Choose one by number or '0' to exit: ")) - 1 # Subtracting 1 so that the print() # index can start from 0 if title < -1 or title > len(search_results["data"]) - 1: print("Error: Choose the number of an item listed. Or '0' to exit.") if (title == -1): # If the user inputs 0 return None print() series = Series(search_results["data"][title]["seriesName"], create_file_name(search_results["data"][title]["seriesName"]), search_results["data"][title]["id"], "https://www.thetvdb.com/series/" + search_results["data"][title]["slug"]) return series