You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.5 KiB
68 lines
2.5 KiB
#!/usr/bin/python -OO
|
|
# Copyright 2009-2017 The SABnzbd-Team <team@sabnzbd.org>
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
"""
|
|
sabnzbd.utils.upload - File association functions for adding nzb files to sabnzbd
|
|
"""
|
|
|
|
import urllib
|
|
import logging
|
|
import os
|
|
from sabnzbd.encoding import unicoder
|
|
import sabnzbd.cfg as cfg
|
|
from sabnzbd.misc import get_ext, get_filename
|
|
import sabnzbd.newsunpack
|
|
from sabnzbd.constants import VALID_ARCHIVES
|
|
|
|
from sabnzbd.dirscanner import ProcessArchiveFile, ProcessSingleFile
|
|
|
|
|
|
def upload_file(url, fp):
|
|
""" Function for uploading nzbs to a running sabnzbd instance """
|
|
try:
|
|
fp = unicoder(fp).encode('utf-8')
|
|
fp = urllib.quote_plus(fp)
|
|
url = '%s&mode=addlocalfile&name=%s' % (url, fp)
|
|
# Add local apikey if it wasn't already in the registered URL
|
|
apikey = cfg.api_key()
|
|
if apikey and 'apikey' not in url:
|
|
url = '%s&apikey=%s' % (url, apikey)
|
|
if 'apikey' not in url:
|
|
# Use alternative login method
|
|
username = cfg.username()
|
|
password = cfg.password()
|
|
if username and password:
|
|
url = '%s&ma_username=%s&ma_password=%s' % (url, username, password)
|
|
sabnzbd.newsunpack.get_from_url(url)
|
|
except:
|
|
logging.error("Failed to upload file: %s", fp)
|
|
logging.info("Traceback: ", exc_info=True)
|
|
|
|
|
|
def add_local(f):
|
|
""" Function for easily adding nzb/zip/rar/nzb.gz to sabnzbd """
|
|
if os.path.exists(f):
|
|
fn = get_filename(f)
|
|
if fn:
|
|
if get_ext(fn) in VALID_ARCHIVES:
|
|
ProcessArchiveFile(fn, f, keep=True)
|
|
elif get_ext(fn) in ('.nzb', '.gz', '.bz2'):
|
|
ProcessSingleFile(fn, f, keep=True)
|
|
else:
|
|
logging.error("Filename not found: %s", f)
|
|
else:
|
|
logging.error("File not found: %s", f)
|
|
|