141 lines
4.5 KiB
Python
141 lines
4.5 KiB
Python
#!/usr/bin/python3
|
|
|
|
import os
|
|
import sys
|
|
import requests
|
|
|
|
SCRIPT_PATH = os.path.realpath(__file__)
|
|
SCRIPT_DIR = os.path.dirname(SCRIPT_PATH)
|
|
|
|
script_settings = {
|
|
"tdarr": {
|
|
"url": "http://tdarr-server1.ipa.dbb:8265",
|
|
},
|
|
"logging": {
|
|
"log_file": True,
|
|
"log_file_path": os.path.join(SCRIPT_DIR, "debug.log"),
|
|
"output_level": 1
|
|
}
|
|
}
|
|
|
|
|
|
def loggit(logtext, force_err=False):
|
|
if script_settings["logging"]["log_file"]:
|
|
with open(script_settings["logging"]["log_file_path"], 'a') as logwrite:
|
|
logwrite.write(logtext)
|
|
if script_settings["logging"]["output_level"] or force_err:
|
|
sys.stderr.write(logtext)
|
|
else:
|
|
sys.stdout.write(logtext)
|
|
|
|
|
|
def do_tdarr_inform(dbID, arr_file_path):
|
|
headers = {"content-type": "application/json"}
|
|
payload = {
|
|
"data": {
|
|
"scanConfig": {
|
|
"dbID": dbID,
|
|
"arrayOrPath": [
|
|
arr_file_path
|
|
],
|
|
"mode": "scanFolderWatcher"
|
|
}
|
|
}
|
|
}
|
|
response = requests.post("%s/api/v2/scan-files" % script_settings["tdarr"]["url"], json=payload, headers=headers)
|
|
loggit("Tdarr response: %s\n" % response)
|
|
|
|
|
|
def do_file_search(arr_file_path):
|
|
payload = {
|
|
"data": {
|
|
"string": arr_file_path,
|
|
"lessThanGB": 9999,
|
|
"greaterThanGB": 0
|
|
}
|
|
}
|
|
headers = {"content-type": "application/json"}
|
|
response = requests.post("%s/api/v2/search-db" % script_settings["tdarr"]["url"], json=payload, headers=headers)
|
|
loggit("Tdarr response: %s\n" % response.text)
|
|
try:
|
|
response_json = response.json()
|
|
except Exception:
|
|
return None
|
|
if not len(response_json):
|
|
return None
|
|
dbIDs = [x["DB"] for x in response_json if "DB" in list(x.keys())]
|
|
dbIDs = list(set(dbIDs))
|
|
if len(dbIDs) > 1:
|
|
return None
|
|
return dbIDs[0]
|
|
|
|
|
|
def do_reverse_recursive_directory_search(arr_file_path):
|
|
dbID = None
|
|
arr_dir_path = os.path.dirname(arr_file_path)
|
|
while arr_dir_path != os.path.abspath(os.sep) and not arr_dir_path.endswith(":\\"):
|
|
arr_dir_path = os.path.dirname(arr_dir_path)
|
|
dbID = do_file_search(arr_dir_path)
|
|
if dbID:
|
|
break
|
|
return dbID
|
|
|
|
|
|
def main():
|
|
|
|
if "sonarr_eventtype" in os.environ:
|
|
arr_type = "sonarr"
|
|
arr_event_type = os.environ["sonarr_eventtype"]
|
|
arr_file_path_key = "sonarr_episodefile_path"
|
|
elif "radarr_eventtype" in os.environ:
|
|
arr_type = "radarr"
|
|
arr_event_type = os.environ["radarr_eventtype"]
|
|
arr_file_path_key = "radarr_moviefile_path"
|
|
else:
|
|
# loggit("Could not Detect Radarr or Sonarr", True)
|
|
# sys.exit(0)
|
|
|
|
arr_type = "sonarr"
|
|
arr_event_type = "Download"
|
|
arr_file_path_key = "sonarr_episodefile_path"
|
|
print(arr_file_path_key)
|
|
|
|
loggit("tdarr_inform Recieved %s Event from %s\n" % (arr_event_type, arr_type))
|
|
|
|
if arr_event_type == "Test":
|
|
sys.exit(0)
|
|
|
|
elif arr_event_type not in ["Download", "Rename"]:
|
|
# TODO rename sonarr_series_path
|
|
# radarr_moviefile_paths split |
|
|
# radarr_moviefile_previouspaths
|
|
|
|
# TODO Delete
|
|
# sonarr_series_path
|
|
# radarr_deletedpaths
|
|
loggit("%s is not a supported tdarr_inform Event." % arr_event_type, True)
|
|
sys.exit(0)
|
|
|
|
# if arr_file_path_key not in os.environ:
|
|
# loggit("%s Environment variable was missing." % arr_file_path_key, True)
|
|
# sys.exit(1)
|
|
# arr_file_path = os.environ[arr_file_path_key]
|
|
arr_file_path = "/Drivepool/Media/TV/Cartoons/Ben 10 (2016)/Season 2/Ben 10 (2016) - S02E31 - Chicken Nuggets of Wisdom.mkv"
|
|
loggit("Event Item: %s\n" % arr_file_path)
|
|
|
|
loggit("Searching tdarr API for item's library ID\n")
|
|
dbID = do_file_search(arr_file_path)
|
|
if not dbID:
|
|
loggit("No exact match found, searching for library ID from Reverse Recursive Directory matching\n")
|
|
dbID = do_reverse_recursive_directory_search(arr_file_path)
|
|
if not dbID:
|
|
loggit("No match found\n")
|
|
sys.exit(0)
|
|
|
|
loggit("Preparing payload to POST to tdarr API\n")
|
|
# do_tdarr_inform(dbID, arr_file_path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|