Update createlookalike.py

createlookalike.py

@@ -1,4 +1,3 @@
-import asyncio
 import tempfile as tfile
 from datetime import datetime
 from urllib.request import urlopen
@@ -19,10 +18,6 @@ from sklearn.decomposition import PCA
 from scipy.spatial import distance
 from collections import OrderedDict
 
-# from remove import remove_files
-#
-# from generate_csv_file import generate_csv_files
-# from load_data import load_data, get_shops
 from consts import API_KEY
 from schemas import Shop
 
@@ -144,35 +139,49 @@ async def calculate_shop(shop: Shop, feat_extractor) -> None:
                                    }
 
                     # print(design_dict)
-                    if await push_home(design_dict, im, shop):
+                    if await push_home(design_dict, shop):
                         pass
                     else:
                         print("error sending recommendations")
 
-
-                    # idx_closest = sorted(range(len(similar_idx)), key=lambda k: similar_idx[k])
-
-                    # design_json.update(design_dict)
-                    # print(idx_closest)
+                # if calculation is ready send update home
+                if await update_calculation_date(shop):
+                    pass
+                else:
+                    print("error sending shop calculation update")
 
             except Exception as e:
                 print("could not create json with look-a-like for shop:", shop.id, e)
 
         print(f"calculation for {shop.id} ended at  {datetime.now()}")
 
-        # return {'shop_id': shop.id, 'start_time': start, 'end_time': end, 'designs': design_json}
 
 
-async def push_home(design_dict, im, shop):
+async def push_home(design_dict, shop):
     headers: dict[str, str] = {
         "Authorization": API_KEY,
         "Content-type": "application/json",
     }
-
-    print(headers)
     try:
+        url=f"{shop.webhook_url}/fill_recommendations
+        response = requests.post(url, json=design_dict, headers=headers)
+        response.raise_for_status()
+        return True
 
-        response = requests.post(shop.webhook_url, json=design_dict, headers=headers)
+    except Exception as e:
+        print(e)
+        return False
+
+
+async def update_calculation_date(shop):
+    headers: dict[str, str] = {
+        "Authorization": API_KEY,
+        "Content-type": "application/json",
+    }
+    try:
+        url = f"{shop.webhook_url}/shop_updated"
+        data = {"shop_id": shop.id}
+        response = requests.post(url, json=data, headers=headers)
         response.raise_for_status()
         return True