update fn

ericz1803 · Aug 11, 2023 · ff47327 · ff47327
1 parent 865a5ca
commit ff47327
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 9 deletions.
diff --git a/serverless.yml b/serverless.yml
@@ -12,7 +12,7 @@ package:
 
 functions:
   run-rebalance:
-    handler: src/lambda_function.lambda_handler
+    handler: src/main.main
 
 plugins:
   - serverless-python-requirements

diff --git a/src/lambda_function.py b/src/lambda_function.py
diff --git a/src/main.py b/src/main.py
@@ -94,10 +94,15 @@ def rebalance(self):
             )
 
         print(f"Attempted to buy ${total_spent} worth of stocks.") 
-
+
+def getTrader():
+    global trader
+    if not trader:
+        trader = Trader()
+    return trader
 
 def main():
-    trader = Trader()
+    trader = getTrader()
     trader.rebalance()
 
 if __name__ == '__main__':