From 9bb79f700186bd9bee716cd7f8fa010258f47846 Mon Sep 17 00:00:00 2001
From: Luke Van Seters <lukevanseters@gmail.com>
Date: Fri, 30 Jul 2021 13:29:24 -0400
Subject: [PATCH] Add the skeleton

---
 mev_inspect/schemas/strategies.py   | 5 +++++
 mev_inspect/strategies/__init__.py  | 0
 mev_inspect/strategies/arbitrage.py | 8 ++++++++
 scripts/inspect_block.py            | 4 ++++
 4 files changed, 17 insertions(+)
 create mode 100644 mev_inspect/schemas/strategies.py
 create mode 100644 mev_inspect/strategies/__init__.py
 create mode 100644 mev_inspect/strategies/arbitrage.py

diff --git a/mev_inspect/schemas/strategies.py b/mev_inspect/schemas/strategies.py
new file mode 100644
index 0000000..f94dda2
--- /dev/null
+++ b/mev_inspect/schemas/strategies.py
@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class Arbitrage(BaseModel):
+    pass
diff --git a/mev_inspect/strategies/__init__.py b/mev_inspect/strategies/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/mev_inspect/strategies/arbitrage.py b/mev_inspect/strategies/arbitrage.py
new file mode 100644
index 0000000..ccc7410
--- /dev/null
+++ b/mev_inspect/strategies/arbitrage.py
@@ -0,0 +1,8 @@
+from typing import List
+
+from mev_inspect.schemas.classified_traces import ClassifiedTrace
+from mev_inspect.schemas.strategies import Arbitrage
+
+
+def get_arbitrages(traces: List[ClassifiedTrace]) -> List[Arbitrage]:
+    return []
diff --git a/scripts/inspect_block.py b/scripts/inspect_block.py
index 8ae7f6c..a7eaac7 100644
--- a/scripts/inspect_block.py
+++ b/scripts/inspect_block.py
@@ -11,6 +11,7 @@ from mev_inspect.crud.classified_traces import (
 from mev_inspect.db import get_session
 from mev_inspect.classifier_specs import CLASSIFIER_SPECS
 from mev_inspect.trace_classifier import TraceClassifier
+from mev_inspect.strategies.arbitrage import get_arbitrages
 
 
 @click.command()
@@ -39,6 +40,9 @@ def inspect_block(block_number: int, rpc: str):
     write_classified_traces(db_session, classified_traces)
     db_session.close()
 
+    arbitrages = get_arbitrages(classified_traces)
+    print(f"Found {len(arbitrages)} arbitrages")
+
     stats = get_stats(classified_traces)
     print(json.dumps(stats, indent=4))