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

app_id = config.get("bot.app_id")
app_secret = config.get("bot.app_secret")
user_id = config.get("user.open_id")

# 创建client
client = lark.Client.builder() \
    .app_id(app_id) \
    .app_secret(app_secret) \
    .log_level(lark.LogLevel.DEBUG) \
    .build()

def send_message(content: str):
    message = {
        "text": content
    }
    message = json.dumps(message, ensure_ascii=False)
    message = message.encode('unicode_escape').decode()
    # 构造请求对象
    request: CreateMessageRequest = CreateMessageRequest.builder() \
        .receive_id_type("open_id") \
        .request_body(CreateMessageRequestBody.builder()
                      .receive_id(user_id)
                      .msg_type("text")
                      .content(message)
                      .build()) \
        .build()

    # 发起请求
    client.im.v1.message.create(request)

if __name__ == "__main__":
    send_message("hello world")