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