Scraper-for-theTVDB.com/utils.py

54 lines
1.6 KiB
Python

import json
import os
import requests
class APIConnector:
def __init__(self):
with open("login.json", "r") as f:
self.login = json.load(f)
self.auth_headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": "Bearer " + self.login["TOKEN"]
}
def reload_login(self):
with open("login.json", "r") as f:
self.login = json.load(f)
self.auth_headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": "Bearer " + self.login["TOKEN"]
}
def send_http_req(self, api_path):
return requests.get(api_path, headers=self.auth_headers)
def clearFolders(): # TODO implement this
folders = ["banner", "fanart", "poster"]
del_count = 0
for folder in folders:
if os.path.exists(folder):
image_list = os.listdir(folder)
if len(image_list) != 0:
print("Clearing " + folder + "/")
for x in image_list: # TODO check if folder is empty
print("Deleting {}/{}".format(folder, x))
del_path = os.path.join(folder + "\\" + x)
os.remove(del_path)
del_count += 1
print()
else:
print("'{}' is already empty".format(folder))
else:
createFolder(folder)
print("Deleted {} images.\n".format(del_count))
def clearScreen():
IS_WINDOWS = os.name == "nt"
if IS_WINDOWS:
os.system("cls")
else:
os.system("clear")