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 tests
.env .env
src/data/pics src/data
src/logs src/logs

View File

@ -1,11 +1,24 @@
import time import time
import os import os
from dotenv import load_dotenv 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") load_dotenv(dotenv_path=r"../.env")
bot = BotClient() 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( bot.run(
bt_uin=os.getenv("BOT_ACCOUNT"), bt_uin=os.getenv("BOT_ACCOUNT"),

View File

@ -14,15 +14,16 @@ _log = get_log()
class GetFile(BasePlugin): class GetFile(BasePlugin):
name = "GetFile" name = "GetFile"
version = "0.0.1 beta" version = "0.0.1 beta"
author = "pythagodzilla"
info = "A plugin to get images and videos"
@bot.private_event() @bot.private_event()
async def on_receive_images_private_event(self, msg: PrivateMessage): 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: message = msg.raw_message[1: -1]
divided_message = msg.raw_message.split(',')
if message.startswith("CQ:image,"):
divided_message = message.split(',')
for item in divided_message: for item in divided_message:
if "file=" in item: if "file=" in item:
@ -35,17 +36,17 @@ class GetFile(BasePlugin):
response = requests.get(real_url) response = requests.get(real_url)
img = response.content 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) f.write(img)
@bot.group_event() @bot.group_event()
async def on_receive_images_group_event(self, msg: GroupMessage): 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 "[CQ:image," in msg.raw_message:
if msg.raw_message.startswith("[CQ:image,"): message = msg.raw_message[1: -1]
divided_message = msg.raw_message.split(',')
if message.startswith("CQ:image,"):
divided_message = message.split(',')
for item in divided_message: for item in divided_message:
# if "file=" in item: # if "file=" in item:
@ -64,8 +65,6 @@ class GetFile(BasePlugin):
@bot.group_event() @bot.group_event()
async def on_receive_videos_group_event(self, msg: GroupMessage): 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] message = msg.raw_message[1: -1]
@ -87,8 +86,6 @@ class GetFile(BasePlugin):
@bot.private_event() @bot.private_event()
async def on_receive_videos_private_event(self, msg: PrivateMessage): 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] message = msg.raw_message[1: -1]