events updated

service_app_test/test_application/migrate_old_data/company.py

@@ -1,30 +1,27 @@
 from service_app_test.api_configs import BothAPIS
-from service_app_test.bases import FilterObject
+from service_app_test.test_application.migrate_old_data.reader_and_alchemy_bulk_actions import (
+    read_json_file,
+)
+from api_validations.validations_request import InsertCompany
+
+
+requester_dict = lambda data: {"endpoint": "/company/create", "data": data}
+
+
+def get_company_from_json():
+    read_files_json, with_pydantic = read_json_file(json_file="companies"), []
+    read_files = read_files_json.get("companies")
+    for row in read_files:
+        pydantic_row = InsertCompany(**row)
+        with_pydantic.append(pydantic_row.model_dump())
+    if not with_pydantic:
+        raise Exception("No data found")
+    return with_pydantic
 
 
 def migrate_company(requester: BothAPIS):
-    filter_object = FilterObject(
-        page=1,
-        size=100,
-    )
-    response = requester.wag_api.post(
-        endpoint="/company/list",
-        data=filter_object.dump(),
-    )
-    response_json = response.json()
-    print("response_json", response_json)
-    response_datas = response_json["data"]
-    for response_data in response_datas:
-        new_response_data = dict()
-        for key, value in dict(response_data).items():
-            if value is not None and not str(value) == "None":
-                new_response_data[key] = response_data[key]
-        new_response_data.pop("uu_id", None)
-        print("new_response_data", new_response_data)
-        new_response_data["company_tag"] = response_data["formal_name"]
-        response = requester.local_api.post(
-            endpoint="/company/create",
-            data=new_response_data,
-        )
+    for response_data in get_company_from_json():
+        response_data["company_tag"] = response_data["formal_name"]
+        response = requester.local_api.post(**requester_dict(data=response_data))
         print("response", response.text)
     return