github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-08-19 21:03:14 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extract download_id parsing from core.processor.nzbget.process -> parse_download_id

Browse source
This commit is contained in:
Labrys of Knossos 2022-11-25 09:51:04 -05:00
commit a2b2e4f620
1 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

29
core/processor/nzbget.py
View file

@ -6,6 +6,22 @@ from core import logger
from core.processor import nzb from core.processor import nzb
def parse_download_id():
# Check for download_id to pass to CouchPotato
download_id = ''
if 'NZBPR_COUCHPOTATO' in os.environ:
download_id = os.environ['NZBPR_COUCHPOTATO']
elif 'NZBPR_DRONE' in os.environ:
download_id = os.environ['NZBPR_DRONE']
elif 'NZBPR_SONARR' in os.environ:
download_id = os.environ['NZBPR_SONARR']
elif 'NZBPR_RADARR' in os.environ:
download_id = os.environ['NZBPR_RADARR']
elif 'NZBPR_LIDARR' in os.environ:
download_id = os.environ['NZBPR_LIDARR']
return download_id
def process(): def process():
# Check if the script is called from nzbget 11.0 or later # Check if the script is called from nzbget 11.0 or later
if os.environ['NZBOP_VERSION'][0:5] < '11.0': if os.environ['NZBOP_VERSION'][0:5] < '11.0':
@ -54,19 +70,8 @@ def process():
logger.info( logger.info(
'Please check your Par-check/repair settings for future downloads.') 'Please check your Par-check/repair settings for future downloads.')
# Check for download_id to pass to CouchPotato download_id = parse_download_id()
download_id = ''
failure_link = None failure_link = None
if 'NZBPR_COUCHPOTATO' in os.environ:
download_id = os.environ['NZBPR_COUCHPOTATO']
elif 'NZBPR_DRONE' in os.environ:
download_id = os.environ['NZBPR_DRONE']
elif 'NZBPR_SONARR' in os.environ:
download_id = os.environ['NZBPR_SONARR']
elif 'NZBPR_RADARR' in os.environ:
download_id = os.environ['NZBPR_RADARR']
elif 'NZBPR_LIDARR' in os.environ:
download_id = os.environ['NZBPR_LIDARR']
if 'NZBPR__DNZB_FAILURE' in os.environ: if 'NZBPR__DNZB_FAILURE' in os.environ:
failure_link = os.environ['NZBPR__DNZB_FAILURE'] failure_link = os.environ['NZBPR__DNZB_FAILURE']