学习MYSQL有趣的python

SQLAchemy

2019-02-27  本文已影响33人  运维开发_西瓜甜

番外篇

SQLAlchemy

关于字符集

修改 /etc/my.cnf文件添加如下内容:

[mysqld]
collation-server = utf8mb4_unicode_ci
init-connect='SET NAMES utf8mb4'
character-set-server = utf8mb4

[client]
default-character-set=utf8mb4
[mysql]
default-character-set=utf8mb4

之后重启 mysql

查看字符

show variables like 'collation_%';

show variables like 'character_set_%';

查看数据库 db_1803 的字符集、字符序

MariaDB [(none)]> use db_1803
Database changed
MariaDB [db_1803]> SELECT @@character_set_database, @@collation_database;
+--------------------------+----------------------+
| @@character_set_database | @@collation_database |
+--------------------------+----------------------+
| utf8                     | utf8_general_ci      |
+--------------------------+----------------------+

其他的一些设置方法:

修改数据库的字符集

 mysql>use mydb     
 mysql>alter database mydb_name character set utf-8;

创建数据库指定数据库的字符集

mysql>create database mydb_name character set utf-8; 

SQLAlchemy

基本介绍

SQLAlchemy 是 Python 中一个通过 ORM 操作数据库的框架。

SQLAlchemy对象关系映射器提供了一种方法,用于将用户定义的Python类与数据库表相关联,并将这些类(对象)的实例与其对应表中的行相关联。它包括一个透明地同步对象及其相关行之间状态的所有变化的系统,称为工作单元,以及根据用户定义的类及其定义的彼此之间的关系表达数据库查询的系统。

官方解释:
它提供了一整套众所周知的企业级持久性模式,旨在实现高效,高性能的数据库访问,并采用简单的Pythonic域语言。

可以让我们使用类和对象的方式操作数据库,从而从繁琐的 sql 语句中解脱出来。

ORM 就是 Object Relational Mapper 的简写,就是关系对象映射器的意思。

架构图
架构图

Schema / Types 定义了类到表之间的映射框架(规则)

SQL Expression Language 封装好的 SQL 语句

Engine 操作者

Connection Pooling 连接池

Dialect 根据用户的配置,调用不同的数据库 API(Oracle, postgresql, Mysql) 并执行对应的 SQL语句

安装

shell> pip3 install sqlalchemy

DBAPI

SQLAlchemy本身无法操作数据库,其必须使用 pymsql 等第三方插件,从而实现对数据库的操作,如:

pymysql 使用如下方式连接:
    mysql+pymysql://<username>:<password>@<host>/<dbname>[?<options>]

更多参考:https://docs.sqlalchemy.org/en/latest/dialects/index.html

使用 EngineConnectionPoolingDialect 对数据库进行操作的流程:

Engine (SQLAlchemy 引擎) 使用 ConnectionPooling 连接数据库,之后再通过 Dialect 执行 SQL 语句。

连接数据库

下面一连接 Mysql 为例来演示连接语法的具体实现

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from sqlalchemy import create_engine
 
engine = create_engine(
  "mysql+pymysql://root:123@172.16.153.160:3306/dbname?charset=utf8mb4", 
  echo=True, 
  max_overflow=5)

echo 标志是设置SQLAlchemy日志记录的快捷方式。 启用它后,我们将看到所有生成的SQL。

Max_overflow 指定了连接池的最大连接数。

create_engine() 的返回值是一个实例引擎,它代表了一个数据库的核心接口。

此时的连接是惰性的,当create_engine()第一次返回的引擎,其实并没有试图连接到数据库之中; 只有在第一次要求它对数据库执行任务时才会发生这种情况。比如使用了以下的任一方法:

engine.execute()

执行原生 sql 语句

ret = engine.execute("select * from t1;")
# print(dir(engine))
# print(ret.fetchone())
print(ret.fetchall())

ORM 操作

ORM 流程

ORM 操作的流程是,一个 Engine 使用 Schema Type 创建一个特定的结构对象,之后通过 SQL Expression Language 将该对象转换成 SQL 语句, 接着使用 ConnectionPling 连接数据库,最后通过 Dialect 执行 SQL 语句,并获取结果。

声明映射

官方解释:

使用ORM时,配置过程首先描述我们将要处理的数据库表,然后定义我们自己的类,这些类将映射到这些表。在现代SQLAlchemy中,这两个任务通常使用称为Declarative的系统一起执行,这允许我们创建包含指令的类,以描述它们将映射到的实际数据库表。

大白话:

创建一个类,一个类对应了一个表,类的数据属性对应了表中的字段名,这个类称为映射类。

根据映射类创建出一个一个的对象,每个对象对应了表中的一条实际的数据。

image.png

使用Declarative系统映射的类是根据基类定义的,该基类维护相对于该基类的类和表的目录 - 这称为声明性基类。我们的应用程序通常在一个常用的模块中只有一个这个基础的实例。我们使用declarative_base() 函数创建基类,如下所示:

from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

现在我们有了一个Base,我们可以根据它定义任意数量的映射类。

我们将从一个名为 teacher 的表开始。

示例表结构

teacher

id name age city
1 yangge 18 北京
2 qiangge 19 北京
3 shark 19 北京

Teacher将是我们映射此表的类。

在类中,我们定义了有关我们将要映射的表的详细信息,主要是表名,以及列的名称和数据类型:

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String

Base = declarative_base()

class Teacher(Base):
    __tablename__ = 'teacher'
    id = Column(Integer, primary_key=True)
    name = Column(String(12))
    age = Column(String(2))
    city = Column(String(16))
    
    def __repr__(self):
        tpl = "Teacher(id={}, name={}, age={}, city={})"
        return tpl.format(self.id, self.name,
                          self.age, self.city)

__repr__ 方法定义了一个对象的比较易读的显式方式

使用 Declarative 的类至少需要一个__tablename__属性,并且至少有一个 Column属于主键。

此时,Declarative 为我们创建了一个对象,可以在终端中查看这个对象。

Teacher.__table__
Out[3]: Table('teacher', MetaData(bind=None), Column('id', Integer(), table=<teacher>, primary_key=True, nullable=False), Column('name', String(length=12), table=<teacher>), Column('age', String(length=2), table=<teacher>), Column('city', String(length=16), table=<teacher>), schema=None)

这个称为 Table 对象,该Table对象是一个更大的集合的成员,称为MetaData

MetaData 是一个注册表,包括了向数据库发出一组生成表命令的功能。

创建表到数据库中

我们可以使用MetaData 为所有尚不存在的表向数据库发出CREATE TABLE语句。

下面,我们调用该MetaData.create_all()方法,将我们Engine 作为数据库连接源传递。我们将看到首先发出特殊命令以检查teacher表的存在,然后是实际的语句:CREATE TABLE

Base.metadata.create_all(engine)

执行成功后可以检查是否创建成功

终端检查结果

MariaDB [(none)]> use db_1803
Reading table information for completion of table and column names
You can turn off this feature to get a quicker startup with -A

Database changed
MariaDB [db_1803]> show tables;
+-------------------+
| Tables_in_db_1803 |
+-------------------+
| teacher           |
+-------------------+
1 row in set (0.00 sec)

MariaDB [db_1803]>

Navicat 软件检查结果

image.png
创建映射类的实例
yg_teacher = Teacher(name='yangge', 
               age='18', 
               city='BeiJing')

此时,实例对象只是在此刻环境的内存中有效,并没有在表中真正生成数据。

要想生成数据到表中,需要创建一个和数据库沟通的会话对象,利用这个会话对象对数据库中的表进行操作(增加、更新,删除、查询)

创建会话

现在准备开始与数据库交谈,需要使用一个引擎的实例来创建一个 Session类的实例。

from sqlalchemy.orm import sessionmaker

# 把当前的引擎绑定给这个会话
Session = sessionmaker(bind=engine) 

# 实例化
session = Session()
通过 Session 实例的方法进行操作表中的数据

刚才我们已经创建了一条数据的实例了,要想把这条数据实例真正的增加到数据库中的teacher 表中,需要先把这条数据对象添加到会话实例中,再通过会话实例的 commit() 方法提交事务到数据库,此时,数据库的 teacher 表中才会有这条数据。

添加单条数据到 session 中

session.add(yg_teacher)

此时这个数据并没有被同步的数据库中,而是处于等待的状态。

如果我们查询数据库,则首先刷新所有待处理信息,然后立即发出查询。

our_teacher = session.query(User).filter_by(
    name='yangge').first() 
print(our_teacher)

此时得到的结果是并不是 数据库表中的最终数据,而是映射类的一个对象

提交

session.commit()

此时,数据真正的被写入到数据库中了

增加多条数据

session.add_all([
    Teacher(name='qiangge', age='19',city='北京'),
    Teacher(name='shark', age='19',city='北京'),
])

一起提交

session.commit()

回滚

在 commit() 之前,对实例对象的属性所做的更改,可以进行回滚。回到更改之前。

session.rollback()

示例:

假如目前向表中添加一条新的数据,同时查询出一条已有的数据,并且把已有的数据某个字段的值进行更改。

# 新增数据
session.add(Teacher(name='shark2', 
                    age='18',
                    city='ZhengZhou'))

# 修改已有数据的字段值
shark=session.query(Teacher).filter_by(name='shark').first()
print(shark.age)
shark.age = '28'
shark.name = 'shark3'

# 查询操作结果
two_user = session.query(Teacher).filter(Teacher.name.in_(['shark2','shark3'])).all()
print(two_user)

#输出的结果
[Teacher(id=4, name=shark, age=38, city='北京'),
 Teacher(id=23, name=shark2, age=18, city=ZhengZhou)]

此时数据的更改只是发生于会话对象的事务中,并没有发生在数据库中。

现在进行回滚操作

session.rollback()

之后再次进行查询操作会发现已恢复到修改之前

[Teacher(id=4, name=shark, age=23, city='北京')]

进阶操作

导入模块介绍
# 创建连接相关
from sqlalchemy import create_engine

# 和 sqlapi 交互,执行转换后的 sql 语句,用于创建基类
from sqlalchemy.ext.declarative import declarative_base

# 创建表中的字段(列)
from sqlalchemy import Column

# 表中字段的属性
from sqlalchemy import Integer, String, ForeignKey
from sqlalchemy import UniqueConstraint, Index
from sqlalchemy.orm import sessionmaker, relationship
连接数据库
# 创建连接对象,并使用 pymsql 引擎
conn_str = "mysql+pymysql://{user}:{pwd}@{}:3306/{db_name}?charset='utf8mb4"
connect_info = conn_str.format(user='root',
                              pwd='123456',
                              db_name='db_1803')

engine = create_engine(connect_info, max_overflow=5)
创建表
# 创建基类
Base = declarative_base()

# 创建单表
class Person(Base):
    __tablename__ = 'person'  # 表名
    id = Column(Integer, primary_key=True)
    # 必须指定长度,在PostgreSQL上不需要
    name = Column(String(32))
    age = Column(Integer)      # 整型

    __table_args__ = (
    # 设置联合唯一
    UniqueConstraint('id', 'name', name='uix_id_name'),
    
    # 建立索引   
    Index('uix_id_name', 'name'),
    )


def init_db():
    """创建所有定义的表到数据库中"""
    Base.metadata.create_all(engine)


def drop_db():
    """从数据库中删除所有定义的表"""
    Base.metadata.drop_all(engine)
    
# 执行创建表
#init_db()
    
# 创建会话实例对象
Session = sessionmaker(bind=engine)
session = Session()
查询数据
  1. 迭代查询结果集
for name, age, in session.query(Teacher.name, Teacher.age):
    print(name, age)
    
# 输出结果
yangge 18
qiangge 19
shark 23
  1. 给列起别名

可以使用 label() 给每个列名起别名

for row in session.query(Teacher.name.label('t_name')).all():
    print(row.t_name)
  1. 控制返回的查询结果集

    all() 返回的是所有的结果集,是列表

    first() 返回的是所有结果集中的第一个数据

ret = session.query(Teacher).all()
# 结果 [obj1,obj2,obj3]

ret = session.query(Teacher).first()
# 结果 obj1

​ 提取结果集中的所有数据,假如没有或者数据多于一条则会报错

​ 找到后返回的是一个元组

scalar() 调用 one() 方法,找不到,返回 None

​ 找到后返回的是赤裸裸的数据

session.query(Teacher).all()[0:2]

​ 排序

# 正序
session.query(Teacher).order_by(Teacher.name).all()

# 倒序
session.query(Teacher).order_by(Teacher.name.desc()).all()

# 先按名字排序,假如有相同的再安装 id 排序
session.query(Teacher).order_by(Teacher.name,       
                                Teacher.id.desc()).all()
session.query(Teacher).filter(Teacher.age=='18').count()
  1. 结果集内看到的是对象 、还是数据

query(类名) 返回的就是对象

query(类名.字段名) 返回的就是含有数据的元组对象

# 所有数据,且结果集中是一个一个的对象
ret = session.query(Teacher).all()
# 结果 [obj1, obj2, obj3]

#  指定字段查询,返回所有的数据,是一个列表,列表内是一个一个的元组
ret = session.query(Teachers.name, Teachers.arg).all()
# 结果 [('yangge', '18'), ('qiangge', '19'), ('shark', '23')]
  1. 条件查询

filter_by() 接收的是关键字参数

filter() 允许使用 python 的比较或关系运算符,实现更灵活的查询

# filter_by()
ret = session.query(Teacher).filter_by(name='yangge').first()
# 结果 Teacher(id=2, name=yangge, age=18, city=BeiJing)

# filter()
ret = session.query(Teacher).filter(Teacher.age>'20').first()
# 结果 Teacher(id=4, name=shark, age=23, city='北京')
  1. 关系运算符的查询

以下适用于 filter()

以下查询都是以这个查询对象为基础的过滤

query = session.query(Teacher)
query.filter(Teacher.name == 'shark').all()
query.filter(Teacher.name != 'shark').all()
query.filter(Teacher.name.like('%sha%')).all()

query.filter(Teacher.name.ilike('Sha___')).all()
query.filter(Teacher.id.in_([2,3])).all()
query.filter(~Teacher.id.in_([2,3])).all()
query.filter(Teacher.id.between(1, 3)).all()
query.filter(Teacher.name == None).all()
# 或者
query.filter(Teacher.name.is_(None)).all()
query.filter(Teacher.name != None).all()
# 或者
query.filter(Teacher.name.isnot(None)).all()
# 使用 and_()
from sqlalchemy import and_
query.filter(and_(Teacher.name == 'shark', 
                  Teacher.city == '北京')).all()

# 或者使用逗号
query.filter(Teacher.name == 'shark', 
             Teacher.city == '北京').all()
from sqlalchemy import or_
query.filter(or_(User.name == 'shark', 
                 User.name == 'xiguatian')).all()
query.filter(
    or_(
        Teacher.id <= 2,
        and_(Teacher.name == 'shark', Teacher.id > 3)
    )).all()
  1. 嵌套的查询
# 嵌套,从最内层的查询结果中再查询想要的数据
session.query(Teacher).filter(
    Teacher.id.in_(
        session.query(Teacher.id
                      ).filter_by(
            name='yangge'))).all()
  1. 分组统计查询
from sqlalchemy.sql import func
# 统计表中所有的数据
session.query(func.count('*')).select_from(Teacher).first()

# 以年龄分组,并统计每组的数据数量
session.query(func.count(Teacher.age),Teacher.age
                    ).group_by(Teacher.age).all()

# 以年龄为分组,并统计每组的最大/最小 id 号,年龄总和/平均值,
session.query(
    func.max(Teacher.id),
    func.min(Teacher.id),
    func.sum(Teacher.age),
    func.avg(Teacher.age),
    Teacher.id
                    ).group_by(Teacher.age).all()

# 从分组的数据中再查找需要的数据
session.query(
    func.max(Teacher.id),
    func.min(Teacher.age),
    func.sum(Teacher.age),
    func.avg(Teacher.age),

    Teacher.id
    ).group_by(Teacher.age
               ).having(func.min(Teacher.id) > 2).all()
  1. 组合
# 再创建一个表
class Student(Base):
    __tablename__ = 'student'
    id = Column(Integer,primary_key=True)
    name = Column(String(12))
    age = Column(String(2))
    city = Column(String(16))

# 组合  用一条数据将两个表中的要查询的数据组合在一张表里展示出来
q1 = session.query(Teacher.name).filter(Teacher.id > 2)
q2 = session.query(Student.name).filter(Student.id < 2)
## 去重
ret = q1.union(q2).all()
## 不去重 
q1 = session.query(Teacher.name).filter(Teacher.id > 2)
q2 = session.query(Student.name).filter(Student.id < 2)
ret = q1.union_all(q2).all()
更新数据
session.query(Teacher).filter(Teacher.id == 3).update(
    {"name" : "xiguatian"})

session.query(Teacher).filter(Teacher.id > 3).update(
    {Teacher.name: Teacher.name + "_云计算讲师"},
    synchronize_session=False) 
# 不同步,数据的更新在 commit 之后

session.query(Teacher).filter(Teacher.id > 2).update(
    {"age": int(Teacher.age + 1)}, 
    synchronize_session="evaluate") 

session.commit()
删除数据
session.query(Teacher).filter(Teacher.id > 4).delete()
session.commit()

连表操作

INNER JOIN 内连接
LEFT JOIN 左连接
RIGHT JOIN 右连接
FULL JOIN 完全连接

一对多

示例表:

groups

id name full_name cn_name
1 Other 默认组
2 PM Product Manager 产品经理
3 RD Research and Development engineer 开发
4 QA Qualtiy Assurance 测试
5 OP Operator 运维
6 DBA Database Administrator 数据库管理员

users

id name goup_id
1 Yangge 2
2 Tom 2
3 Rose 3
4 Shark 3
5 Xiguatian 5
6 Jack 6
7 new_user 1

创建表的原生语句

CREATE TABLE groups (
    id INTEGER NOT NULL AUTO_INCREMENT, 
    name VARCHAR(5) NOT NULL, 
    full_name VARCHAR(64), 
    cn_name VARCHAR(64), 
    PRIMARY KEY (id), 
    UNIQUE (name)
)

CREATE TABLE users (
    id INTEGER NOT NULL AUTO_INCREMENT, 
    name VARCHAR(32) NOT NULL, 
    group_id INTEGER, 
    PRIMARY KEY (id), 
    FOREIGN KEY(group_id) REFERENCES groups(id)
)
  1. 创建表
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship

engine = create_engine("mysql+pymysql://root:123456@172.16.153.160:3306/db_1802", echo=True, max_overflow=5)
Base = declarative_base()
class Groups(Base):
    __tablename__= 'groups'
    id = Column(Integer, primary_key=True)
    name = Column(String(12), 
                  unique=True,     # 值必须唯一
                  nullable=False)  # 不允许为空
    full_name = Column(String(64), nullable=True)
    cn_name = Column(String(64))

class Users(Base):
    __tablename__= 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String(32), 
                  unique=True,
                  index=True,  # 此列建立索引
                  nullable=False)
    group_id = Column(Integer, 
                      ForeignKey('groups.id'),  # 定义外键
                      default=1)  # 默认值
    
    # 下面此列与创建表无关,仅用于查询使用,group 正向查询,user 反向查询
    group = relationship('Groups',  # 字符串类型的映射类名称。 
                         backref='user')
    
# 创建所有的表结构
Base.metadata.create_all(engine)
  1. 添加数据
from sqlalchemy.orm import sessionmaker

# 把当前的引擎绑定给这个会话
Session = sessionmaker(bind=engine) 

# 实例化
session = Session()

session.add_all([
    Groups(name='Other',),
    Groups(name='PM',full_name='Product Manager', cn_name='产品经理'),
    Groups(name='RD',full_name='Research and Development engineer', cn_name='开发'),
    Groups(name='QA',full_name='Product Manager', cn_name='测试'),
    Groups(name='OP',full_name='Product Manager', cn_name='运维'),
    Groups(name='DBA',full_name='Product Manager', cn_name='数据库管理员'),
])

session.commit()

session.add_all([
    Users(name='Yangge', group_id=2),
    Users(name='Tom', group_id=2),
    Users(name='Rose', group_id=3),
    Users(name='shark', group_id=3),
    Users(name='xiguatian', group_id=5),
    Users(name='Jack', group_id=6),
    Users(name='new_user'),
])

session.commit()

补充

print(session.query(Groups).join(Users).all())
  1. 查询数据

问题一: 查询用户 Shark 是属于哪个组 ,打印出组名?

user_obj = session.query(Users).filter_by(name='Shark').scalar()
print(user_obj.group.name)

scalar() 假如存在,必须得到一个对象,不存在则返回 None

问题二: 查询组 RD 组中的成员都有谁 ?

group_obj = session.query(Groups).filter_by(name='RD').one()
for u_obj in group_obj.user:
    print(u_obj.name)
  1. 更新数据

现在把用户 new_user 加入到 测试组 QA

g_id = session.query(Groups.id).filter(Groups.name=='QA').scalar()
user_obj = session.query(Users).filter(Users.group_id == 1).update(
    {Users.group_id: g_id}
)
session.commit()
  1. 删除数据

删除用户 jack , 之后在删除组 DBA

# 删用户
session.query(Users).filter_by(name='jack').delete()
session.commit()
# 删组
query_obj = session.query(Groups).filter_by(name='DBA')
g_obj = query_obj.scalar()
u_obj = g_obj.user
if not u_obj:
    query_obj.delete()
    session.commit()
多对多 (扩展)

表示例

server

id hostname ip
1 web 172.16.153.10
2 elk 172.16.153.10

sys_user

id name password key
1 root
2 shark

host_to_sys_user

id host_id sys_user_id
1 1 1
2 1 2
3 2 1
  1. 创建表
# 多对多
# 一个服务器中可以有个用户
# 同名的用户可以出现在多台服务器中,也就是多台服务器可能存在相同的用户名
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy import Column, Integer, String, ForeignKey

# 需要安装 sqlalchemy-utils 和 passlib
# pip3 install sqlalchemy-utils passlib
from sqlalchemy_utils import IPAddressType, PasswordType

from sqlalchemy.orm import relationship

engine = create_engine("mysql+pymysql://root:123456@172.16.153.160:3306/db_1803",
                       echo=True,
                       max_overflow=5)
Base = declarative_base()

class ServersToSysUsers(Base):
    __tablename__ = 'servers_to_sys_users'
    nid = Column(Integer, primary_key=True, autoincrement=True)
    server_id = Column(Integer, ForeignKey('servers.id'))
    group_id = Column(Integer, ForeignKey('sys_users.id'))
    
    # 方式一
    #servers = relationship("Servers", backref='ser_to_user')
    #sys_users = relationship("SysUsers", backref='ser_to_user')

class Servers(Base):
    __tablename__ = 'servers'

    id = Column(Integer, primary_key=True, autoincrement=True)
    hostname = Column(String(64), unique=True, nullable=False)
    ip = Column(IPAddressType)  # 输入字符串,输出时是对象
    
    # 方式二
    sys_users = relationship('SysUsers',
                           secondary=ServersToSysUsers.__table__,
                           backref='servers')


class SysUsers(Base):
    __tablename__ = 'sys_users'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), unique=True, nullable=False)
    password = Column(PasswordType(
        schemes=[
            'pbkdf2_sha512',
            'md5_crypt'
        ],
        deprecated=['md5_crypt']))
    key = Column(String(128),nullable=True)


Base.metadata.create_all(engine)
  1. 添加数据
session.add_all([
    Servers(hostname='web', ip='1.1.1.1'),
    Servers(hostname='elk', ip='1.1.1.2'),
    SysUsers(name='root',password='upsa'),
    SysUsers(name='shark',password='upsa'),
])
session.commit()

session.add_all([
    ServersToSysUsers(server_id=1,sys_users_id=1),
    ServersToSysUsers(server_id=1,sys_users_id=2),
    ServersToSysUsers(server_id=2,sys_users_id=1),
])
session.commit()
  1. 查询数据

问题一: 查询服务器 web 下的所有用户名和密码

# 方式一的查询
query = session.query(Servers).filter_by(hostname='web')
if query:
    for obj in query.scalar().ser_to_user:
        print(obj.sys_users.name)

# 方式二的查询
query = session.query(Servers).filter_by(hostname='web')
if query:
    for u_obj in query.scalar().sys_users:
        print(u_obj.name, u_obj.password)

问题二: 查询用户 root 存在于哪些服务器上

# 方式一的查询 (作业练习)

# 方式二的查询:
query = session.query(SysUsers).filter_by(name='root')
if query:
    for server_obj in query.scalar().servers:
        print(server_obj.hostname,server_obj.ip)
  1. 更新数据(作业练习)

增加一条服务器信息

主机名: mysql

IP 地址: 1.1.1.100

可以登录的账户: db_admin, select, db_op

同时完善其他表

  1. 删除数据(作业练习)

删除服务器主机名为: mysql 的相关信息

扩展功能

使用 sqlalchemy-utils 会得到更多的功能

pip3 install sqlalchemy-utils

更多字段的数据类型

http://sqlalchemy-utils.readthedocs.io/en/latest/data_types.html

表关系实战

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from sqlalchemy import create_engine,and_,or_,func,Table
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String,ForeignKey,UniqueConstraint,DateTime
from  sqlalchemy.orm import sessionmaker,relationship

Base = declarative_base() #生成一个SqlORM 基类

# 服务器账号和组
# HostUser2Group = Table('hostuser_2_group',Base.metadata,
#     Column('hostuser_id',ForeignKey('host_user.id'),primary_key=True),
#     Column('group_id',ForeignKey('group.id'),primary_key=True),
# )

# 用户和组关系表,用户可以属于多个组,一个组可以有多个人
UserProfile2Group = Table('userprofile_2_group',Base.metadata,
    Column('userprofile_id',ForeignKey('user_profile.id'),primary_key=True),
    Column('group_id',ForeignKey('group.id'),primary_key=True),
)

# 程序登陆用户和服务器账户,一个人可以有多个服务器账号,一个服务器账号可以给多个人用
UserProfile2HostUser= Table('userprofile_2_hostuser',Base.metadata,
    Column('userprofile_id',ForeignKey('user_profile.id'),primary_key=True),
    Column('hostuser_id',ForeignKey('host_user.id'),primary_key=True),
)


class Host(Base):
    __tablename__='host'
    id = Column(Integer,primary_key=True,autoincrement=True)
    hostname = Column(String(64),unique=True,nullable=False)
    ip_addr = Column(String(128),unique=True,nullable=False)
    port = Column(Integer,default=22)
    def __repr__(self):
        return  "<id=%s,hostname=%s, ip_addr=%s>" %(self.id,
                                                    self.hostname,
                                                    self.ip_addr)


class HostUser(Base):
    __tablename__ = 'host_user'
    id = Column(Integer,primary_key=True)
    AuthTypes = [
        (u'ssh-passwd',u'SSH/Password'),
        (u'ssh-key',u'SSH/KEY'),
    ]
    # auth_type = Column(ChoiceType(AuthTypes))
    auth_type = Column(String(64))
    username = Column(String(64),unique=True,nullable=False)
    password = Column(String(255))

    host_id = Column(Integer,ForeignKey('host.id'))
    
    # groups = relationship('Group',
    #                       secondary=HostUser2Group,
    #                       backref='host_list')

    __table_args__ = (UniqueConstraint('host_id','username', name='_host_username_uc'),)

    def __repr__(self):
        return  "<id=%s,name=%s>" %(self.id,self.username)


class Group(Base):
    __tablename__ = 'group'
    id = Column(Integer,primary_key=True)
    name = Column(String(64),unique=True,nullable=False)
    def __repr__(self):
        return  "<id=%s,name=%s>" %(self.id,self.name)


class UserProfile(Base):
    __tablename__ = 'user_profile'
    id = Column(Integer,primary_key=True)
    username = Column(String(64),unique=True,nullable=False)
    password = Column(String(255),nullable=False)
    # host_list = relationship('HostUser',
    #                       secondary=UserProfile2HostUser,
    #                       backref='userprofiles')
    # groups = relationship('Group',
    #                       secondary=UserProfile2Group,
    #                       backref='userprofiles')
    def __repr__(self):
        return  "<id=%s,name=%s>" %(self.id,self.username)


class AuditLog(Base):
    __tablename__ = 'audit_log'
    id = Column(Integer,primary_key=True)
    userprofile_id = Column(Integer,ForeignKey('user_profile.id'))
    hostuser_id = Column(Integer,ForeignKey('host_user.id'))
    action_choices2 = [
        (u'cmd',u'CMD'),
        (u'login',u'Login'),
        (u'logout',u'Logout'),
    ]
    action_type = Column(ChoiceType(action_choices2))
    #action_type = Column(String(64))
    cmd = Column(String(255))
    date = Column(DateTime)

    # user_profile = relationship("UserProfile")
    #bind_host = relationship("BindHost")


engine = create_engine("mysql+pymsql://root:123@localhost:3306/stupid_jumpserver",echo=False)
Base.metadata.create_all(engine) # 创建所有表结构

方式一的查询作业答案

query = session.query(SysUsers).filter_by(name='root')
if query:
    for obj in query.scalar().ser_to_user:
        print(obj.servers.hostname)
上一篇下一篇

猜你喜欢

热点阅读