diff --git a/great_ai/__main__.py b/great_ai/__main__.py index dfc30d5..f07f252 100644 --- a/great_ai/__main__.py +++ b/great_ai/__main__.py @@ -40,7 +40,7 @@ GREAT_AI_LOGGING_CONFIG = { } -def main() -> None: +def serve() -> None: args = parse_arguments() should_auto_reload = not _is_in_production_mode(logger=None) @@ -200,10 +200,14 @@ class GreatAIReload(BaseReload): sock.close() -if __name__ == "__main__": +def main() -> None: try: - main() + serve() except KeyboardInterrupt: exit() except Exception as e: logger.error(e) + + +if __name__ == "__main__": + main() diff --git a/great_ai/large_file/__main__.py b/great_ai/large_file/__main__.py index 576f781..512a7a3 100644 --- a/great_ai/large_file/__main__.py +++ b/great_ai/large_file/__main__.py @@ -11,7 +11,7 @@ from .parse_arguments import parse_arguments logger = get_logger("large_file") -def main() -> None: +def handle_command() -> None: parser, args = parse_arguments() large_file = get_class(args) @@ -61,11 +61,15 @@ def get_class(args: Namespace) -> Type[LargeFileBase]: return large_file -if __name__ == "__main__": +def main() -> None: try: - main() + handle_command() except KeyboardInterrupt: logger.warning("Exiting") exit() except Exception as e: logger.exception(e) + + +if __name__ == "__main__": + main() diff --git a/great_ai/parse_arguments.py b/great_ai/parse_arguments.py index e816b6b..60974cd 100644 --- a/great_ai/parse_arguments.py +++ b/great_ai/parse_arguments.py @@ -12,7 +12,7 @@ def parse_arguments() -> Namespace: help="the name of the file containing your to-be-served function such as `main.py`\n", ) - default_host = "0.0.0.0" + default_host = "127.0.0.1" parser.add_argument( "--host", type=str,