diff --git a/tdarr_inform.py b/tdarr_inform.py index 1b26e02..e4e2fbc 100644 --- a/tdarr_inform.py +++ b/tdarr_inform.py @@ -44,11 +44,12 @@ def main(): logwrite = open(log_file, 'a') if "sonarr_eventtype" in os.environ: - logwrite.write("Recieved Event from Sonarr\n") - ENV_FILE = os.path.join(SCRIPT_DIR, "env_sonarr") + arr_type = "sonarr" elif "radarr_eventtype" in os.environ: - logwrite.write("Recieved Event from Sonarr\n") - ENV_FILE = os.path.join(SCRIPT_DIR, "env_radarr") + arr_type = "radarr" + logwrite.write("Recieved Event from %s\n" % arr_type) + + ENV_FILE = os.path.join(SCRIPT_DIR, "env_%s" % arr_type) envs = {}