feat: use return error codes for easier integration in other projects

medusa/_cli/__main__.py

@@ -44,7 +44,7 @@ def main():
     sys.tracebacklimit = 1000
 
     try:
-        importlib.import_module(f"medusa._cli.{cmd}").main()
+        sys.exit(importlib.import_module(f"medusa._cli.{cmd}").main())
     except Exception as e:
         tb_item = sys.exc_info()[2]
         traceback.print_tb(tb_item)

medusa/_cli/analyse.py

@@ -17,26 +17,28 @@
 """
 
 
-def main():
+def main() -> int:
     args = docopt(__doc__)
 
-    if args["<contract>"]:
-        path = args["<contract>"]
-        print_output = args["--print-output"]
+    if not (path := args["<contract>"]):
+        return 1
 
-        # Get Vyper AST
-        vyper_ast = get_vyper_ast(path)
+    print_output = args["--print-output"]
 
-        # Perform analysis and format the result as a string
-        output_dict = analyse(vyper_ast)
-        formatted_analysis = format_analysis(output_dict)
+    # Get Vyper AST
+    vyper_ast = get_vyper_ast(path)
 
-        # Write to output file
-        if output_file := args["--output"]:
-            write_analysis(formatted_analysis, output_file)
+    # Perform analysis and format the result as a string
+    output_dict = analyse(vyper_ast)
+    formatted_analysis = format_analysis(output_dict)
 
-        # Print analysis to console
-        if print_output:
-            print(formatted_analysis)
+    # Write to output file
+    if output_file := args["--output"]:
+        write_analysis(formatted_analysis, output_file)
 
-        print(f"\nSuccessfully analysed {path}!")
+    # Print analysis to console
+    if print_output:
+        print(formatted_analysis)
+
+    print(f"\nSuccessfully analysed {path}!")
+    return int(bool(output_dict))

medusa/_cli/transform.py

@@ -12,13 +12,17 @@
 """
 
 
-def main():
+def main() -> int:
     args = docopt(__doc__)
 
-    if args["<contract>"]:
-        path = args["<contract>"]
+    if not args["<contract>"]:
+        return 1
 
-        # Get Vyper AST
-        vyper_ast = get_vyper_ast(path)
-        ast_dict = vyper_ast.to_dict()
-        print(json.dumps(ast_dict, indent=4, sort_keys=True))
+    path = args["<contract>"]
+
+    # Get Vyper AST
+    vyper_ast = get_vyper_ast(path)
+    ast_dict = vyper_ast.to_dict()
+    print(json.dumps(ast_dict, indent=4, sort_keys=True))
+
+    return 0