feat: add wells endpoint and health check to transfer service

Author: jirhiker

transfers/entrypoint.py

@@ -16,16 +16,22 @@
 
 from fastapi import FastAPI
 
-from transfers.transfer import main_transfer
+from core.dependencies import session_dependency
+from transfers.well_transfer import transfer_wells
 
 app = FastAPI(title="Transfer Service")
 
 
-@app.get("/trigger")
-async def trigger():
-    main_transfer()
+@app.get("/wells")
+async def wells(session: session_dependency):
 
-    return
+    results = transfer_wells(session, limit=50)
+    return results
+
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}
 
 
 # ============= EOF =============================================

transfers/well_transfer.py

@@ -46,7 +46,8 @@ def transfer_wells(session, limit=None):
 
     n = len(wdf)
     start_time = time.time()
-
+    results = {"n": n, }
+    made_things = []
     for i, row in enumerate(wdf.itertuples()):
         if limit and i >= limit:
             print("Reached limit of", limit, "rows. Stopping migration.")
@@ -107,6 +108,10 @@ def transfer_wells(session, limit=None):
         assoc.location = location
         assoc.thing = well
         session.add(assoc)
+        made_things.append(row.PointID)
+
+    results["made_things"] = made_things
+    return results
 
 
 def transfer_wellscreens(session, limit=None):