test.py

import sys
import pprint

from solidity_parser import parser

from app.src.react_agent.utils import ExtractEntryFunctions, FunctionParser, RedisUtil, StorageParser

root_directory = "/Users/jetty/upside_academy/audit_project/contracts"
src_directory = "/Users/jetty/upside_academy/audit_project/contracts/src"

# _entry_points = ExtractEntryFunctions.extract_entry(root_directory)
# print(f"entry points: {_entry_points}")
function_code = FunctionParser.process_directory(src_directory, "function_code.json")
RedisUtil.set_function_codes_from_json(function_code)
print(f"{RedisUtil.get_function_code('BlockRewardController.onlyDistributor')}")
print("--------------------------------")
storage_variables = StorageParser.parse_project(root_directory, src_directory)
RedisUtil.set_storage_variables_from_json(storage_variables)
print(f"{RedisUtil.get_storage_variables('BlockRewardController')}")