MySQL权限管理小程序

作业：

     参考表结构：

         用户类型

         用户信息

         权限

         用户类型&权限

    功能：          

1. 登陆、注册、找回密码
2. 权限管理
3. 角色管理
4. 角色分配权限
5. 动态显示当前登陆用户权限菜单

注：可基于反射实现将权限和代码结合"

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,Integer,ForeignKey,CHAR,VARCHAR,UniqueConstraint
from sqlalchemy.orm import sessionmaker
#建立链接
engine = create_engine("mysql+pymysql://root:@127.0.0.1:3306/s作业db?charset=utf8", max_overflow=5)
Base = declarative_base()
Session = sessionmaker(bind=engine)
session = Session()
#创建用户信息表
class User(Base):
    __tablename__ = 'user'
    uid = Column(Integer,primary_key=True,autoincrement=True)
    name = Column(CHAR(20),nullable=False)
    pwd = Column(CHAR(20),nullable=False)
    roll_id = Column(Integer,ForeignKey('roll.rid'))
#__table_args__的参数必须是元组的形式
    __table_args__ = (
        UniqueConstraint(name,pwd,name='uni_np'),  #用户名和密码不能重复！
    )
#创建权限表
class Authority(Base):
    __tablename__ ='authority'
    aid = Column(Integer,primary_key=True,autoincrement=True)
    auth = Column(VARCHAR(255),nullable=True)

#创建角色表
class Roll(Base):
    __tablename__ ='roll'
    rid = Column(Integer,primary_key=True,autoincrement=True)
    dep = Column(VARCHAR(255),nullable=False)

#角色权限关系表
class Ro_au(Base):
    __tablename__ ='ro_au'
    id = Column(Integer,primary_key=True,autoincrement=True)
    rid = Column(Integer,ForeignKey('roll.rid'))
    aid = Column(Integer,ForeignKey('authority.aid'))
#创建原始关系表
def original_table():
    session.add_all([
        Roll(dep='项目经理'),
        Roll(dep='实验室'),
        Roll(dep='办公室'),
        Roll(dep='人事部'),
        Roll(dep='工程部'),
        Roll(dep='普通员工'),
        Authority(auth='查看个人信息'),
        Authority(auth='更改用户信息'),
        Authority(auth='删除用户'),
        Authority(auth='设定项目整体方案'),
        Authority(auth='查看项目进度'),
        Authority(auth='查看项目经费'),
        Authority(auth='查看项目计划'),
        Authority(auth='项目技术方案'),
        Authority(auth='项目支出管理'),
        Authority(auth='项目人事安排'),
        Authority(auth='项目后勤保障'),
    ])
    session.commit()
    session.add_all([
        User(name='李一',pwd='123',roll_id=1),
        User(name='李二',pwd='456',roll_id=2),
        User(name='李三',pwd='789',roll_id=2),
        User(name='李四',pwd='1234',roll_id=3),
        User(name='李五',pwd='2345',roll_id=4),
        User(name='李六',pwd='3456',roll_id=5),
        User(name='李七',pwd='4567',roll_id=6),
    ])
    session.add_all([
        Ro_au(rid=1,aid=1),Ro_au(rid=1,aid=5),Ro_au(rid=1,aid=6),
        Ro_au(rid=2, aid=1),Ro_au(rid=2, aid=4),Ro_au(rid=2, aid=8),
        Ro_au(rid=3, aid=1),Ro_au(rid=3, aid=5),Ro_au(rid=3, aid=6),Ro_au(rid=3, aid=7),
        Ro_au(rid=4, aid=1),Ro_au(rid=4, aid=2),Ro_au(rid=4, aid=3),
        Ro_au(rid=5, aid=1),Ro_au(rid=5, aid=9),Ro_au(rid=5, aid=11),
        Ro_au(rid=6, aid=1),
    ])
    session.commit()

#创建数据库所有表
def creat_db():
    Base.metadata.create_all(engine)

#删除数据库所用表
def drop_db():
    Base.metadata.drop_all(engine)
#显示个人权限
def show_auth(user_name):
    user_roll_id = session.query(User.roll_id).filter(User.name == user_name).subquery()
    user_dep_list = session.query(Roll.dep).filter(Roll.rid == user_roll_id).all()
    for row in user_dep_list:
        user_dep = row[0]
        print('您当前所属部门名称：', user_dep)
    print('33[33m您具有的权限如下:33[37m')
    auth_list = session.query(Authority.auth).join(Ro_au).filter(Ro_au.rid == user_roll_id).all()
    count = 1
    for auth in auth_list:
        print(count, auth[0])
        count += 1

#显示所有部门信息
def show_department():
    print('公司现有部门如下：')
    dep_list = session.query(Roll).all()
    for dep in dep_list:
        print(dep[0],dep[1])


def login():
    while True:
        user_name = input('请输入用户名【返回3】：
>>>').strip()
        if user_name.lower() == '3':
            break
        user_pwd = input('请输入用户密码【返回3】：
>>>').strip()
        if user_pwd.lower() == '3':
            break
        user_list = session.query(User.name,User.pwd).all()

        if (user_name,user_pwd) in user_list:
            print('登陆成功！')
            choice = input('》》》1、进入个人权限界面'
                           '
》》》2、查看公司具有的所有部门'
                           '
》》》3、返回主界面'
                           '
>>>').strip()
            if choice == '1':
                show_auth(user_name)
                continue
            if choice == '2':
                show_department()
                continue
            if choice.lower() == '3':
                break
        else:
            print('用户名或者密码错误！')
            continue

#用户注册
def register():
    while True:
        user_name = input('请输入用户名【返回B】：
>>>').strip()
        if not user_name:continue
        if user_name.lower() == '3':
            break
        name_list = session.query(User.name).all()
        if (user_name,) in name_list:
            print('该用户名已经存在！')
            break
        else:
            user_pwd = input('请输入用户密码：
>>>').strip()
            if not user_pwd:continue
            dep_list = session.query(Roll.rid,Roll.dep).all()
            for row in dep_list:
                print(row.rid,row.dep)
            user_roll = input('请输入所属部门序号：
>>>').strip()
            if not user_roll:
                print('部门序号不能为空，请重新开始！')
                continue
            if user_roll.isdigit():
                session.add(User(name=user_name,pwd=user_pwd,roll_id=int(user_roll))) #写入数据库
                session.commit()
                print('新的员工信息已经注册成功！')
                break
            else:
                print('您选择的部门有误，请重新选择！')

#找回密码
def find_pwd():
    while True:
        user_name =input('请输入用户名【返回B】：').strip()
        if not user_name:continue
        if user_name.lower() == '3':
            break
        name_list =session.query(User.name).all()
        if (user_name,) in name_list:
            pwd_list =session.query(User.pwd).filter(User.name==user_name).all()
            print('您的密码为：')
            for pwd in pwd_list:
                print(pwd[0])
        else:
            print('您输入的用户名不存在！')

if __name__ == '__main__':
    while True:
        creat_tab_choice = input('是否要在数据库中创建原始关系表？第一次请选择是！'
                                 '**此操作会删除已添加的数据！**【是Y否N】：
>>>').strip()
        if creat_tab_choice.lower() == 'y':
            drop_db()
            creat_db()
            original_table()
            print('基本权限管理原始表已创建！')
        elif creat_tab_choice.lower() == 'n':
            pass
        else:
            print('您的操作有误，请重新选择！')
            continue
        while True:
            print('欢迎进入权限管理系统！'.center(21, '*'))
            choice = input('登录D	注册Z	找回密码M	退出Q:
>>>').strip()
            if choice.lower() == 'q':
                print('成功退出权限管理系统，欢迎您再次使用！')
                break
            elif choice.lower() == 'd':
                login()
            elif choice.lower() == 'z':
                register()
            elif choice.lower() == 'm':
                find_pwd()
            else:
                print('您输入的指令有误，请重新选择！')
                continue
        break