[Obcsapi v1] 微信使用 Remotely Save S3 兼容 发送到 Obsidian

# -*- coding=utf-8

# werobot cos-python-sdk-v5

from qcloud_cos import CosConfig
from qcloud_cos import CosS3Client
import werobot
import time
import requests

# 相关信息

token = "token"  # 自定义
APP_ID = "wxxxxxxxxxxxxxxxxx"  # 微信公众号测试号APP_ID
APP_SECRET = "xxxxxxxxxxxxxxxxxxxxxxxxxxxx"  # 微信公众号测试号APP_SECRET
secret_id = "xxxxxxxxxxxxxxxxxxxxxxxxxxxx"  # 腾讯云 secret_id
secret_key = "xxxxxxxxxxxxxxxxxxxxxxxxx"  # 腾讯云 secret_key
region = "ap-nanjing"  # 腾讯云 COS
cos_token = None  # 腾讯云 COS token
scheme = "https"  # 腾讯云 COS 访问模式
bucket = "test-0123456789"  # 腾讯云 Bucket

# werobot 配置

robot = werobot.WeRoBot(token=token)
robot.config["APP_ID"] = APP_ID
robot.config["APP_SECRET"] = APP_SECRET

# cos 配置

config = CosConfig(
    Region=region,
    SecretId=secret_id,
    SecretKey=secret_key,
    Token=cos_token,
    Scheme=scheme,
)
client = CosS3Client(config)
# rob_client = robot.client
# rob_client.create_menu({
#     "button":[{
#          "type": "click",
#          "name": "用户信息",
#          "key": "info"
#     }]
# })

# 菜单点击 输入信息 这个菜单影响快速输入信息 不方便

# @robot.key_click("info")
# def music(message):
#     return '你点击了“用户信息”按钮'

# 关注回复


@robot.subscribe
def subscribe(message):
    return "这是恐咖兵糖的测试公众号"


# 文字消息回复 put_object append_object
@robot.text
def text_reply(message):
    yaml_str = message.content
    yaml = (
        "---\ncontent: " + yaml_str.replace("\n", "<br>") + "\n---\n"
    )  # 将内容存入 yaml 字段 方便 dataview 使用 替换换行防止出错
    btyes = bytes(yaml + message.content, encoding="utf8")
    name = time.strftime("%Y%m%d%H%M%S", time.localtime())
    response = client.put_object(
        Bucket=bucket,
        Body=btyes,
        Key="微信收藏/" + name + ".md",
    )
    return response["ETag"]


# 图片消息
@robot.image
def image_reply(message):
    # btyes = bytes("![]("+message.img+")", encoding = "utf8") # 直接使用腾讯链接显示图片
    file = requests.get(message.img)
    name = time.strftime("%Y%m%d%H%M%S", time.localtime())
    response1 = client.put_object(
        Bucket=bucket,
        Body=file.content,
        Key="微信收藏/附件/" + name + ".jpg",
    )
    conent = "![[附件/" + name + ".jpg]]"
    yaml = "---\ncontent: " + conent + "\n---\n"
    btyes = bytes(yaml + conent, encoding="utf8")
    response2 = client.put_object(
        Bucket=bucket,
        Body=btyes,
        Key="微信收藏/" + name + ".md",
    )
    return response1["ETag"] + response2["ETag"]


@robot.error_page
def make_error_page(url):
    return "404"


@robot.handler
def error_message(message):
    return "不支持的消息类型"


robot.config["HOST"] = "0.0.0.0"
robot.config["PORT"] = "8008"
robot.run()