def test_start_twice_should_raise(self):
checker = NewVersionChecker(False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(10)
with self.assertRaises(Exception):
checker.start(10)
checker.stop()
execute_mock.assert_not_called()
def test_is_started(self):
checker = NewVersionChecker(False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(10)
self.assertTrue(checker.is_started())
checker.stop()
self.assertFalse(checker.is_started())
execute_mock.assert_not_called()
def test_is_started(self):
checker = NewVersionChecker(Mock(), False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(10)
self.assertTrue(checker.is_started())
checker.stop()
self.assertFalse(checker.is_started())
execute_mock.assert_not_called()
def test_timer_stop_dont_call_execute(self):
checker = NewVersionChecker(False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(1)
checker.stop()
execute_mock.assert_not_called()
self.assertLess(execute_mock.call_count, 4)
def test_start_twice_should_raise(self):
checker = NewVersionChecker(Mock(), False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(10)
with self.assertRaises(Exception):
checker.start(10)
checker.stop()
execute_mock.assert_not_called()
def test_timer_stop_dont_call_execute(self):
checker = NewVersionChecker(Mock(), False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(1)
checker.stop()
execute_mock.assert_not_called()
self.assertLess(execute_mock.call_count, 4)
def test_timer_calls(self):
checker = NewVersionChecker(False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(0.1)
sleep(0.3)
checker.stop()
self.assertGreaterEqual(execute_mock.call_count, 2)
self.assertLess(execute_mock.call_count, 4)
sleep(0.3)
self.assertLess(execute_mock.call_count, 4)
def test_update(self, include_prerelease, is_started, start_interval, enabled, interval, start_called, stop_called):
checker = NewVersionChecker(False)
def start_side_effect(i):
checker.interval = i
start_mock = Mock(side_effect=start_side_effect)
stop_mock = Mock()
is_started_mock = Mock(return_value=is_started)
checker.interval = start_interval
checker.start = start_mock
checker.stop = stop_mock
checker.is_started = is_started_mock
checker.update(include_prerelease, enabled, interval)
self.assertEqual(checker.interval, interval)
self.assertEqual(checker.include_prereleases, include_prerelease)
if start_called:
start_mock.assert_called_once_with(interval)
else:
start_mock.assert_not_called()
if stop_called:
stop_mock.assert_called_once()
else:
stop_mock.assert_not_called()
def test_update(self, include_prerelease, is_started, start_interval,
enabled, interval, start_called, stop_called):
checker = NewVersionChecker(Mock(), False)
def start_side_effect(i):
checker.interval = i
start_mock = Mock(side_effect=start_side_effect)
stop_mock = Mock()
is_started_mock = Mock(return_value=is_started)
checker.interval = start_interval
checker.start = start_mock
checker.stop = stop_mock
checker.is_started = is_started_mock
checker.update(include_prerelease, enabled, interval)
self.assertEqual(checker.interval, interval)
self.assertEqual(checker.include_prereleases, include_prerelease)
if start_called:
start_mock.assert_called_once_with(interval)
else:
start_mock.assert_not_called()
if stop_called:
stop_mock.assert_called_once()
else:
stop_mock.assert_not_called()
def test_timer_calls(self):
checker = NewVersionChecker(Mock(), False)
execute_mock = Mock(return_value=True)
checker.execute = execute_mock
checker.start(0.1)
sleep(0.3)
checker.stop()
self.assertGreaterEqual(execute_mock.call_count, 2)
self.assertLess(execute_mock.call_count, 4)
sleep(0.3)
self.assertLess(execute_mock.call_count, 4)
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
log.bind(config_is_file=True)
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)
if config.debug:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.WARN)
log = structlog.get_logger()
log.info("Configuration finished", config=config.__dict__)
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, log)
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:
print('Stopping engine')
engine_runner.stop()
print('Stopping new_version_checker')
new_version_checker.stop()
server.stop()
print('Server stopped')
