add script to update lambda env vars, fix imports

Author: peterrauscher

config_lambdas.py

@@ -0,0 +1,36 @@
+import json
+import subprocess
+
+
+def get_lambdas():
+    functions = {}
+    res = subprocess.Popen(
+        ["aws", "lambda", "list-functions"],
+        stdout=subprocess.PIPE,
+    )
+    output = res.communicate()[0]
+    functions.update(json.loads(output))
+    return functions["Functions"]
+
+
+# for lambda_function in get_lambdas():
+#     function_name = lambda_function["FunctionName"]
+
+with open(".env", "r") as f:
+    env_vars = [
+        line for line in f.read().splitlines() if line.startswith("SOLESEARCH_")
+    ]
+    env = ",".join(env_vars)
+    subprocess.run(
+        [
+            "aws",
+            "lambda",
+            "update-function-configuration",
+            "--region",
+            "us-east-1",
+            "--function-name",
+            "solesearch-api",
+            "--environment",
+            f"Variables={{{env}}}",
+        ]
+    )

pyproject.toml

@@ -31,11 +31,11 @@ path = "src/api/__about__.py"
 allow-direct-references = true
 
 [tool.hatch.envs.default.scripts]
-start = "python -m api.server"
+start = "python -m src.api.main"
 
 [tool.hatch.envs.dev]
 pre-install-commands = ["pip install -e ../core"]
-scripts.start = "uvicorn src.api.main:app --reload"
+scripts.start = "python -m src.api.main"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/api"]

src/api/data/queries.py

@@ -4,7 +4,8 @@
 from bson import ObjectId
 
 from api.data.instance import DEFAULT_LIMIT, DEFAULT_OFFSET, db, sneakers
-from api.data.models import Audience, SortKey, SortOrder
+from api.data.models import SortKey, SortOrder
+from core.models.details import Audience
 
 
 def find_sneaker_by_id(id: str = ""):

src/api/main.py

@@ -5,10 +5,10 @@
 from mangum import Mangum
 from starlette.middleware.sessions import SessionMiddleware
 
-from api.routes import auth, sneakers
+if not os.environ.get("AWS_EXECUTION_ENV"):
+    load_dotenv(os.path.join(os.getcwd(), ".env"))
 
-dotenv_path = os.path.join(os.path.dirname(__file__), ".env")
-load_dotenv(dotenv_path)
+from api.routes import auth, sneakers
 
 app = FastAPI(
     redoc_url=None,
@@ -20,3 +20,8 @@
 app.include_router(sneakers.router)
 app.include_router(auth.router)
 handler = Mangum(app)
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app, host="localhost", port=8000)

src/api/routes/sneakers.py

@@ -3,8 +3,9 @@
 from fastapi import APIRouter, Query
 
 from api.data.instance import DEFAULT_LIMIT, DEFAULT_OFFSET
-from api.data.models import Audience, SortKey, SortOrder
+from api.data.models import SortKey, SortOrder
 from api.data.queries import find_sneaker_by_id, find_sneaker_by_sku, find_sneakers
+from core.models.details import Audience
 
 router = APIRouter(
     prefix="/sneakers",