[Obcsapi v2] 微信到 Obsidian 2.0

# -*- coding=utf-8

# README
# 需要安装的依赖
# werobot cos-python-sdk-v5
# 可用 Alist 或其他可挂载 S3 的开源网盘 实现web编辑.md

from qcloud_cos import CosConfig
from qcloud_cos import CosS3Client
import werobot
import time
import get_add as cosga

# 相关信息

token = "token"  # 自定义
APP_ID = "wxxxxxxxxxxxxxxxxx"  # 微信公众号测试号APP_ID
APP_SECRET = "xxxxxxxxxxxxxxxxxxxxxxxxxxxx"  # 微信公众号测试号APP_SECRET


# werobot 配置

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

rob_client = robot.client
rob_client.delete_menu()
# 
# rob_client.create_menu({
#     "button":[{
#          "type": "click",
#          "name": "今日",
#          "key": "info"
#     }]
# })

# message.source 为 OpenID 。获取->测试号二维码->用户列表（最多100个）->微信号

@robot.key_click("info")
def menu_click(message):
    if (message.source != "o6_bmjrPTlm6_2sgVt7hMZOPfL2M"):
        return '你不是恐咖兵糖'
    return cosga.get_daily_today()

# 关注回复


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


# 文字消息回复 put_object append_object
@robot.text
def text_reply(message):
    if (message.source != "o6_bmjrPTlm6_2sgVt7hMZOPfL2M"):
        return '你不是恐咖兵糖'
    str = cosga.add_memos_in_daily(message.content) # ETag
    return "📩 已保存"


# 图片消息
@robot.image
def image_reply(message):
    if (message.source != "o6_bmjrPTlm6_2sgVt7hMZOPfL2M"):
        return '你不是恐咖兵糖'
    now_key = cosga.save_asset(message.img)
    str = cosga.add_memos_in_daily("![]("+now_key+")") # ETag
    return  "📩 已保存"


@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()

# -*- coding=utf-8

# 需要安装的依赖
# werobot cos-python-sdk-v5

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

# 相关信息
secret_id = "xxxxxxxxxxxxxxxxxxxxxxxxxxxx"  # 腾讯云 secret_i
secret_key = "xxxxxxxxxxxxxxxxxxxxxxxxx"  # 腾讯云 secret_key
region = "ap-nanjing"  # 腾讯云 COS
cos_token = None  # 腾讯云 COS token
scheme = "https"  # 腾讯云 COS 访问模式
bucket = "test-0123456789"  # 腾讯云 Bucket
config = CosConfig(
    Region=region,
    SecretId=secret_id,
    SecretKey=secret_key,
    Token=cos_token,
    Scheme=scheme,
)

client = CosS3Client(config)
   

# name = time.strftime("%Y%m%d%H%M%S", time.localtime())
def add_memos_in_daily(text):
    # 生成ObjectKey => 日志/2022-08-08.md
    daily_note_dir_path = "日志/"
    today_daily_file_key = daily_note_dir_path + time.strftime("%Y-%m-%d", time.localtime()) + ".md"
    todo = "todo"
    if todo in text:
        message = time.strftime("\n- [ ] %H:%M ", time.localtime()) + text
    else:
        message = time.strftime("\n- %H:%M ", time.localtime()) + text
    response3 = client.object_exists(
        Bucket=bucket,
        Key=today_daily_file_key
     )
    # 判断本日日志是否存在
    if(response3==True):
         # 存在 读取文件
        response2 = client.get_object(
            Bucket=bucket,
            Key=today_daily_file_key
        )
        str1=response2['Body'].get_raw_stream().read().decode('utf-8')
        btyes = bytes(str1+ message, encoding="utf8")
    else:
        btyes = bytes(message, encoding="utf8")


    #btyes = bytes(str1+"- 20:01 somgthing", encoding="utf8")	
    # name = time.strftime("%Y%m%d%H%M%S", time.localtime())
    response_sum = client.put_object(
            Bucket=bucket,
            Body=btyes,
            Key=today_daily_file_key
        )
    return response_sum["ETag"]

def save_asset(file):
    # 生成ObjectKey => 日志/附件/202208/20220808131104.jpg 
    file = requests.get(file).content # wechat file is url 为了防止过期，下载下来。
    # file = bytes("![]("+file+")", encoding = "utf8") # 直接使用腾讯微信链接显示图片
    daily_asset_dir_path = "日志/附件/" + time.strftime("%Y%m",time.localtime()) + "/" 
    today_daily_asset_file_key = daily_asset_dir_path + time.strftime("%Y%m%d%H%M%S", time.localtime()) + ".jpg"
    now_key = today_daily_asset_file_key
    response_file = client.put_object(
        Bucket=bucket,
        Body=file,
        Key=now_key
    )
    print(response_file["ETag"])
    return now_key

# 已弃用 原因: 超过一定长度后微信显示服务错误
def get_daily_today():
    # 生成ObjectKey => 日志/2022-08-08.md
    daily_note_dir_path = "日志/"
    today_daily_file_key = daily_note_dir_path + time.strftime("%Y-%m-%d", time.localtime()) + ".md"
    response3 = client.object_exists(
        Bucket=bucket,
        Key=today_daily_file_key
    )
    if(response3==True):
    # 存在 读取文件
        response2 = client.get_object(
            Bucket=bucket,
            Key=today_daily_file_key
        )
        str1=response2['Body'].get_raw_stream().read().decode('utf-8')
        return str1
    else:
        return '今日无日志'