From 2d36ef9488b85b922f5055ca62b1e5f6350ee038 Mon Sep 17 00:00:00 2001
From: harisang <harisangelidakis@gmail.com>
Date: Thu, 17 Oct 2024 01:14:45 +0300
Subject: [PATCH] Add more env variables

---
 src/daemon.py | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/src/daemon.py b/src/daemon.py
index acd58da..1273862 100644
--- a/src/daemon.py
+++ b/src/daemon.py
@@ -8,25 +8,17 @@
 def main() -> None:
     # valid chain names: mainnet, xdai, arbitrum_one
     chain_name = os.getenv("CHAIN_NAME")
+    process_imbalances = os.getenv("PROCESS_IMBALANCES", True)
+    process_prices = os.getenv("PROCESS_PRICES", False)
+    process_fees = os.getenv("PROCESS_FEES", False)
+
     if chain_name is None:
         logger.error("CHAIN_NAME environment variable is not set.")
         return
 
-    process_imbalances = True
-    process_fees = False
-    process_prices = True
-
-    if chain_name == "arbitrum_one":
-        process_imbalances = False
-        process_prices = False
-
-    if chain_name == "xdai":
-        process_prices = False
-
     processor = TransactionProcessor(
         chain_name, process_imbalances, process_fees, process_prices
     )
-
     processor.run()