feat: allow reading of token from env

Author: opensaucerer

lighthouse/config.py

@@ -4,6 +4,6 @@
 class Config:
     """Config class for lighthouse"""
 
-    lighthouse_api = "https://api.lighthouse.storage"
+    lighthouse_api = "http://13.234.35.183:5050"  # "https://api.lighthouse.storage"
     lighthouse_node = "https://node.lighthouse.storage"
     lighthouse_bls_node = "https://encryption.lighthouse.storage"

main.py

@@ -1,12 +1,17 @@
 #!/usr/bin/env python3
 
+import os
 import lighthouse.deploy as d
 from lighthouse import types as t
 
 
 class Lighthouse:
-    def __init__(self, token: str):
-        self.token = token
+    def __init__(self, token: str = ""):
+        self.token = token or os.environ.get("LIGHTHOUSE_TOKEN", "")
+        if not self.token:
+            raise Exception(
+                "No token provided: Please provide a token or set the LIGHTHOUSE_TOKEN environment variable"
+            )
 
     def deploy(self, source: str) -> t.Deploy:
         """

tests/test_deploy.py

@@ -13,14 +13,14 @@ def setUp(self) -> None:
 
     def test_deploy_file(self):
         """test deploy function"""
-        l = Lighthouse(os.environ.get("LH_TOKEN", ""))
+        l = Lighthouse()  # will use env var
         res = l.deploy("tests/testdir/testfile.txt")
         self.assertNotEqual(res.get("data"), None, "data is None")
         self.assertNotEqual(res.get("data").get("Hash"), None, "data is None")
 
     def test_deploy_dir(self):
         """test deploy function"""
-        l = Lighthouse(os.environ["LH_TOKEN"])
+        l = Lighthouse(os.environ["LIGHTHOUSE_TOKEN"])
         res = l.deploy("tests/testdir/")
         self.assertNotEqual(res.get("data"), None, "data is None")
         self.assertIsInstance(res.get("data"), dict, "data is a dict")