iShotaBot/modules/post.py

from pyrogram import Client, filters
from defs.glover import admin
from pyrogram.types import Message

from defs.post import LofterPost
from init import bot


@bot.on_message(
    filters.incoming
    & filters.private
    & filters.user(admin)
    & filters.command(["lofter_post"])
)
async def lofter_post_command(client: Client, message: Message):
    """
    抓取 lofter 粮单
    """
    data = message.text.split(" ")
    offset = 0
    if len(data) < 2:
        await message.reply("参数错误")
        return
    elif len(data) == 2:
        url = data[1]
    else:
        url = data[1]
        offset = data[2]
    data = LofterPost(url, offset)
    client.loop.create_task(data.upload(message))