Merge pull request #23 from lukevs/project-structure

Add project structure

mev_inspect/block.py

@@ -4,7 +4,7 @@ from typing import List
 
 from web3 import Web3
 
-from schemas import Block, BlockCall, BlockCallType
+from mev_inspect.schemas import Block, BlockCall, BlockCallType
 
 
 cache_directory = './cache'

mev_inspect/config.py

@@ -0,0 +1,13 @@
+import os
+import configparser
+
+
+THIS_FILE_DIRECTORY = os.path.dirname(__file__)
+CONFIG_PATH = os.path.join(THIS_FILE_DIRECTORY, "config.ini")
+
+
+def load_config():
+    config = configparser.ConfigParser()
+    config.read(CONFIG_PATH)
+
+    return config

mev_inspect/inspector_uniswap.py

@@ -1,11 +1,11 @@
-from web3 import Web3
-import configparser
 import json
-import utils
 
-## Config file is used for addresses/ABIs
-config = configparser.ConfigParser()
-config.read('./utils/config.ini')
+from web3 import Web3
+
+from mev_inspect import utils
+from mev_inspect.config import load_config
+
+config = load_config()
 
 uniswap_router_abi = json.loads(config['ABI']['UniswapV2Router'])
 uniswap_router_address = (config['ADDRESSES']['UniswapV2Router'])

mev_inspect/processor.py

@@ -1,4 +1,4 @@
-from schemas.utils import to_original_json_dict
+from mev_inspect.schemas.utils import to_original_json_dict
 
 
 class Processor:

mev_inspect/tokenflow.py

@@ -1,10 +1,10 @@
-# from web3 import Web3, HTTPProvider
-from pathlib import Path
 import json
-import configparser
+from pathlib import Path
+
+from mev_inspect.config import load_config
+
+config = load_config()
 
-config = configparser.ConfigParser()
-config.read('./utils/config.ini')
 rpc_url = config['RPC']['Endpoint']
 weth_address = config['ADDRESSES']['WETH']
 # w3 = Web3(HTTPProvider(rpc_url))

testing_file.py

@@ -1,10 +1,12 @@
-from processor import Processor
-from web3.providers import base
-from inspector_uniswap import UniswapInspector
-import block
-from web3 import Web3
 import argparse
 
+from web3.providers import base
+from web3 import Web3
+
+from mev_inspect import block
+from mev_inspect.inspector_uniswap import UniswapInspector
+from mev_inspect.processor import Processor
+
 parser = argparse.ArgumentParser(description='Inspect some blocks.')
 parser.add_argument('-block_number', metavar='b', type=int, nargs='+',
                     help='the block number you are targetting, eventually this will need to be changed')

tests/liquidation_test.py

@@ -1,6 +1,7 @@
 import unittest
-import inspector_compound
-import inspector_aave
+
+from mev_inspect import inspector_compound
+from mev_inspect import inspector_aave
 
 class TestLiquidations (unittest.TestCase):
     def test_compound_liquidation(self):

tests/tokenflow_test.py

@@ -1,5 +1,7 @@
 import unittest
-import tokenflow
+
+from mev_inspect import tokenflow
+
 
 class TestTokenFlow (unittest.TestCase):
     def test_simple_arb(self):