some little changes

This commit is contained in:
Pyhtagodzilla 2025-07-25 22:13:16 +08:00
parent 29a96bb1ff
commit a30c3699a1
3 changed files with 26 additions and 16 deletions

2
.gitignore vendored
View File

@ -1,4 +1,4 @@
tests
.env
src/data/pics
src/data
src/logs

View File

@ -1,11 +1,24 @@
import time
import os
from dotenv import load_dotenv
from ncatbot.core import BotClient
from ncatbot.core import BotClient, GroupMessage, PrivateMessage
from ncatbot.utils import get_log
from sqlalchemy.orm import defer
load_dotenv(dotenv_path=r"../.env")
bot = BotClient()
_log = get_log()
@bot.group_event()
async def on_receive_group_message(msg: GroupMessage):
_log.info(f"The message's meta data is: {str(msg)}")
_log.info(f"The message's raw message is: {msg.raw_message}")
@bot.private_event()
async def on_receive_private_event(msg: PrivateMessage):
_log.info(f"The message's meta data is: {str(msg)}")
_log.info(f"The message's raw message is: {msg.raw_message}")
bot.run(
bt_uin=os.getenv("BOT_ACCOUNT"),

View File

@ -14,15 +14,16 @@ _log = get_log()
class GetFile(BasePlugin):
name = "GetFile"
version = "0.0.1 beta"
author = "pythagodzilla"
info = "A plugin to get images and videos"
@bot.private_event()
async def on_receive_images_private_event(self, msg: PrivateMessage):
meta_message = str(msg)
_log.info(f"The message's meta data is: {meta_message}")
_log.info(f"The message's raw message is: {msg.raw_message}")
if "[CQ:image," in msg.raw_message:
divided_message = msg.raw_message.split(',')
message = msg.raw_message[1: -1]
if message.startswith("CQ:image,"):
divided_message = message.split(',')
for item in divided_message:
if "file=" in item:
@ -35,17 +36,17 @@ class GetFile(BasePlugin):
response = requests.get(real_url)
img = response.content
with open(f"D:/Projects/memebot/src/data/pics/{file_name}", 'wb') as f:
with open(f"../data/pics/{file_name}", 'wb') as f:
f.write(img)
@bot.group_event()
async def on_receive_images_group_event(self, msg: GroupMessage):
_log.info(f"The message's meta data is: {str(msg)}")
_log.info(f"The message's raw message is: {msg.raw_message}")
# if "[CQ:image," in msg.raw_message:
if msg.raw_message.startswith("[CQ:image,"):
divided_message = msg.raw_message.split(',')
message = msg.raw_message[1: -1]
if message.startswith("CQ:image,"):
divided_message = message.split(',')
for item in divided_message:
# if "file=" in item:
@ -64,8 +65,6 @@ class GetFile(BasePlugin):
@bot.group_event()
async def on_receive_videos_group_event(self, msg: GroupMessage):
_log.info(f"The message's meta data is: {str(msg)}")
_log.info(f"The message's raw message is: {msg.raw_message}")
message = msg.raw_message[1: -1]
@ -87,8 +86,6 @@ class GetFile(BasePlugin):
@bot.private_event()
async def on_receive_videos_private_event(self, msg: PrivateMessage):
_log.info(f"The message's meta data is: {str(msg)}")
_log.info(f"The message's raw message is: {msg.raw_message}")
message = msg.raw_message[1: -1]