MySQL权限管理小程序
作业：
参考表结构：
用户类型
用户信息
权限
用户类型&权限
功能：
1. 登陆、注册、找回密码
2. 权限管理
3. 角色管理
4. 角色分配权限
5. 动态显示当前登陆用户权限菜单
注：可基于反射实现将权限和代码结合"
1 from sqlalchemy import create_engine 2 from sqlalchemy.ext.declarative import declarative_base 3 from sqlalchemy import Column,Integer,ForeignKey,CHAR,VARCHAR,UniqueConstraint 4 from sqlalchemy.orm import sessionmaker 5 #建立链接 6 engine = create_engine("mysql+pymysql://root:@127.0.0.1:3306/s作业db?charset=utf8", max_overflow=5) 7 Base = declarative_base() 8 Session = sessionmaker(bind=engine) 9 session = Session() 10 #创建用户信息表 11 class User(Base): 12 __tablename__ = 'user' 13 uid = Column(Integer,primary_key=True,autoincrement=True) 14 name = Column(CHAR(20),nullable=False) 15 pwd = Column(CHAR(20),nullable=False) 16 roll_id = Column(Integer,ForeignKey('roll.rid')) 17 #__table_args__的参数必须是元组的形式 18 __table_args__ = ( 19 UniqueConstraint(name,pwd,name='uni_np'), #用户名和密码不能重复！ 20 ) 21 #创建权限表 22 class Authority(Base): 23 __tablename__ ='authority' 24 aid = Column(Integer,primary_key=True,autoincrement=True) 25 auth = Column(VARCHAR(255),nullable=True) 26 27 #创建角色表 28 class Roll(Base): 29 __tablename__ ='roll' 30 rid = Column(Integer,primary_key=True,autoincrement=True) 31 dep = Column(VARCHAR(255),nullable=False) 32 33 #角色权限关系表 34 class Ro_au(Base): 35 __tablename__ ='ro_au' 36 id = Column(Integer,primary_key=True,autoincrement=True) 37 rid = Column(Integer,ForeignKey('roll.rid')) 38 aid = Column(Integer,ForeignKey('authority.aid')) 39 #创建原始关系表 40 def original_table(): 41 session.add_all([ 42 Roll(dep='项目经理'), 43 Roll(dep='实验室'), 44 Roll(dep='办公室'), 45 Roll(dep='人事部'), 46 Roll(dep='工程部'), 47 Roll(dep='普通员工'), 48 Authority(auth='查看个人信息'), 49 Authority(auth='更改用户信息'), 50 Authority(auth='删除用户'), 51 Authority(auth='设定项目整体方案'), 52 Authority(auth='查看项目进度'), 53 Authority(auth='查看项目经费'), 54 Authority(auth='查看项目计划'), 55 Authority(auth='项目技术方案'), 56 Authority(auth='项目支出管理'), 57 Authority(auth='项目人事安排'), 58 Authority(auth='项目后勤保障'), 59 ]) 60 session.commit() 61 session.add_all([ 62 User(name='李一',pwd='123',roll_id=1), 63 User(name='李二',pwd='456',roll_id=2), 64 User(name='李三',pwd='789',roll_id=2), 65 User(name='李四',pwd='1234',roll_id=3), 66 User(name='李五',pwd='2345',roll_id=4), 67 User(name='李六',pwd='3456',roll_id=5), 68 User(name='李七',pwd='4567',roll_id=6), 69 ]) 70 session.add_all([ 71 Ro_au(rid=1,aid=1),Ro_au(rid=1,aid=5),Ro_au(rid=1,aid=6), 72 Ro_au(rid=2, aid=1),Ro_au(rid=2, aid=4),Ro_au(rid=2, aid=8), 73 Ro_au(rid=3, aid=1),Ro_au(rid=3, aid=5),Ro_au(rid=3, aid=6),Ro_au(rid=3, aid=7), 74 Ro_au(rid=4, aid=1),Ro_au(rid=4, aid=2),Ro_au(rid=4, aid=3), 75 Ro_au(rid=5, aid=1),Ro_au(rid=5, aid=9),Ro_au(rid=5, aid=11), 76 Ro_au(rid=6, aid=1), 77 ]) 78 session.commit() 79 80 #创建数据库所有表 81 def creat_db(): 82 Base.metadata.create_all(engine) 83 84 #删除数据库所用表 85 def drop_db(): 86 Base.metadata.drop_all(engine) 87 #显示个人权限 88 def show_auth(user_name): 89 user_roll_id = session.query(User.roll_id).filter(User.name == user_name).subquery() 90 user_dep_list = session.query(Roll.dep).filter(Roll.rid == user_roll_id).all() 91 for row in user_dep_list: 92 user_dep = row[0] 93 print('您当前所属部门名称：', user_dep) 94 print('\33[33m您具有的权限如下:\33[37m') 95 auth_list = session.query(Authority.auth).join(Ro_au).filter(Ro_au.rid == user_roll_id).all() 96 count = 1 97 for auth in auth_list: 98 print(count, auth[0]) 99 count += 1 100 101 #显示所有部门信息 102 def show_department(): 103 print('公司现有部门如下：') 104 dep_list = session.query(Roll).all() 105 for dep in dep_list: 106 print(dep[0],dep[1]) 107 108 109 def login(): 110 while True: 111 user_name = input('请输入用户名【返回3】：\n>>>').strip() 112 if user_name.lower() == '3': 113 break 114 user_pwd = input('请输入用户密码【返回3】：\n>>>').strip() 115 if user_pwd.lower() == '3': 116 break 117 user_list = session.query(User.name,User.pwd).all() 118 119 if (user_name,user_pwd) in user_list: 120 print('登陆成功！') 121 choice = input('》》》1、进入个人权限界面' 122 '\n》》》2、查看公司具有的所有部门' 123 '\n》》》3、返回主界面' 124 '\n>>>').strip() 125 if choice == '1': 126 show_auth(user_name) 127 continue 128 if choice == '2': 129 show_department() 130 continue 131 if choice.lower() == '3': 132 break 133 else: 134 print('用户名或者密码错误！') 135 continue 136 137 #用户注册 138 def register(): 139 while True: 140 user_name = input('请输入用户名【返回B】：\n>>>').strip() 141 if not user_name:continue 142 if user_name.lower() == '3': 143 break 144 name_list = session.query(User.name).all() 145 if (user_name,) in name_list: 146 print('该用户名已经存在！') 147 break 148 else: 149 user_pwd = input('请输入用户密码：\n>>>').strip() 150 if not user_pwd:continue 151 dep_list = session.query(Roll.rid,Roll.dep).all() 152 for row in dep_list: 153 print(row.rid,row.dep) 154 user_roll = input('请输入所属部门序号：\n>>>').strip() 155 if not user_roll: 156 print('部门序号不能为空，请重新开始！') 157 continue 158 if user_roll.isdigit(): 159 session.add(User(name=user_name,pwd=user_pwd,roll_id=int(user_roll))) #写入数据库 160 session.commit() 161 print('新的员工信息已经注册成功！') 162 break 163 else: 164 print('您选择的部门有误，请重新选择！') 165 166 #找回密码 167 def find_pwd(): 168 while True: 169 user_name =input('请输入用户名【返回B】：').strip() 170 if not user_name:continue 171 if user_name.lower() == '3': 172 break 173 name_list =session.query(User.name).all() 174 if (user_name,) in name_list: 175 pwd_list =session.query(User.pwd).filter(User.name==user_name).all() 176 print('您的密码为：') 177 for pwd in pwd_list: 178 print(pwd[0]) 179 else: 180 print('您输入的用户名不存在！') 181 182 if __name__ == '__main__': 183 while True: 184 creat_tab_choice = input('是否要在数据库中创建原始关系表？第一次请选择是！' 185 '**此操作会删除已添加的数据！**【是Y否N】：\n>>>').strip() 186 if creat_tab_choice.lower() == 'y': 187 drop_db() 188 creat_db() 189 original_table() 190 print('基本权限管理原始表已创建！') 191 elif creat_tab_choice.lower() == 'n': 192 pass 193 else: 194 print('您的操作有误，请重新选择！') 195 continue 196 while True: 197 print('欢迎进入权限管理系统！'.center(21, '*')) 198 choice = input('登录D\t注册Z\t找回密码M\t退出Q:\n>>>').strip() 199 if choice.lower() == 'q': 200 print('成功退出权限管理系统，欢迎您再次使用！') 201 break 202 elif choice.lower() == 'd': 203 login() 204 elif choice.lower() == 'z': 205 register() 206 elif choice.lower() == 'm': 207 find_pwd() 208 else: 209 print('您输入的指令有误，请重新选择！') 210 continue 211 break
