def test_linux_features(self, mock_subproc_popen, userxbit):
# Should break on no-execute permissions
userxbit.return_value = False
with pytest.raises(IOError):
misc.build_and_run_command([self.script_path, "input 1"])
userxbit.return_value = True
# Check if python-call is added if not supplied by shebang
temp_file_fd, temp_file_path = tempfile.mkstemp(suffix=".py")
os.close(temp_file_fd)
misc.build_and_run_command([temp_file_path, "input 1"])
assert mock_subproc_popen.call_args[0][0] == ["python", temp_file_path, "input 1"]
os.remove(temp_file_path)
# Have to fake these for it to work
newsunpack.IONICE_COMMAND = "ionice"
newsunpack.NICE_COMMAND = "nice"
userxbit.return_value = True
misc.build_and_run_command([self.script_path, "input 1"])
assert mock_subproc_popen.call_args[0][0] == [
"nice",
"--adjustment=-7",
"ionice",
"-t",
"-n9",
"-c7",
self.script_path,
"input 1",
]
def test_win(self, mock_subproc_popen):
# Needed for priority and startupinfo check
import win32process
import win32con
misc.build_and_run_command(["test.cmd", "input 1"])
assert mock_subproc_popen.call_args[0][0] == ["test.cmd", "input 1"]
assert mock_subproc_popen.call_args[1]["creationflags"] == win32process.NORMAL_PRIORITY_CLASS
assert mock_subproc_popen.call_args[1]["startupinfo"].dwFlags == win32process.STARTF_USESHOWWINDOW
assert mock_subproc_popen.call_args[1]["startupinfo"].wShowWindow == win32con.SW_HIDE
misc.build_and_run_command(["test.py", "input 1"])
assert mock_subproc_popen.call_args[0][0] == ["python.exe", "test.py", "input 1"]
assert mock_subproc_popen.call_args[1]["creationflags"] == win32process.NORMAL_PRIORITY_CLASS
assert mock_subproc_popen.call_args[1]["startupinfo"].dwFlags == win32process.STARTF_USESHOWWINDOW
assert mock_subproc_popen.call_args[1]["startupinfo"].wShowWindow == win32con.SW_HIDE
# See: https://github.com/sabnzbd/sabnzbd/issues/1043
misc.build_and_run_command(["UnRar.exe", "\\\\?\\C:\\path\\"])
assert mock_subproc_popen.call_args[0][0] == ["UnRar.exe", "\\\\?\\C:\\path\\"]
misc.build_and_run_command(["UnRar.exe", "\\\\?\\C:\\path\\"], flatten_command=True)
assert mock_subproc_popen.call_args[0][0] == '"UnRar.exe" "\\\\?\\C:\\path\\"'
def send_nscript(title, msg, gtype, force=False, test=None):
""" Run user's notification script """
if test:
script = test.get("nscript_script")
nscript_parameters = test.get("nscript_parameters")
else:
script = sabnzbd.cfg.nscript_script()
nscript_parameters = sabnzbd.cfg.nscript_parameters()
nscript_parameters = nscript_parameters.split()
if not script:
return T("Cannot send, missing required data")
title = "SABnzbd: " + T(NOTIFICATION.get(gtype, "other"))
if force or check_classes(gtype, "nscript"):
script_path = make_script_path(script)
if script_path:
ret = -1
output = None
try:
p = build_and_run_command([script_path, gtype, title, msg] +
nscript_parameters,
env=create_env())
output = platform_btou(p.stdout.read())
ret = p.wait()
except:
logging.info("Failed script %s, Traceback: ",
script,
exc_info=True)
if ret:
logging.error(
T('Script returned exit code %s and output "%s"'), ret,
output)
return T('Script returned exit code %s and output "%s"') % (
ret, output)
else:
logging.info("Successfully executed notification script %s",
script_path)
else:
return T('Notification script "%s" does not exist') % script_path
return ""
def create_unrar_instance(self):
""" Start the unrar instance using the user's options """
# Generate extraction path and save for post-proc
if not self.unpack_dir_info:
try:
self.unpack_dir_info = prepare_extraction_path(self.nzo)
except:
# Prevent fatal crash if directory creation fails
self.abort()
return
# Get the information
extraction_path, _, _, one_folder, _ = self.unpack_dir_info
# Set options
if self.nzo.password:
password_command = "-p%s" % self.nzo.password
else:
password_command = "-p-"
if one_folder or cfg.flat_unpack():
action = "e"
else:
action = "x"
# The first NZF
self.rarfile_nzf = self.have_next_volume()
# Ignore if maybe this set is not there any more
# This can happen due to race/timing issues when creating the sets
if not self.rarfile_nzf:
return
# Generate command
rarfile_path = os.path.join(self.nzo.downpath,
self.rarfile_nzf.filename)
if sabnzbd.WIN32:
# For Unrar to support long-path, we need to cricumvent Python's list2cmdline
# See: https://github.com/sabnzbd/sabnzbd/issues/1043
command = [
"%s" % sabnzbd.newsunpack.RAR_COMMAND,
action,
"-vp",
"-idp",
"-o+",
"-ai",
password_command,
"%s" % clip_path(rarfile_path),
"%s\\" % long_path(extraction_path),
]
else:
# Don't use "-ai" (not needed for non-Windows)
command = [
"%s" % sabnzbd.newsunpack.RAR_COMMAND,
action,
"-vp",
"-idp",
"-o+",
password_command,
"%s" % rarfile_path,
"%s/" % extraction_path,
]
if cfg.ignore_unrar_dates():
command.insert(3, "-tsm-")
# Let's start from the first one!
self.cur_volume = 1
# Need to disable buffer to have direct feedback
self.active_instance = build_and_run_command(command,
flatten_command=True,
bufsize=0)
# Add to runners
ACTIVE_UNPACKERS.append(self)
# Doing the first
logging.info("DirectUnpacked volume %s for %s", self.cur_volume,
self.cur_setname)
def test_std_override(self, mock_subproc_popen, userxbit):
userxbit.return_value = True
misc.build_and_run_command([self.script_path], stderr=subprocess.DEVNULL)
assert mock_subproc_popen.call_args[1]["stderr"] == subprocess.DEVNULL
def test_none_check(self):
with pytest.raises(IOError):
misc.build_and_run_command([None])