migrator functions updated

service_app_test/test_application/migrate_old_data/company.py

@@ -6,26 +6,25 @@ def migrate_company(requester: BothAPIS):
     filter_object = FilterObject(
         page=1,
         size=100,
-        order_field="id",
-        order_type="asc",
     )
     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']
+    print("response_json", response_json)
+    response_datas = response_json["data"]
     for response_data in response_datas:
-        print('response_data', response_data)
-        response_data['active'] = True
-        response_data['deleted'] = False
-        response_data['is_confirmed'] = True
-        response_data['expiry_starts'] = None
-        response_data['expiry_ends'] = None
+        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=response_data,
+            data=new_response_data,
         )
-        print('response', response.text)
+        print("response", response.text)
     return