python 获取mysql数据库列表以及用户权限

系统管理员 2022-01-29 12:36 237阅读 0赞

# [python 获取mysql数据库列表以及用户权限][python _mysql] #

# 一、需求分析 #

需要统计出当前数据库的所有数据库名，以及每个用户的授权信息。

## 获取所有数据库 ##

在mysql里面，使用命令：

show databases

就可以获取所有数据库了

## 获取所有用户 ##

执行命令：

select User from mysql.user

注意：需要排除到默认的用户，比如：

"root", "mysql.sys", "mysql.session"

## 获取用户权限 ##

语法：

show grants for 用户名;

比如：

show grants for test;

执行输出：

GRANT USAGE ON *.* TO 'test'@'%'
    GRANT SELECT ON `DB01`.* TO 'test'@'%'
    GRANT ALL PRIVILEGES ON `DB02`.* TO 'test'@'%' WITH GRANT OPTION

注意：这段信息表示，test用户，对DB01数据库下的所有表，是只读权限。

对对DB02数据库下的所有表，拥有读写权限。

# 二、python实现 #

由于时间关系，这里不一一解释了，代码里面都有注释。

完整代码如下：

![ContractedBlock.gif][] ![ExpandedBlockStart.gif][]

#!/usr/bin/env python
    # coding: utf-8
    
    import json
    import pymysql
    
    
    class Mysql(object):
        # mysql 端口号,注意：必须是int类型
        def __init__(self, host, user, passwd, port, db_name):
            self.host = host
            self.user = user
            self.passwd = passwd
            self.port = port
            self.db_name = db_name
    
        def select(self, sql):
            """
            执行sql命令
            :param sql: sql语句
            :return: 元祖
            """
            try:
                conn = pymysql.connect(
                    host=self.host,
                    user=self.user,
                    passwd=self.passwd,
                    port=self.port,
                    database=self.db_name,
                    charset='utf8',
                    cursorclass=pymysql.cursors.DictCursor
                )
                cur = conn.cursor()  # 创建游标
                # conn.cursor()
                cur.execute(sql)  # 执行sql命令
                res = cur.fetchall()  # 获取执行的返回结果
                cur.close()
                conn.close()
                return res
            except Exception as e:
                print(e)
                return False
    
        def get_all_db(self):
            """
            获取所有数据库名
            :return: list
            """
            # 排除自带的数据库
            exclude_list = ["sys", "information_schema", "mysql", "performance_schema"]
            sql = "show databases"  # 显示所有数据库
            res = self.select(sql)
            # print(res)
            if not res:  # 判断结果非空
                return False
    
            db_list = []  # 数据库列表
            for i in res:
                db_name = i['Database']
                # 判断不在排除列表时
                if db_name not in exclude_list:
                    db_list.append(db_name)
                    # print(db_name)
    
            if not db_list:
                return False
    
            return db_list
    
        def get_user_list(self):
            """
            获取用户列表
            :return: list
            """
            # 排除自带的用户
            exclude_list = ["root", "mysql.sys", "mysql.session"]
            sql = "select User from mysql.user"
            res = self.select(sql)
            # print(res)
            if not res:  # 判断结果非空
                return False
    
            user_list = []
            for i in res:
                db_name = i['User']
                # 判断不在排除列表时
                if db_name not in exclude_list:
                    user_list.append(db_name)
    
            if not user_list:
                return False
    
            return user_list
    
        def get_user_power(self):
            """
            获取用户权限
            :return: {}
    
            {
                "test":{  # 用户名
                    "read":["db1","db2"],  # 只拥有读取权限的数据库
                    "all":["db1","db2"],  # 拥有读写权限的数据库
                },
                ...
            }
            """
            info_dict = {}  # 最终结果字典
            # 获取用户列表
            user_list = self.get_user_list()
            if not user_list:
                return False
    
            # 查询每一个用户的权限
            for user in user_list:
                # print("user",user)
                sql = "show grants for {}".format(user)
                res = self.select(sql)
                if not res:
                    return False
    
                for i in res:
                    key = 'Grants for {}@%'.format(user)
                    # print("key",key)
                    # 判断key值存在时
                    if i.get(key):
                        # print(i[key])
                        # 包含ALL或者SELECT时
                        if "ALL" in i[key] or "SELECT" in i[key]:
                            # print(i[key])
                            if not info_dict.get(user):
                                info_dict[user] = {
           "read": [], "all": []}
    
                            cut_str = i[key].split()  # 空格切割
                            # print(cut_str,len(cut_str))
                            power = cut_str[1]  # 权限，比如ALL，SELECT
    
                            if len(cut_str) == 6:  # 判断切割长度
                                # 去除左边的`
                                tmp_str = cut_str[3].lstrip("`")
                            else:
                                tmp_str = cut_str[4].lstrip("`")
    
                            # 替换字符串
                            tmp_str = tmp_str.replace('`.*', '')
                            value = tmp_str.replace('\_', '-')
    
                            # 判断权限为select 时
                            if power.lower() == "select":
                                if value not in info_dict[user].get("read"):
                                    # 只读列表
                                    info_dict[user]["read"].append(value)
                            else:
                                if value not in info_dict[user].get("all"):
                                    # 所有权限列表
                                    info_dict[user]["all"].append(value)
    
            # print(info_dict)
            return info_dict
    
    
    if __name__ == '__main__':
        host = "192.168.10.10"
        user = "root"
        passwd = "123456"
        port = 3306
        db_name = "mysql"
    
        obj = Mysql(host, user, passwd, port, db_name)
        all_db_list = obj.get_all_db()
        user_power = obj.get_user_power()
    
        print("all_db_list",all_db_list)
        print("user_power",user_power)

posted @ 2019-05-21 16:38 [肖祥][Link 1] 阅读( ...) 评论( ...) [编辑][Link 2] 收藏

[python _mysql]: https://www.cnblogs.com/xiao987334176/p/10900673.html
[ContractedBlock.gif]: https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif
[ExpandedBlockStart.gif]: /images/20220129/5ac79244d42442a7ae907a8d4fbff4bc.png
[Link 1]: https://www.cnblogs.com/xiao987334176/
[Link 2]: https://i.cnblogs.com/EditPosts.aspx?postid=10900673