def main():
args = CmdlineParser().parse_args()
# Parse environment variables from Dockerfile.
# If an environment variable is not empty and it's not defined in the arguments, then we'll use the environment
# variable.
if args.config_file_name is None and len(
os.environ.get("CONFIG_FILE_NAME", "")) > 0:
args.config_file_name = os.environ["CONFIG_FILE_NAME"]
if args.config_password is None and len(
os.environ.get("CONFIG_PASSWORD", "")) > 0:
args.config_password = os.environ["CONFIG_PASSWORD"]
# If no password is given from the command line, prompt for one.
secrets_manager_cls = ETHKeyFileSecretManger
client_config_map = load_client_config_map_from_file()
if args.config_password is None:
secrets_manager = login_prompt(secrets_manager_cls,
style=load_style(client_config_map))
if not secrets_manager:
return
else:
secrets_manager = secrets_manager_cls(args.config_password)
asyncio.get_event_loop().run_until_complete(
quick_start(args, secrets_manager))
def main():
chdir_to_data_directory()
secrets_manager_cls = ETHKeyFileSecretManger
ev_loop: asyncio.AbstractEventLoop = asyncio.get_event_loop()
client_config_map = load_client_config_map_from_file()
if login_prompt(secrets_manager_cls, style=load_style(client_config_map)):
ev_loop.run_until_complete(main_async(client_config_map))
def test_login_success(
self,
new_password_required_mock: MagicMock,
login_mock: MagicMock,
input_dialog_mock: MagicMock,
message_dialog_mock: MagicMock,
):
new_password_required_mock.return_value = False
run_mock = MagicMock()
run_mock.run.return_value = self.password
input_dialog_mock.return_value = run_mock
login_mock.return_value = True
self.assertTrue(login_prompt(ETHKeyFileSecretManger, style=load_style(self.client_config_map)))
self.assertEqual(1, len(login_mock.mock_calls))
message_dialog_mock.assert_not_called()
def test_login_error_retries(
self,
new_password_required_mock: MagicMock,
login_mock: MagicMock,
input_dialog_mock: MagicMock,
message_dialog_mock: MagicMock,
):
new_password_required_mock.return_value = False
run_mock = MagicMock()
run_mock.run.return_value = "somePassword"
input_dialog_mock.return_value = run_mock
message_dialog_mock.return_value = run_mock
login_mock.side_effect = [False, True]
self.assertTrue(login_prompt(ETHKeyFileSecretManger, style=load_style(self.client_config_map)))
self.assertEqual(2, len(login_mock.mock_calls))
message_dialog_mock.assert_called()
def main():
args = CmdlineParser().parse_args()
# Parse environment variables from Dockerfile.
# If an environment variable is not empty and it's not defined in the arguments, then we'll use the environment
# variable.
if args.config_file_name is None and len(os.environ.get("CONFIG_FILE_NAME", "")) > 0:
args.config_file_name = os.environ["CONFIG_FILE_NAME"]
if args.config_password is None and len(os.environ.get("CONFIG_PASSWORD", "")) > 0:
args.config_password = os.environ["CONFIG_PASSWORD"]
# If no password is given from the command line, prompt for one.
if args.config_password is None:
if not login_prompt():
return
asyncio.get_event_loop().run_until_complete(quick_start(args))
init_logging("hummingbot_logs.yml")
read_system_configs_from_yml()
hb = HummingbotApplication.main_application()
with patch_stdout(log_field=hb.app.log_field):
dev_mode = check_dev_mode()
if dev_mode:
hb.app.log("Running from dev branches. Full remote logging will be enabled.")
init_logging("hummingbot_logs.yml",
override_log_level=global_config_map.get("log_level").value,
dev_mode=dev_mode)
tasks: List[Coroutine] = [hb.run()]
if global_config_map.get("debug_console").value:
if not hasattr(__builtins__, "help"):
import _sitebuiltins
__builtins__.help = _sitebuiltins._Helper()
from hummingbot.core.management.console import start_management_console
management_port: int = detect_available_port(8211)
tasks.append(start_management_console(locals(), host="localhost", port=management_port))
await safe_gather(*tasks)
if __name__ == "__main__":
chdir_to_data_directory()
if login_prompt():
ev_loop: asyncio.AbstractEventLoop = asyncio.get_event_loop()
ev_loop.run_until_complete(main())
def main():
chdir_to_data_directory()
if login_prompt():
ev_loop: asyncio.AbstractEventLoop = asyncio.get_event_loop()
ev_loop.run_until_complete(main_async())
