import json

import lark_oapi as lark
from lark_oapi.api.im.v1 import *
from config import CONFIG

def main():
    # 创建client
    client = lark.Client.builder() \
        .app_id(CONFIG['app_id']) \
        .app_secret(CONFIG['app_secret']) \
        .log_level(lark.LogLevel.DEBUG) \
        .build()

    # 构造请求对象
    request: CreateMessageRequest = CreateMessageRequest.builder() \
        .receive_id_type("open_id") \
        .request_body(CreateMessageRequestBody.builder()
            .receive_id(CONFIG['open_id'])
            .msg_type("text")
            .content("{\"text\":\"请打开微信或者登录微信\"}")
            .build()) \
        .build()

    # 发起请求
    response: CreateMessageResponse = client.im.v1.message.create(request)

    # 处理失败返回
    if not response.success():
        lark.logger.error(
            f"client.im.v1.message.create failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
        return

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

if __name__ == "__main__":
    main()