def test_is_new_version_checker_enabled(self, include_prerelease, enabled, interval):
with patch('monitorrent.settings_manager.SettingsManager.new_version_check_interval',
new_callable=PropertyMock) as new_version_check_interval_mock:
new_version_check_interval_mock.return_value = interval
settings_manager = SettingsManager()
get_is_new_version_checker_enabled_mock = Mock(return_value=enabled)
get_new_version_check_include_prerelease_mock = Mock(return_value=include_prerelease)
settings_manager.get_is_new_version_checker_enabled = get_is_new_version_checker_enabled_mock
settings_manager.get_new_version_check_include_prerelease = get_new_version_check_include_prerelease_mock
new_version_checker = NewVersionChecker(False)
new_version_checker.execute = Mock()
settings_new_version_checker = SettingsNewVersionChecker(settings_manager, new_version_checker)
self.api.add_route('/api/settings/new_version_checker', settings_new_version_checker)
body = self.simulate_request("/api/settings/new_version_checker", decode='utf-8')
self.assertEqual(self.srmock.status, falcon.HTTP_OK)
self.assertTrue('application/json' in self.srmock.headers_dict['Content-Type'])
result = json.loads(body)
expected = {'include_prerelease': include_prerelease, 'enabled': enabled, 'interval': interval}
self.assertEqual(result, expected)
get_is_new_version_checker_enabled_mock.assert_called_once_with()
get_new_version_check_include_prerelease_mock.assert_called_once_with()
new_version_check_interval_mock.assert_called_once_with()
def test_patch_test(self, patch_include_prerelease, patch_enabled, patch_interval, update_include_prerelease, update_enabled, update_interval):
with patch('monitorrent.settings_manager.SettingsManager.new_version_check_interval',
new_callable=PropertyMock) as new_version_check_interval_mock:
new_version_check_interval_mock.return_value = 4000
settings_manager = SettingsManager()
settings_manager.get_new_version_check_include_prerelease = Mock(return_value=False)
settings_manager.set_new_version_check_include_prerelease = Mock()
settings_manager.get_is_new_version_checker_enabled = Mock(return_value=True)
settings_manager.set_is_new_version_checker_enabled = Mock()
settings_manager.new_version_check_interval = PropertyMock(return_value=3600)
new_version_checker = NewVersionChecker(False)
update_mock = Mock()
new_version_checker.update = update_mock
settings_new_version_checker = SettingsNewVersionChecker(settings_manager, new_version_checker)
self.api.add_route('/api/settings/new_version_checker', settings_new_version_checker)
request = dict()
if patch_include_prerelease is not None:
request['include_prerelease'] = patch_include_prerelease
if patch_enabled is not None:
request['enabled'] = patch_enabled
if patch_interval is not None:
request['interval'] = patch_interval
self.simulate_request('/api/settings/new_version_checker', method="PATCH", body=json.dumps(request))
self.assertEqual(self.srmock.status, falcon.HTTP_NO_CONTENT)
update_mock.assert_called_once_with(update_include_prerelease, update_enabled, update_interval)
def test_bad_request(self, body):
with patch('monitorrent.settings_manager.SettingsManager.new_version_check_interval',
new_callable=PropertyMock) as new_version_check_interval_mock:
new_version_check_interval_mock.return_value = 4000
settings_manager = SettingsManager()
get_is_new_version_checker_enabled = Mock()
settings_manager.get_is_new_version_checker_enabled = get_is_new_version_checker_enabled
new_version_checker = NewVersionChecker(False)
new_version_checker.execute = Mock()
settings_new_version_checker = SettingsNewVersionChecker(settings_manager, new_version_checker)
self.api.add_route('/api/settings/new_version_checker', settings_new_version_checker)
self.simulate_request("/api/settings/new_version_checker", method="PATCH", body=json.dumps(body) if body else None)
self.assertEqual(self.srmock.status, falcon.HTTP_BAD_REQUEST)
def test_patch_test(self, patch_include_prerelease, patch_enabled,
patch_interval, update_include_prerelease,
update_enabled, update_interval):
with patch(
'monitorrent.settings_manager.SettingsManager.new_version_check_interval',
new_callable=PropertyMock) as new_version_check_interval_mock:
new_version_check_interval_mock.return_value = 4000
settings_manager = SettingsManager()
settings_manager.get_new_version_check_include_prerelease = Mock(
return_value=False)
settings_manager.set_new_version_check_include_prerelease = Mock()
settings_manager.get_is_new_version_checker_enabled = Mock(
return_value=True)
settings_manager.set_is_new_version_checker_enabled = Mock()
settings_manager.new_version_check_interval = PropertyMock(
return_value=3600)
new_version_checker = NewVersionChecker(Mock(), False)
update_mock = Mock()
new_version_checker.update = update_mock
settings_new_version_checker = SettingsNewVersionChecker(
settings_manager, new_version_checker)
self.api.add_route('/api/settings/new_version_checker',
settings_new_version_checker)
request = dict()
if patch_include_prerelease is not None:
request['include_prerelease'] = patch_include_prerelease
if patch_enabled is not None:
request['enabled'] = patch_enabled
if patch_interval is not None:
request['interval'] = patch_interval
self.simulate_request('/api/settings/new_version_checker',
method="PATCH",
body=json.dumps(request))
self.assertEqual(self.srmock.status, falcon.HTTP_NO_CONTENT)
update_mock.assert_called_once_with(update_include_prerelease,
update_enabled,
update_interval)
def test_is_new_version_checker_enabled(self, include_prerelease, enabled,
interval):
with patch(
'monitorrent.settings_manager.SettingsManager.new_version_check_interval',
new_callable=PropertyMock) as new_version_check_interval_mock:
new_version_check_interval_mock.return_value = interval
settings_manager = SettingsManager()
get_is_new_version_checker_enabled_mock = Mock(
return_value=enabled)
get_new_version_check_include_prerelease_mock = Mock(
return_value=include_prerelease)
settings_manager.get_is_new_version_checker_enabled = get_is_new_version_checker_enabled_mock
settings_manager.get_new_version_check_include_prerelease = get_new_version_check_include_prerelease_mock
new_version_checker = NewVersionChecker(Mock(), False)
new_version_checker.execute = Mock()
settings_new_version_checker = SettingsNewVersionChecker(
settings_manager, new_version_checker)
self.api.add_route('/api/settings/new_version_checker',
settings_new_version_checker)
body = self.simulate_request("/api/settings/new_version_checker",
decode='utf-8')
self.assertEqual(self.srmock.status, falcon.HTTP_OK)
self.assertTrue(
'application/json' in self.srmock.headers_dict['Content-Type'])
result = json.loads(body)
expected = {
'include_prerelease': include_prerelease,
'enabled': enabled,
'interval': interval
}
self.assertEqual(result, expected)
get_is_new_version_checker_enabled_mock.assert_called_once_with()
get_new_version_check_include_prerelease_mock.assert_called_once_with(
)
new_version_check_interval_mock.assert_called_once_with()
def test_bad_request(self, body):
with patch(
'monitorrent.settings_manager.SettingsManager.new_version_check_interval',
new_callable=PropertyMock) as new_version_check_interval_mock:
new_version_check_interval_mock.return_value = 4000
settings_manager = SettingsManager()
get_is_new_version_checker_enabled = Mock()
settings_manager.get_is_new_version_checker_enabled = get_is_new_version_checker_enabled
new_version_checker = NewVersionChecker(Mock(), False)
new_version_checker.execute = Mock()
settings_new_version_checker = SettingsNewVersionChecker(
settings_manager, new_version_checker)
self.api.add_route('/api/settings/new_version_checker',
settings_new_version_checker)
self.simulate_request("/api/settings/new_version_checker",
method="PATCH",
body=json.dumps(body) if body else None)
self.assertEqual(self.srmock.status, falcon.HTTP_BAD_REQUEST)
def main():
def try_int(s, base=10, val=None):
if s is None:
return None
try:
return int(s, base)
except ValueError:
return val
class Config(object):
debug = False
ip = '0.0.0.0'
port = 6687
db_path = 'monitorrent.db'
config = 'config.py'
def __init__(self, parsed_args):
if parsed_args.config is not None and not os.path.isfile(parsed_args.config):
warnings.warn('File not found: {}'.format(parsed_args.config))
config_path = parsed_args.config or self.config
if os.path.isfile(config_path):
# noinspection PyBroadException
try:
parsed_config = {}
with open(config_path) as config_file:
six.exec_(compile(config_file.read(), config_path, 'exec'), {}, parsed_config)
self.debug = parsed_config.get('debug', self.debug)
self.ip = parsed_config.get('ip', self.ip)
self.port = parsed_config.get('port', self.port)
self.db_path = parsed_config.get('db_path', self.db_path)
except:
ex, val, tb = sys.exc_info()
warnings.warn('Error reading: {0}: {1} ({2}'.format(parsed_args.config, ex, val))
env_debug = (os.environ.get('MONITORRENT_DEBUG', None) in ['true', 'True', '1'])
self.debug = parsed_args.debug or env_debug or self.debug
self.ip = parsed_args.ip or os.environ.get('MONITORRENT_IP', None) or self.ip
self.port = parsed_args.port or try_int(os.environ.get('MONITORRENT_PORT', None)) or self.port
self.db_path = parsed_args.db_path or os.environ.get('MONITORRENT_DB_PATH', None) or self.db_path
parser = argparse.ArgumentParser(description='Monitorrent server')
parser.add_argument('--debug', action='store_true',
help='Run in debug mode. Secret key is always the same.')
parser.add_argument('--ip', type=str, dest='ip',
help='Bind interface. Default is {0}'.format(Config.ip))
parser.add_argument('--port', type=int, dest='port',
help='Port for server. Default is {0}'.format(Config.port))
parser.add_argument('--db-path', type=str, dest='db_path',
help='Path to SQL lite database. Default is to {0}'.format(Config.db_path))
parser.add_argument('--config', type=str, dest='config',
default=os.environ.get('MONITORRENT_CONFIG', None),
help='Path to config file (default {0})'.format(Config.config))
parsed_args = parser.parse_args()
config = Config(parsed_args)
db_connection_string = "sqlite:///" + config.db_path
init_db_engine(db_connection_string, False)
load_plugins()
upgrade()
create_db()
settings_manager = SettingsManager()
tracker_manager = TrackersManager(settings_manager, get_plugins('tracker'))
clients_manager = DbClientsManager(settings_manager, get_plugins('client'))
notifier_manager = NotifierManager(settings_manager, get_plugins('notifier'))
log_manager = ExecuteLogManager()
engine_runner_logger = DbLoggerWrapper(log_manager, settings_manager)
engine_runner = DBEngineRunner(engine_runner_logger, settings_manager, tracker_manager,
clients_manager, notifier_manager)
include_prerelease = settings_manager.get_new_version_check_include_prerelease()
new_version_checker = NewVersionChecker(notifier_manager, include_prerelease)
if settings_manager.get_is_new_version_checker_enabled():
# noinspection PyBroadException
try:
new_version_checker.execute()
except:
pass
new_version_checker.start(settings_manager.new_version_check_interval)
debug = config.debug
if debug:
secret_key = 'Secret!'
token = 'monitorrent'
else:
secret_key = os.urandom(24)
token = ''.join(random.choice(string.ascii_letters) for _ in range(8))
app = create_app(secret_key, token, tracker_manager, clients_manager, notifier_manager, settings_manager,
engine_runner, log_manager, new_version_checker)
server_start_params = (config.ip, config.port)
server = wsgi.Server(server_start_params, app)
print('Server started on {0}:{1}'.format(*server_start_params))
try:
server.start()
except KeyboardInterrupt:
engine_runner.stop()
server.stop()
class SettingsManagerTest(DbTestCase):
def setUp(self):
super(SettingsManagerTest, self).setUp()
self.settings_manager = SettingsManager()
def test_get_default_password(self):
self.assertEqual('monitorrent', self.settings_manager.get_password())
def test_set_password(self):
value = 'p@$$w0rd!'
self.settings_manager.set_password(value)
self.assertEqual(value, self.settings_manager.get_password())
def test_get_default_is_authentication_enabled(self):
self.assertTrue(self.settings_manager.get_is_authentication_enabled())
def test_set_is_authentication_enabled(self):
value = False
self.settings_manager.set_is_authentication_enabled(value)
self.assertEqual(value, self.settings_manager.get_is_authentication_enabled())
def test_enable_disable_authentication(self):
self.settings_manager.disable_authentication()
self.assertFalse(self.settings_manager.get_is_authentication_enabled())
self.settings_manager.enable_authentication()
self.assertTrue(self.settings_manager.get_is_authentication_enabled())
def test_default_client(self):
self.assertIsNone(self.settings_manager.get_default_client())
client = 'transmission'
self.settings_manager.set_default_client(client)
self.assertEqual(client, self.settings_manager.get_default_client())
def test_get_is_developer_mode(self):
self.assertFalse(self.settings_manager.get_is_developer_mode())
@data(True, False)
def test_set_is_developer_mode(self, value):
self.settings_manager.set_is_developer_mode(value)
self.assertEqual(value, self.settings_manager.get_is_developer_mode())
def test_get_default_requests_timeout(self):
self.assertEqual(10, self.settings_manager.requests_timeout)
def test_set_requests_timeout(self):
self.assertEqual(10, self.settings_manager.requests_timeout)
self.settings_manager.requests_timeout = 20
self.assertEqual(20, self.settings_manager.requests_timeout)
def test_set_float_requests_timeout(self):
self.assertEqual(10, self.settings_manager.requests_timeout)
self.settings_manager.requests_timeout = 20.7
self.assertEqual(20.7, self.settings_manager.requests_timeout)
def test_get_default_plugin_settings(self):
self.assertEqual(10, self.settings_manager.tracker_settings.requests_timeout)
def test_set_plugin_settings(self):
plugin_settings = self.settings_manager.tracker_settings
self.assertEqual(10, plugin_settings.requests_timeout)
plugin_settings.requests_timeout = 20
self.settings_manager.tracker_settings = plugin_settings
self.assertEqual(20, self.settings_manager.tracker_settings.requests_timeout)
def test_set_float_plugin_settings(self):
plugin_settings = self.settings_manager.tracker_settings
self.assertEqual(10, plugin_settings.requests_timeout)
plugin_settings.requests_timeout = 20.3
self.settings_manager.tracker_settings = plugin_settings
self.assertEqual(20.3, self.settings_manager.tracker_settings.requests_timeout)
def test_get_remove_logs_interval(self):
self.assertEqual(10, self.settings_manager.remove_logs_interval)
def test_set_remove_logs_interval(self):
self.assertEqual(10, self.settings_manager.remove_logs_interval)
self.settings_manager.remove_logs_interval = 20
self.assertEqual(20, self.settings_manager.remove_logs_interval)
def test_get_is_proxy_enabled(self):
self.assertFalse(self.settings_manager.get_is_proxy_enabled())
@data(True, False)
def test_set_is_proxy_enabled(self, value):
self.settings_manager.set_is_proxy_enabled(value)
self.assertEqual(self.settings_manager.get_is_proxy_enabled(), value)
def test_get_proxy(self):
self.assertEqual(self.settings_manager.get_proxy('http'), None)
self.assertEqual(self.settings_manager.get_proxy('https'), None)
def test_set_proxy(self):
self.assertEqual(self.settings_manager.get_proxy('http'), None)
self.assertEqual(self.settings_manager.get_proxy('https'), None)
self.settings_manager.set_proxy('http', 'http://1.1.1.1:8888')
self.assertEqual(self.settings_manager.get_proxy('http'), 'http://1.1.1.1:8888')
self.assertEqual(self.settings_manager.get_proxy('https'), None)
expected = {
'http': 'http://1.1.1.1:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
self.settings_manager.set_proxy('https', 'http://2.2.2.2:8888')
self.assertEqual(self.settings_manager.get_proxy('http'), 'http://1.1.1.1:8888')
self.assertEqual(self.settings_manager.get_proxy('https'), 'http://2.2.2.2:8888')
expected = {
'http': 'http://1.1.1.1:8888',
'https': 'http://2.2.2.2:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
def test_unset_proxy(self):
self.settings_manager.set_proxy('http', 'http://1.1.1.1:8888')
self.settings_manager.set_proxy('https', 'http://2.2.2.2:8888')
expected = {
'http': 'http://1.1.1.1:8888',
'https': 'http://2.2.2.2:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
self.settings_manager.set_proxy('https', None)
self.settings_manager.set_proxy('http://rutor.org', None)
expected = {
'http': 'http://1.1.1.1:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
@data(True, False)
def test_set_is_new_version_check_enabled(self, value):
self.settings_manager.set_is_new_version_checker_enabled(value)
self.assertEqual(self.settings_manager.get_is_new_version_checker_enabled(), value)
def test_get_new_version_check_include_prerelease(self):
self.assertFalse(self.settings_manager.get_new_version_check_include_prerelease())
@data(True, False)
def test_set_new_version_check_include_prerelease(self, value):
self.settings_manager.set_new_version_check_include_prerelease(value)
self.assertEqual(self.settings_manager.get_new_version_check_include_prerelease(), value)
def test_get_new_version_check_interval_default_value(self):
self.assertEqual(3600, self.settings_manager.new_version_check_interval)
def test_get_new_version_check_interval_set_value(self):
self.assertEqual(3600, self.settings_manager.new_version_check_interval)
self.settings_manager.new_version_check_interval = 7200
self.assertEqual(7200, self.settings_manager.new_version_check_interval)
class SettingsManagerTest(DbTestCase):
def setUp(self):
super(SettingsManagerTest, self).setUp()
self.settings_manager = SettingsManager()
def test_get_default_password(self):
self.assertEqual('monitorrent', self.settings_manager.get_password())
def test_set_password(self):
value = 'p@$$w0rd!'
self.settings_manager.set_password(value)
self.assertEqual(value, self.settings_manager.get_password())
def test_get_default_is_authentication_enabled(self):
self.assertTrue(self.settings_manager.get_is_authentication_enabled())
def test_set_is_authentication_enabled(self):
value = False
self.settings_manager.set_is_authentication_enabled(value)
self.assertEqual(value, self.settings_manager.get_is_authentication_enabled())
def test_enable_disable_authentication(self):
self.settings_manager.disable_authentication()
self.assertFalse(self.settings_manager.get_is_authentication_enabled())
self.settings_manager.enable_authentication()
self.assertTrue(self.settings_manager.get_is_authentication_enabled())
def test_default_client(self):
self.assertIsNone(self.settings_manager.get_default_client())
client = 'transmission'
self.settings_manager.set_default_client(client)
self.assertEqual(client, self.settings_manager.get_default_client())
def test_get_is_developer_mode(self):
self.assertFalse(self.settings_manager.get_is_developer_mode())
@data(True, False)
def test_set_is_developer_mode(self, value):
self.settings_manager.set_is_developer_mode(value)
self.assertEqual(value, self.settings_manager.get_is_developer_mode())
def test_get_default_requests_timeout(self):
self.assertEqual(10, self.settings_manager.requests_timeout)
def test_set_requests_timeout(self):
self.assertEqual(10, self.settings_manager.requests_timeout)
self.settings_manager.requests_timeout = 20
self.assertEqual(20, self.settings_manager.requests_timeout)
def test_set_float_requests_timeout(self):
self.assertEqual(10, self.settings_manager.requests_timeout)
self.settings_manager.requests_timeout = 20.7
self.assertEqual(20.7, self.settings_manager.requests_timeout)
def test_get_default_plugin_settings(self):
self.assertEqual(10, self.settings_manager.tracker_settings.requests_timeout)
def test_set_plugin_settings(self):
plugin_settings = self.settings_manager.tracker_settings
self.assertEqual(10, plugin_settings.requests_timeout)
plugin_settings.requests_timeout = 20
self.settings_manager.tracker_settings = plugin_settings
self.assertEqual(20, self.settings_manager.tracker_settings.requests_timeout)
def test_set_float_plugin_settings(self):
plugin_settings = self.settings_manager.tracker_settings
self.assertEqual(10, plugin_settings.requests_timeout)
plugin_settings.requests_timeout = 20.3
self.settings_manager.tracker_settings = plugin_settings
self.assertEqual(20.3, self.settings_manager.tracker_settings.requests_timeout)
def test_get_remove_logs_interval(self):
self.assertEqual(10, self.settings_manager.remove_logs_interval)
def test_set_remove_logs_interval(self):
self.assertEqual(10, self.settings_manager.remove_logs_interval)
self.settings_manager.remove_logs_interval = 20
self.assertEqual(20, self.settings_manager.remove_logs_interval)
def test_get_is_proxy_enabled(self):
self.assertFalse(self.settings_manager.get_is_proxy_enabled())
@data(True, False)
def test_set_is_proxy_enabled(self, value):
self.settings_manager.set_is_proxy_enabled(value)
self.assertEqual(self.settings_manager.get_is_proxy_enabled(), value)
def test_get_proxy(self):
self.assertEqual(self.settings_manager.get_proxy('http'), None)
self.assertEqual(self.settings_manager.get_proxy('https'), None)
def test_set_proxy(self):
self.assertEqual(self.settings_manager.get_proxy('http'), None)
self.assertEqual(self.settings_manager.get_proxy('https'), None)
self.settings_manager.set_proxy('http', 'http://1.1.1.1:8888')
self.assertEqual(self.settings_manager.get_proxy('http'), 'http://1.1.1.1:8888')
self.assertEqual(self.settings_manager.get_proxy('https'), None)
expected = {
'http': 'http://1.1.1.1:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
self.settings_manager.set_proxy('https', 'http://2.2.2.2:8888')
self.assertEqual(self.settings_manager.get_proxy('http'), 'http://1.1.1.1:8888')
self.assertEqual(self.settings_manager.get_proxy('https'), 'http://2.2.2.2:8888')
expected = {
'http': 'http://1.1.1.1:8888',
'https': 'http://2.2.2.2:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
def test_unset_proxy(self):
self.settings_manager.set_proxy('http', 'http://1.1.1.1:8888')
self.settings_manager.set_proxy('https', 'http://2.2.2.2:8888')
expected = {
'http': 'http://1.1.1.1:8888',
'https': 'http://2.2.2.2:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
self.settings_manager.set_proxy('https', None)
self.settings_manager.set_proxy('http://rutor.org', None)
expected = {
'http': 'http://1.1.1.1:8888'
}
self.assertEqual(self.settings_manager.get_proxies(), expected)
@data(True, False)
def test_set_is_new_version_check_enabled(self, value):
self.settings_manager.set_is_new_version_checker_enabled(value)
self.assertEqual(self.settings_manager.get_is_new_version_checker_enabled(), value)
def test_get_new_version_check_include_prerelease(self):
self.assertFalse(self.settings_manager.get_new_version_check_include_prerelease())
@data(True, False)
def test_set_new_version_check_include_prerelease(self, value):
self.settings_manager.set_new_version_check_include_prerelease(value)
self.assertEqual(self.settings_manager.get_new_version_check_include_prerelease(), value)
def test_get_new_version_check_interval_default_value(self):
self.assertEqual(3600, self.settings_manager.new_version_check_interval)
def test_get_new_version_check_interval_set_value(self):
self.assertEqual(3600, self.settings_manager.new_version_check_interval)
self.settings_manager.new_version_check_interval = 7200
self.assertEqual(7200, self.settings_manager.new_version_check_interval)
def test_get_external_notifications_levels(self):
self.assertEqual(self.settings_manager.get_external_notifications_levels(),
['DOWNLOAD', 'ERROR', 'STATUS_CHANGED'])
self.settings_manager.set_external_notifications_levels(['DOWNLOAD', 'ERROR'])
self.assertEqual(self.settings_manager.get_external_notifications_levels(), ['DOWNLOAD', 'ERROR'])
self.settings_manager.set_external_notifications_levels(None)
self.assertEqual(self.settings_manager.get_external_notifications_levels(),
['DOWNLOAD', 'ERROR', 'STATUS_CHANGED'])
self.settings_manager.set_external_notifications_levels([])
self.assertEqual(self.settings_manager.get_external_notifications_levels(), [])
def test_set_unknown_notification_levels_levels(self):
self.assertEqual(self.settings_manager.get_external_notifications_levels(),
['DOWNLOAD', 'ERROR', 'STATUS_CHANGED'])
self.settings_manager.set_external_notifications_levels(['DOWNLOAD', 'WRONG'])
self.assertEqual(self.settings_manager.get_external_notifications_levels(), ['DOWNLOAD'])
self.settings_manager.set_external_notifications_levels(['UNKNOWN', 'WRONG'])
self.assertEqual(self.settings_manager.get_external_notifications_levels(), [])
self.settings_manager.set_external_notifications_levels(None)
self.assertEqual(self.settings_manager.get_external_notifications_levels(),
['DOWNLOAD', 'ERROR', 'STATUS_CHANGED'])
def test_remove_not_existing_settings_success(self):
self.settings_manager.set_external_notifications_levels(None)
self.assertEqual(self.settings_manager.get_external_notifications_levels(),
['DOWNLOAD', 'ERROR', 'STATUS_CHANGED'])
def test_get_existing_external_notifications_levels_success(self):
self.assertEqual(self.settings_manager.get_existing_external_notifications_levels(),
['DOWNLOAD', 'ERROR', 'STATUS_CHANGED'])
def main():
def try_int(s, base=10, val=None):
if s is None:
return None
try:
return int(s, base)
except ValueError:
return val
class Config(object):
debug = False
ip = '0.0.0.0'
port = 6687
db_path = 'monitorrent.db'
config = 'config.py'
def __init__(self, parsed_args):
if parsed_args.config is not None and not os.path.isfile(parsed_args.config):
warnings.warn('File not found: {}'.format(parsed_args.config))
config_path = parsed_args.config or self.config
if os.path.isfile(config_path):
# noinspection PyBroadException
try:
parsed_config = {}
with open(config_path) as config_file:
six.exec_(compile(config_file.read(), config_path, 'exec'), {}, parsed_config)
self.debug = parsed_config.get('debug', self.debug)
self.ip = parsed_config.get('ip', self.ip)
self.port = parsed_config.get('port', self.port)
self.db_path = parsed_config.get('db_path', self.db_path)
except:
ex, val, tb = sys.exc_info()
warnings.warn('Error reading: {0}: {1} ({2}'.format(parsed_args.config, ex, val))
env_debug = (os.environ.get('MONITORRENT_DEBUG', None) in ['true', 'True', '1'])
self.debug = parsed_args.debug or env_debug or self.debug
self.ip = parsed_args.ip or os.environ.get('MONITORRENT_IP', None) or self.ip
self.port = parsed_args.port or try_int(os.environ.get('MONITORRENT_PORT', None)) or self.port
self.db_path = parsed_args.db_path or os.environ.get('MONITORRENT_DB_PATH', None) or self.db_path
parser = argparse.ArgumentParser(description='Monitorrent server')
parser.add_argument('--debug', action='store_true',
help='Run in debug mode. Secret key is always the same.')
parser.add_argument('--ip', type=str, dest='ip',
help='Bind interface. Default is {0}'.format(Config.ip))
parser.add_argument('--port', type=int, dest='port',
help='Port for server. Default is {0}'.format(Config.port))
parser.add_argument('--db-path', type=str, dest='db_path',
help='Path to SQL lite database. Default is to {0}'.format(Config.db_path))
parser.add_argument('--config', type=str, dest='config',
default=os.environ.get('MONITORRENT_CONFIG', None),
help='Path to config file (default {0})'.format(Config.config))
parsed_args = parser.parse_args()
config = Config(parsed_args)
db_connection_string = "sqlite:///" + config.db_path
init_db_engine(db_connection_string, False)
load_plugins()
upgrade()
create_db()
settings_manager = SettingsManager()
tracker_manager = TrackersManager(settings_manager, get_plugins('tracker'))
clients_manager = DbClientsManager(get_plugins('client'), settings_manager)
notifier_manager = NotifierManager(get_plugins('notifier'))
log_manager = ExecuteLogManager()
engine_runner_logger = DbLoggerWrapper(None, log_manager, settings_manager)
engine_runner = DBEngineRunner(engine_runner_logger, tracker_manager, clients_manager)
new_version_checker = NewVersionChecker(settings_manager.get_new_version_check_include_prerelease())
if settings_manager.get_is_new_version_checker_enabled():
# noinspection PyBroadException
try:
new_version_checker.execute()
except:
pass
new_version_checker.start(settings_manager.new_version_check_interval)
debug = config.debug
if debug:
secret_key = 'Secret!'
token = 'monitorrent'
else:
secret_key = os.urandom(24)
token = ''.join(random.choice(string.ascii_letters) for _ in range(8))
app = create_app(secret_key, token, tracker_manager, clients_manager, notifier_manager, settings_manager,
engine_runner, log_manager, new_version_checker)
d = wsgiserver.WSGIPathInfoDispatcher({'/': app})
server_start_params = (config.ip, config.port)
server = wsgiserver.CherryPyWSGIServer(server_start_params, d)
print('Server started on {0}:{1}'.format(*server_start_params))
try:
server.start()
except KeyboardInterrupt:
engine_runner.stop()
server.stop()
