#337 fix running airdropper (#347)

Co-authored-by: ivanl <[email protected]>

proxy/__main__.py

@@ -8,7 +8,26 @@
     :copyright: (c) 2013-present by Abhinav Singh and contributors.
     :license: BSD, see LICENSE for more details.
 """
+
 from .proxy import entry_point
+import os
+from .indexer.airdropper import run_airdropper
 
 if __name__ == '__main__':
-    entry_point()
+    airdropper_mode = os.environ.get('AIRDROPPER_MODE', 'False').lower() in [1, 'true', 'True']
+    if airdropper_mode:
+        print("Will run in airdropper mode")
+        solana_url = os.environ['SOLANA_URL']
+        evm_loader_id = os.environ['EVM_LOADER']
+        faucet_url = os.environ['FAUCET_URL']
+        wrapper_whitelist = os.environ['INDEXER_ERC20_WRAPPER_WHITELIST'].split(',')
+        airdrop_amount = int(os.environ['AIRDROP_AMOUNT'])
+        log_level = os.environ['LOG_LEVEL']
+        run_airdropper(solana_url,
+                       evm_loader_id,
+                       faucet_url,
+                       wrapper_whitelist,
+                       airdrop_amount,
+                       log_level)
+    else:
+        entry_point()

proxy/indexer/airdropper.py

@@ -157,19 +157,3 @@ def run_airdropper(solana_url,
                             airdrop_amount,
                             log_level)
     airdropper.run()
-
-
-if __name__ == "__main__":
-    solana_url = os.environ.get('SOLANA_URL', 'http://localhost:8899')
-    evm_loader_id = os.environ.get('EVM_LOADER_ID', '53DfF883gyixYNXnM7s5xhdeyV8mVk9T4i2hGV9vG9io')
-    faucet_url = os.environ.get('FAUCET_URL', 'http://localhost:3333')
-    wrapper_whitelist = os.environ.get('INDEXER_ERC20_WRAPPER_WHITELIST', '').split(',')
-    airdrop_amount = os.environ.get('AIRDROP_AMOUNT', 0)
-    log_level = os.environ.get('LOG_LEVEL', 'INFO')
-
-    run_airdropper(solana_url,
-                   evm_loader_id,
-                   faucet_url,
-                   wrapper_whitelist,
-                   airdrop_amount,
-                   log_level)