tas

.gitignore

@@ -13,4 +13,5 @@ config_exp
 test_smartmuv.py
 dist
 smartmuv.egg-info
-test_env
+test_env
+output_data

src/ast_parsing/ast_parser.py

@@ -99,7 +99,7 @@ def check_if_stmt(statement, definition, functions, variables, mappings):
     return functions, variables, mappings
 
 
-def parse_ast(children, cont_name):
+def parse_ast(children):
     all_contracts_details = {}
     variables = []
     state_vars = []

try_smartmuv.py

@@ -40,17 +40,30 @@ def read_source_code(contract_name, input_dir):
         for slot in slots_details:
             print(slot)
 
+        # save to json file
+        with open(f"tests/output_data/{contract_name}_slot_details.json", "w") as f:
+            json.dump(slots_details, f)
+
+
     elif option == "2":
         results, slots_details, slots_and_data, block_number = extract_regular_variables(
             contract_name, source_code, cont_addr, compiler_version, "mainnet")
         print("\nDetails of Extracted Regular Variables are:\n")
         for var in results:
             print(var)
-
+
+        # save to json file
+        with open(f"tests/output_data/{contract_name}_elementary_data.json", "w") as f:
+            json.dump(results, f)
+
     elif option == "3":
         contract_state, _, variables_slot_results, slots_and_data, key_analysis_result, block_number = extract_contract_state(
             contract_name, source_code, cont_addr, compiler_version, "mainnet")
         print("\nDetails of Extracted Contract State:\n")
         for var in contract_state:
             print(var)
+
+        # save to json file
+        with open(f"tests/output_data/{contract_name}_complete_data.json", "w") as f:
+            json.dump(contract_state, f)