go-cqhttp权限管理系统的实现代码_Golang_

|-- 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): # 创建一个群的表 __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): # 创建一个权限管理表 __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) # 是否开启小功能，如疫情数据查询等 dailyBrief = db.Column(db.INTEGER, default=0) # 是否开启每日简报功能 groupId = db.Column(db.INTEGER, db.ForeignKey("group.id", ondelete="CASCADE")) # 外键约束，同时进行级联删除 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() # 操作数据库 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: """基础的配置字""" 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", ] # 超级用户，其可以私发消息给机器人，管理机器人对所有群的权限，当然也可以存储到数据库中，这里为了方便，存储在配置文件中，大家可根据这篇文章执行实现 ADMIN = 3500515050 # 开发者 class Help: """帮助文档""" ADMIN_HELP = """开发者命令提示： 1 添加群:\n/admin:add QQ群号 QQ群名 2. 删除功能:\n/admin:delete QQ群号 3 查找功能:\n/admin:get QQ群号 4. 获取所有群:\n/admin:show 5. 修改权限命令:\n/admin:changeAuth QQ群号 |聊天功能|入群欢迎|管理群|戳一戳|拓展功能|定时功能|(比如110011)\n 如果还是有问题，请与开发人员联系哦！""" GROUP_ADMIN = """[CQ:at,qq=%d]命令提示： 1. 查看群权限:\n/admin:get 2. 修改群权限：\n/admin:change |聊天功能|入群欢迎|管理群|戳一戳|拓展功能|定时功能|(比如#admin:change# 110011)\n 如果还是有问题，请与开发人员联系哦！""" class ProductConfig(Config, Mes, Url): """生产环境配置""" pass class DevelopConfig(Config, Mes, Url): """开发环境配置""" 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 # 后面主要是通过这个接口来对消息的分发 

#!/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() # 首先查看一下数据库中是否收录了这个群 if data is None: # 如果没有则添加，同时，也检查其是否有逻辑删除 g = Group(nick, gid) # 创建一个群 g.group2auth = [GroupAuthority()] # 给群赋予默认的权限 session.add(g) # 添加到数据库中 session.commit() # 提交事务 return { "status": 200 } # 返回状态码 # 如果群存在，直接修改群名，以及使其可以检测到 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)