|-- App
|   |-- exts.py
|   |-- __init__.py
|   |-- models.py
|   |-- script.py
|   |-- settings.py
|   `-- views
|       |-- goCqhttp.py
|       |-- __init__.py
|-- app.py

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "models.py"
__time__ = "2022/9/11 19:56"

from App.exts import db

class Group(db.Model):  # 創(chuàng)建一個群的表
    __tablename__ = "group"
    id = db.Column(db.INTEGER, primary_key=True, autoincrement=True)
    name = db.Column(db.String(200), nullable=True)
    qqId = db.Column(db.String(20), nullable=False, unique=True, index=True)  
    isDetect = db.Column(db.BOOLEAN, default=True)  # 是否開啟，進行邏輯刪除
    auth = db.Column(db.SmallInteger, default=0)  # 在群里面的的地位，默認為群成員，1為管理員，2為群主

    def __init__(self, name, qqId):
        self.name = name
        self.qqId = qqId

class GroupAuthority(db.Model):  # 創(chuàng)建一個權(quán)限管理表
    __tablename__ = "groupAuth"
    id = db.Column(db.INTEGER, primary_key=True, autoincrement=True)
    chat = db.Column(db.INTEGER, default=0, nullable=False)  # 是否開啟智能聊天的功能
    welcome = db.Column(db.INTEGER, default=1, nullable=False)  # 是否開啟新成員入群歡迎的功能
    banTalk = db.Column(db.INTEGER, default=0)  # 群禁言功能，以及消息撤回功能
    click = db.Column(db.INTEGER, default=1)  # 戳一戳功能，默認開啟
    smallFunction = db.Column(db.INTEGER, default=1)  # 是否開啟小功能，如疫情數(shù)據(jù)查詢等
    dailyBrief = db.Column(db.INTEGER, default=0)  # 是否開啟每日簡報功能
    groupId = db.Column(db.INTEGER, db.ForeignKey("group.id", ondelete="CASCADE"))  # 外鍵約束，同時進行級聯(lián)刪除
    auth2group = db.relationship("Group", backref=db.backref("group2auth"))  # 使用代理

    def __init__(self, is_privade=False, chat=None, welcome=None, bantalk=None, click=None, smallFunction=None,
                 dailyBrief=None):
        if is_privade:
            self.chat = chat
            self.welcome = welcome
            self.banTalk = bantalk
            self.click = click
            self.smallFunction = smallFunction
            self.dailyBrief = dailyBrief

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "exts.py"
__time__ = "2022/9/11 23:39"

from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

db = SQLAlchemy()  # 操作數(shù)據(jù)庫
def init_exts(app):
    db.init_app(app)
    Migrate().init_app(app, db)  # 使用app初始化Migrate
    app.config["db"] = db

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "settings.py"
__time__ = "2022/9/11 19:17"

class Config:
    """基礎(chǔ)的配置字"""
    SQLALCHEMY_DATABASE_URI = "mysql+pymysql://root:123456@127.0.0.1:3306/bot?charset=utf8"  
    SQLALCHEMY_TRACK_MODIFICATIONS = False
    SQLALCHEMY_ECHO = True
    SUPER_USER = ["3500515050", ]  # 超級用戶，其可以私發(fā)消息給機器人，管理機器人對所有群的權(quán)限，當然也可以存儲到數(shù)據(jù)庫中，這里為了方便，存儲在配置文件中，大家可根據(jù)這篇文章執(zhí)行實現(xiàn)
    ADMIN = 3500515050  # 開發(fā)者

class Help:
    """幫助文檔"""
    ADMIN_HELP = """開發(fā)者命令提示：
1 添加群:\n/admin:add QQ群號 QQ群名
2. 刪除功能:\n/admin:delete QQ群號
3 查找功能:\n/admin:get QQ群號
4. 獲取所有群:\n/admin:show
5. 修改權(quán)限命令:\n/admin:changeAuth QQ群號 |聊天功能|入群歡迎|管理群|戳一戳|拓展功能|定時功能|(比如110011)\n
如果還是有問題，請與開發(fā)人員聯(lián)系哦！"""
    GROUP_ADMIN = """[CQ:at,qq=%d]命令提示：
1. 查看群權(quán)限:\n/admin:get
2. 修改群權(quán)限：\n/admin:change |聊天功能|入群歡迎|管理群|戳一戳|拓展功能|定時功能|(比如#admin:change# 110011)\n
如果還是有問題，請與開發(fā)人員聯(lián)系哦！"""

class ProductConfig(Config, Mes, Url):
    """生產(chǎn)環(huán)境配置"""
    pass
class DevelopConfig(Config, Mes, Url):
    """開發(fā)環(huán)境配置"""
    DEBUG = True
envs = {
    "product": ProductConfig,
    "develop": DevelopConfig
}

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "__init__.py.py"
__time__ = "2022/9/11 19:30"

from App.views.goCqhttp import AcceptMes
from flask_restful import Api
def init_app(app):
    api = Api(app)
    api.add_resource(AcceptMes, "/", endpoint="index")

# !/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "goCqhttp.py"
__time__ = "2022/9/11 19:57"

from flask_restful import Resource
from flask import request
import asyncio
from App.events.private import PriChatMes
from App.events.group import GroupChatMes
from App.events.groupAndPri import GroupAndPri
from flask import current_app
from App.models import Group
from App.events.groupAndPri import Command
from App.script.requests_tools import Sender
class AcceptMes(Resource):

    def post(self):
		pass  # 后面主要是通過這個接口來對消息的分發(fā)

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "__init__.py.py"
__time__ = "2022/9/11 14:25"

from flask import Flask
from App.settings import envs
from App.views import init_app
from App.exts import init_exts

def create_app(env):
    app = Flask(__name__)
    app.config.from_object(envs.get(env))
    init_exts(app)
    init_app(app)  # 將路由傳入app中
    return app

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "app.py"
__time__ = "2022/9/11 19:17"

from App import create_app
from flask_script import Manager

app = create_app("develop")
manage = Manager(app)

if __name__ == '__main__':
    manage.run()

flask db init  # 初始化操作
flask db migrate  # 遷移操作
flask db upgrade  # 更新操作

from flask import current_app
from App.models import Group, GroupAuthority
from base64 import b64decode  # 對中文進行編碼
async def add(gid, nick):
    # 傳入QQ群號和qq群的昵稱就可以添加了
    session = current_app.config["db"].session
    data = session.query(Group).filter_by(qqId=gid).first()  # 首先查看一下數(shù)據(jù)庫中是否收錄了這個群
    if data is None:  # 如果沒有則添加，同時，也檢查其是否有邏輯刪除
        g = Group(nick, gid)  # 創(chuàng)建一個群
        g.group2auth = [GroupAuthority()]  # 給群賦予默認的權(quán)限
        session.add(g)  # 添加到數(shù)據(jù)庫中
        session.commit()  # 提交事務(wù)
        return {  
            "status": 200
        }  # 返回狀態(tài)碼
   	# 如果群存在，直接修改群名，以及使其可以檢測到
    data.isDetect = True
    data.name = nick
    session.commit()
    return {
        "status": 300,
        "error": "該群號已存在！",
    }  

async def delete(gid):
    db = current_app.config["db"]
    session = db.session
    group = session.query(Group).filter(db.and_(Group.qqId == gid, Group.isDetect)).first()
    if group:
        group.isDetect = False
        session.commit()
        return "刪除成功"
    return "該群不存在"

async def show():
    session = current_app.config["db"].session
    data = [f"|_ {b64decode(i.name).decode()} _|_ {i.qqId} _|_ {'yes' if i.isDetect else 'no'} _|" for i in
            session.query(Group).all()]
    return "\n".join(data)

async def get(gid):
    db = current_app.config["db"]
    session = db.session
    data = session.query(Group).filter(db.and_(Group.qqId == gid, Group.isDetect)).first()
    if data is None:
        return "在該群不支持，請與開發(fā)者聯(lián)系！"
    name = b64decode(data.name).decode()  # qq群名稱
    chat = "1. 已開啟聊天功能" if data.group2auth[0].chat else "1. 未開啟聊天功能"  # 有沒有開啟智能聊天的功能
    welcome = "2. 已開啟入群歡迎功能" if data.group2auth[0].welcome else "2. 未開啟入群歡迎功能"  # 是否開啟入群歡迎的功能
    banTalk = "3. 已開啟管理群功能" if data.group2auth[0].banTalk else "3. 未開啟管理群功能"  # 是否開啟管理員的功能
    click = "4. 已開啟戳一戳功能" if data.group2auth[0].click else "4. 未開啟戳一戳功能"  # 戳一戳功能
    smallFunction = "5. 已開啟拓展功能" if data.group2auth[0].smallFunction else "5. 未開啟拓展功能"  # 是否開啟小功能
    dailyBrief = "6. 已開啟定時發(fā)消息功能" if data.group2auth[0].dailyBrief else "6. 未開啟定時發(fā)消息功能"  # 是否開啟每日自動播報的功能
    return f"{name}\n{chat}\n{welcome}\n{banTalk}\n{click}\n{smallFunction}\n{dailyBrief}\n"

async def changeAuth(gid, data, ty):
    """
    傳入群號，要修改的數(shù)據(jù)其為6位數(shù)字，如："011011"，每一個數(shù)字代表一個權(quán)限，1為真|0為假
 	最后傳入的是調(diào)用這個函數(shù)的類型，是群管理員還是超級管理員
    """
    db = current_app.config["db"]
    session = db.session  # 獲取數(shù)據(jù)庫連接
    try:
        for i in data:
            if int(i) not in [0, 1] or len(data) != 6:
                raise ValueError
        data = (int(i) for i in data)  # 獲取到對應(yīng)的權(quán)限數(shù)字
        group = session.query(Group).filter(db.and_(Group.qqId == gid, Group.isDetect)).first()  # 獲取對象
        if group:
            group.group2auth = [GroupAuthority(True, *data)]  # 使用元組拆包的方式來傳參
            session.commit()  # 提交數(shù)據(jù)
            _ = await Admin.get(gid)  # 獲取到群里面的權(quán)限
            ret = f"[CQ:at,qq=%d]設(shè)置成功，設(shè)置后的權(quán)限為：\n{_}" if ty == "group" else f"設(shè)置成功，設(shè)置后的權(quán)限為：\n{_}"  
        else: 
            ret = "該群不支持機器人"
    except Exception as e:
        ret = "[CQ:at,qq=%d]設(shè)置失敗，請查看幫助文檔！" if ty == "group" else "設(shè)置失敗，請查看幫助文檔！"
    return ret

# !/usr/bin/python3
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "goCqhttp.py"
__time__ = "2022/9/11 19:57"

from flask_restful import Resource
from flask import request
import asyncio
from flask import current_app
from App.models import Group
from App.script import *
from base64 import b64encode

# 用于發(fā)送信息的函數(shù)
async def send(id, message, ty):
    """
        用于發(fā)送消息的函數(shù)
        :param id: qq號，或者qq群號
        :param message: 發(fā)送的消息
        :param ty: 傳入的
        :return: None
        """
    async with httpx.AsyncClient(base_url="http://127.0.0.1:5700") as client:
        # 如果發(fā)送的為私聊消息
        if ty == "group":
            params = {
                "group_id": id,
                "message": message,
            }
            await client.get("/send_group_msg", params=params)
            elif ty == "private":
                params = {
                    "user_id": id,
                    "message": message,
                }
                await client.get("/send_private_msg", params=params)
class AcceptMes(Resource):

    def post(self):
        # 這里對消息進行分發(fā)，暫時先設(shè)置一個簡單的分發(fā)
        _ = request.json
        if _.get("message_type") == "private":  # 說明有好友發(fā)送信息過來
            uid = _["sender"]["user_id"]  # 獲取發(fā)信息的好友qq號
        	message = _["raw_message"]  # 獲取發(fā)送過來的消息
            if message.startswith("/admin:") and str(uid) in current_app.config["SUPER_USER"]:
                # 其為超級用戶的命令，通過正則解析出命令，并且做出回應(yīng)，那么我們就需要創(chuàng)建一個函數(shù)，來解析我們的命令，這個函數(shù)我們后面來實現(xiàn)
            	asyncio.run(super_command(uid, message))  # 處理私聊的信息

        elif _.get("message_type") == "group" and message.startswith("/admin:") and resp["sender"]["role"] in ["owner", "admin"]:
            # 獲取群消息，并且，其為群主或者管理員發(fā)送的消息的話
            db = current_app.config["db"]
            session = db.session
            group = session.query(Group).filter(db.and_(Group.qqId == _["group_id"], Group.isDetect)).first()
            if not group:
                # 如果群里面不支持機器人的話，直接返回
                return
            uid = _["sender"]["user_id"]  # 獲取發(fā)信息的好友qq號
            message = _.get("raw_message")
            gid = resp["group_id"]  # 獲取群號
            asyncio.run(admin_command(uid, gid, message))  # 這里也是先用一個函數(shù)名來占位，我們后面來實現(xiàn)這個函數(shù)

async def super_command(uid, message):  # 處理開發(fā)者的命令
    com_1 = re.findall(r"/admin:(.*)", message.split()[0])[0]
    try:
        com_2 = message.split()[1]  # qq群號
    except Exception as e:
        com_2 = None
    if com_1 == "add" and com_2.isdecimal():  # 這說明其為添加qq群
        com_3 = message.split()[2]  # qq昵稱
        ret = await add(com_2, b64encode(com_3.encode()))
        if ret["status"] == 200:
            await send(uid, f"{com_2}添加成功，該群名為{com_3}", "private")
        else:
            await send(uid, ret["error"], "private")
    elif com_1 == "get" and com_2.isdecimal():  # 獲取指定的qq群的權(quán)限信息
        ret = await get(com_2)
        await send(uid, ret, ty="private")
    elif com_1 == "delete" and com_2.isdecimal():  # 刪除qq群
        ret = await delete(com_2)
        await send(uid, ret, ty="private")
    elif com_1 == "changeAuth" and com_2.isdecimal():  # 修改權(quán)限
        data = list(message.split()[2])
        ret = await changeAuth(com_2, data, "private")
        await send(uid, ret, "private")
    elif com_1 == "show":  # 展示所有群信息
        ret = await show()
        await send(uid, ret, "private")
    else:  # 如果都不是的話，發(fā)送幫助文檔
        ret = current_app.config["ADMIN_HELP"]
        await send(uid, ret, "private")

async def admin_command(uid, gid, message):  # 群管理員命令
    com_1 = re.findall(r"/admin:(.*)", message.split()[0])[0]  # 把命令解析出來
    if com_1 == "get":  # 獲取本群的權(quán)限信息
        ret = await get(gid)
        await send(gid, ret, ty="group")
    elif com_1 == "change":  # 修改權(quán)限信息
            data = list(message.split()[1])
        ret = await changeAuth(gid, data, "group")
        await send(gid, ret % uid, "group")
    else:  # 默認為幫助文檔
        ret = current_app.config["GROUP_ADMIN"] % uid
        await send(gid, ret, "group")