wag-managment-api-service-v.../service_app_test/test_application/migrate_old_data/company.py

31 lines
1.0 KiB
Python

from service_app_test.api_configs import BothAPIS
from service_app_test.bases import FilterObject
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,
)
print("response", response.text)
return