# Code generated by Lark OpenAPI.

import lark_oapi as lark
from lark_oapi.api.corehr.v1 import *


def main():
    # 创建client
    client = lark.Client.builder() \
        .app_id(lark.APP_ID) \
        .app_secret(lark.APP_SECRET) \
        .log_level(lark.LogLevel.DEBUG) \
        .build()

    # 构造请求对象
    request: CreateEmploymentRequest = CreateEmploymentRequest.builder() \
        .client_token("12454646") \
        .request_body(EmploymentCreate.builder()
                      .seniority_date("2020-01-01")
                      .employee_number("1000000")
                      .effective_time("2020-01-01")
                      .expiration_time("2021-01-01")
                      .employment_type(Enum.builder().build())
                      .person_id("6919733936050406926")
                      .primary_employment(True)
                      .employment_status(Enum.builder().build())
                      .custom_fields([])
                      .work_email_list([])
                      .reason_for_offboarding(Enum.builder().build())
                      .ats_application_id("6838119494196871234")
                      .rehire(Enum.builder().build())
                      .rehire_employment_id("7051837122449425964")
                      .build()) \
        .build()

    # 发起请求
    response: CreateEmploymentResponse = client.corehr.v1.employment.create(request)

    # 处理失败返回
    if not response.success():
        lark.logger.error(
            f"client.corehr.v1.employment.create failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}")
        return

    # 处理业务结果
    lark.logger.info(lark.JSON.marshal(response.data, indent=4))


# 异步方式
async def amain():
    # 创建client
    client = lark.Client.builder() \
        .app_id(lark.APP_ID) \
        .app_secret(lark.APP_SECRET) \
        .log_level(lark.LogLevel.DEBUG) \
        .build()

    # 构造请求对象
    request: CreateEmploymentRequest = CreateEmploymentRequest.builder() \
        .client_token("12454646") \
        .request_body(EmploymentCreate.builder()
                      .seniority_date("2020-01-01")
                      .employee_number("1000000")
                      .effective_time("2020-01-01")
                      .expiration_time("2021-01-01")
                      .employment_type(Enum.builder().build())
                      .person_id("6919733936050406926")
                      .primary_employment(True)
                      .employment_status(Enum.builder().build())
                      .custom_fields([])
                      .work_email_list([])
                      .reason_for_offboarding(Enum.builder().build())
                      .ats_application_id("6838119494196871234")
                      .rehire(Enum.builder().build())
                      .rehire_employment_id("7051837122449425964")
                      .build()) \
        .build()

    # 发起请求
    response: CreateEmploymentResponse = await client.corehr.v1.employment.acreate(request)

    # 处理失败返回
    if not response.success():
        lark.logger.error(
            f"client.corehr.v1.employment.acreate failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}")
        return

    # 处理业务结果
    lark.logger.info(lark.JSON.marshal(response.data, indent=4))


if __name__ == "__main__":
    # asyncio.run(amain()) 异步方式
    main()
