ORM框架SQLAlchemy与权限管理系统的数据库设计

SQLAlchemy是Python编程语言下的一款ORM框架，该框架建立在数据库API之上，使用对象关系映射进行数据库操作，即：将对象转换成SQL，然后使用数据API执行SQL并获取执行结果。

执行流程：

1、使用者通过ORM对象提交命令
2、将命令交给SQLAlchemy Core（Schema/Types  SQL Expression Language）转换成SQL
3、使用 Engine/ConnectionPooling/Dialect 进行数据库操作
4、匹配使用者事先配置好的egine
5、egine从连接池中取出一个链接
6、基于该链接通过Dialect调用DB API，将SQL转交给它去执行

1、关于DB API

SQLAlchemy本身无法操作数据库，其必须依赖pymsql等第三方插件，Dialect用于和数据API进行交流，根据配置文件的不同调用不同的数据库API，从而实现对数据库的操作。

 

1、MySQL-Python
mysql+mysqldb://<user>:<password>@<host>[:<port>]/<dbname>
2、pymysql
mysql+pymysql://<username>:<password>@<host>/<dbname>[?<options>]
3、MySQL-Connector
mysql+mysqlconnector://<user>:<password>@<host>[:<port>]/<dbname>
4、cx_Oracle
oracle+cx_oracle://user:pass@host:port/dbname[?key=value&key=value...]

更多详见：http://docs.sqlalchemy.org/en/latest/dialects/index.html 

2、创建表操作

在ORM框架下的Alchemy中，类==>表table　　对象==>数据行column ；

四张表:业务线,服务,用户,角色，利用ORM创建出它们，并建立好它们直接的关系

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,Integer,String,DateTime,Enum,ForeignKey,UniqueConstraint,ForeignKeyConstraint,Index
from sqlalchemy.orm import sessionmaker

egine=create_engine('mysql+pymysql://root@127.0.0.1:3306/db1?charset=utf8',max_overflow=5)

Base=declarative_base()

#创建单表:业务线
class Business(Base):
    __tablename__='business'
    id=Column(Integer,primary_key=True,autoincrement=True)
    bname=Column(String(32),nullable=False,index=True)

#多对一:多个服务可以属于一个业务线,多个业务线不能包含同一个服务
class Service(Base):
    __tablename__='service'
    id=Column(Integer,primary_key=True,autoincrement=True)
    sname=Column(String(32),nullable=False,index=True)
    ip=Column(String(15),nullable=False)
    port=Column(Integer,nullable=False)

    business_id=Column(Integer,ForeignKey('business.id'))

    __table_args__=(
        UniqueConstraint(ip,port,name='uix_ip_port'),
        Index('ix_id_sname',id,sname)
    )

#一对一:一种角色只能管理一条业务线,一条业务线只能被一种角色管理
class Role(Base):
    __tablename__='role'
    id=Column(Integer,primary_key=True,autoincrement=True)
    rname=Column(String(32),nullable=False,index=True)
    priv=Column(String(64),nullable=False)

    business_id=Column(Integer,ForeignKey('business.id'),unique=True)


#多对多:多个用户可以是同一个role,多个role可以包含同一个用户
class Users(Base):
    __tablename__='users'
    id=Column(Integer,primary_key=True,autoincrement=True)
    uname=Column(String(32),nullable=False,index=True)


class Users2Role(Base):
    __tablename__='users2role'
    id=Column(Integer,primary_key=True,autoincrement=True)
    uid=Column(Integer,ForeignKey('users.id'))
    rid=Column(Integer,ForeignKey('role.id'))

    __table_args__=(
        UniqueConstraint(uid,rid,name='uix_uid_rid'),
    )


def init_db():
    Base.metadata.create_all(egine)

def drop_db():
    Base.metadata.drop_all(egine)

if __name__ == '__main__':
    init_db()

3、对于表的增删改查

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,Integer,String,ForeignKey
from sqlalchemy.orm import sessionmaker

egine=create_engine('mysql+pymysql://root@127.0.0.1:3306/db1?charset=utf8',max_overflow=5)

Base=declarative_base()

class Dep(Base):
    __tablename__='dep'
    id=Column(Integer,primary_key=True,autoincrement=True)
    dname=Column(String(64),nullable=False,index=True)

class Emp(Base):
    __tablename__='emp'
    id=Column(Integer,primary_key=True,autoincrement=True)
    ename=Column(String(32),nullable=False,index=True)
    dep_id=Column(Integer,ForeignKey('dep.id'))

def init_db():
    Base.metadata.create_all(egine)

def drop_db():
    Base.metadata.drop_all(egine)

drop_db()
init_db()
Session=sessionmaker(bind=egine)
session=Session()

【增】

　　有两种row增加的方式：单行增加session.add( )；多行增加session.addall( )

#增
row_obj=Dep(dname='销售') #按关键字传参,无需指定id,因其是自增长的
session.add(row_obj)
session.add_all([
    Dep(dname='技术'),
    Dep(dname='运营'),
    Dep(dname='人事'),
])

session.commit()

【删】

session.query(Dep).filter(Dep.id > 3).delete()
session.commit()

【改】

#改
session.query(Dep).filter(Dep.id > 0).update({'dname':'哇哈哈'})
session.query(Dep).filter(Dep.id>0).update
({'dname':Dep.dname+'_SB'},synchronize_session=False)
session.query(Dep).filter(Dep.id>0).update
({'id':Dep.id*100},synchronize_session='evaluate')

session.commit()

　　使用Alchemy 进行增、删、改的操作时最后都需要进行向数据库提交数据：session.commit( )

【查】

#查所有,取所有字段
res=session.query(Dep).all() #for row in res:print(row.id,row.dname)

#查所有,取指定字段
res=session.query(Dep.dname).order_by(Dep.id).all() #for row in res:print(row.dname)

res=session.query(Dep.dname).first()
print(res) # ('哇哈哈_SB',)

#过滤查
res=session.query(Dep).filter(Dep.id > 1,Dep.id <1000) #逗号分隔,默认为and
print([(row.id,row.dname) for row in res])

4、子查询

　个人感觉使用连表操作能很大程度代替子查询的工作量，而且Alchemy实现起来也很方便！

　详细内容参考海峰老师博客，链接：http://www.cnblogs.com/linhaifeng/articles/7560153.html

5、权限管理系统的设计

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:761379545@127.0.0.1:3306/test_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='alex',pwd='3714',roll_id=1),
        User(name='egon',pwd='123456',roll_id=2),
        User(name='sl',pwd='761379545',roll_id=2),
        User(name='eric',pwd='123456',roll_id=3),
        User(name='mick',pwd='123456',roll_id=4),
        User(name='john',pwd='123456',roll_id=5),
        User(name='kaka',pwd='123456',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('请输入用户名【返回B】：
>>>').strip()
        if user_name.lower() == 'b':
            break
        user_pwd = input('请输入用户密码【返回B】：
>>>').strip()
        if user_pwd.lower() == 'b':
            break
        user_list = session.query(User.name,User.pwd).all()

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

#用户注册
def register():
    while True:
        user_name = input('请输入用户名【返回B】：
>>>').strip()
        if not user_name:continue
        if user_name.lower() == 'b':
            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() == 'b':
            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('登录L	注册R	找回密码F	退出Q:
>>>').strip()
            if choice.lower() == 'q':
                print('成功退出权限管理系统，欢迎您再次使用！')
                break
            elif choice.lower() == 'l':
                login()
            elif choice.lower() == 'r':
                register()
            elif choice.lower() == 'f':
                find_pwd()
            else:
                print('您输入的指令有误，请重新选择！')
                continue
        break

　　数据库中的逻辑结构图如下：

       

　　通过该模块内容的学习，对ORM有了深刻的认识，通过Alchemy语句映射SQL语句来达到我们操作数据库的目的！