一、数据创建

如果想要和 MySQL 交谈也得先通过 SQLAlchemy 建立一个会话:

from sqlalchemy.orm import sessionmaker

# 创建与数据库的会话session class ,注意,这里返回给session的是个class,不是实例
Session_class = sessionmaker(bind=engine)
session = Session_class()   # 生成session实例,相当于游标

通过session对象,session可以记录和跟踪数据的改变,在适当的时候提交,并且支持强大的ORM的功能。生成会话之后,我们需要一些测试数据。而 Faker 就是用来生成虚假数据的库。安装:

$ pip install faker

下面结合 Faker 库创建一些测试数据:

def Create(session):
    #创建了10个用户
    faker_users = [User(
        username=faker.name(),
        password=faker.word(),
        email=faker.email(),
        ) for i in range(10)]
    session.add_all(faker_users)

    # 5个分类
    faker_categories = [Category(name=faker.word()) for i in range(5) ]
    session.add_all( faker_categories )
    # 20个标签
    faker_tags = [Tag(name=faker.word()) for i in range(20)]
    session.add_all(faker_tags)
    # 100篇文章
    for i in range(100):
        article = Article(
            title = faker.sentence(),
            content = ' '.join( faker.sentences(nb=random.randint(10,20))),
            author = random.choice(faker_users),
            category = random.choice(faker_categories),
            )
        for tag in random.sample(faker_tags,random.randint(2,5)):
            article.tags.append(tag)
        session.add(article)

    session.commit()

if __name__ == '__main__':
    #Base.metadata.create_all(engine)

    faker = Factory.create()
    Session_class = sessionmaker(bind=engine)
    session = Session_class()

    Create(session)

在上面的代码中我们创建了10个用户,5个分类,20个标签,100篇文章,并且为每篇文章随机选择了2~5个标签。

使用 SQLAlchemy 往数据库中添加数据,我们只需要创建相关类的实例,调用 session.add() 添加一个,或者 session.add_all() 一次添加多个, 最后 session.commit() 就可以了。

二、查询

数据库最常用的就是查询功能,这里查询(Query)对象就是通过Session会话的query()方法获取的,需要注意的是这个方法的参数数目是可变的,也就是说我们可以传入任意多的参数数目,参数的类型可以是任意的类组合或者是类的名称。

1、返回列表和标量

(1).比如:

def Query_Iteration(session):
    # 此处得到Query对象返回的是一组可迭代的User实例表
    for user in session.query(User):
        print( "( username : %s ) , ( password : %s ),( email : %s )" 
            % (user.username,user.password , user.email) )

此处得到Query对象返回的是一组可迭代的User实例表,后我们通过for in语句访问,依次输出 用户名:user.username , 密码:user.password 和 邮箱:user.email 。
(2).当然如果只需要“用户名”和“密码”,对于对象实例的其他属性不感兴趣的话,也可以直接查询它们(类的属性名称)

def Query_Iteration1(session):
    for name,password in session.query(User.username , User.password):
        print( " %s , %s " % ( name , password) )

(3).SQLAlchemy的Query对象可以通过Python数组分片来实现控制记录的数目和位置。

def Query_Iteration2(session):
    for user in session.query(User).order_by(User.id)[1:3]:
        print(" %s , %s " % ( user.username,user.password) )

(4).假如我们需要筛选过滤特定结果,则可以使用filter_by()方法,这个方法使用关键词参数:

def Query_Iteration3(session):
    for user in session.query(User).filter_by( username='Cheryl Jackson'):
        print(user.password)

此处说明查询User表,按照User表中username=’Cheryl Jackson’过滤结果。

(5).Query对象的all()、one()以及first()方法将返回非迭代值,比如说all()返回的是一个列表,
first()方法限制并仅作为标量返回结果集的第一条记录,

2、使用原义SQL

Query对象能够灵活的使用原义SQL查询字符串作为查询参数。

def Query_SQL(session):
    users = session.execute('select * from users')
    for u in users:
        print(u)

三、更新和添加

# 更新一个字段
def Update(session):
    a = session.query(Article).get(10)
    a.title = 'test'
    session.add(a)
    session.commit()

session.query(Article).get(10)返回的是Article表中第10条记录。

# 插入新用户,
def Add_User(session):
    new_user = User( username = 'wang' , password='23',email='111@abc.com')
    session.add( new_user )
    session.commit()

四、删除

def Delete(session):
    a = session.query(Article).get(10)
    session.delete(a)
    session.commit()

删除直接调用 delete 删除获取到的对象,提交 session 即可。

五、完整代码

# coding: utf-8

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, String, Integer , Text
from sqlalchemy.orm import relationship
from sqlalchemy import Table,ForeignKey

import random
from sqlalchemy.orm import sessionmaker
from faker import Factory

# 初始化数据库连接:
# '数据库类型+数据库驱动名称://用户名:口令@机器地址:端口号/数据库名'
engine = create_engine('mysql+pymysql://root:0208@localhost:3306/blog?charset=utf8')

# 创建对象的基类:
Base = declarative_base()

class User(Base):
    # 表的名字:
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    username = Column(String(64), nullable=False, index=True)
    password = Column(String(64), nullable=False)
    email    = Column(String(64), nullable=False, index=True)
    articles = relationship('Article' , backref='author')
    userinfo = relationship('UserInfo',backref='user',uselist=False)

    def __repr__(self):
        return '%s(%r)' % (self.__class__.__name__, self.username)

class Article(Base):
    __tablename__ = 'articles'
    id      = Column( Integer, primary_key =True )
    title   = Column( String(255) , nullable=False , index = True )
    content = Column(Text)
    user_id = Column(Integer,ForeignKey('users.id'))
    cate_id = Column(Integer, ForeignKey('categories.id'))
    tags = relationship('Tag', secondary='article_tag', backref='articles')

    def __repr__(self):
        return '%s(%r)' % (self.__class__.__name__ , self.title)

class UserInfo(Base):
    __tablename__ = 'userinfos'
    id = Column(Integer, primary_key =True)
    name = Column( String(64))
    qq = Column(String(11))
    phone=Column(String(11))
    link=Column(String(64))
    user_id = Column(Integer,ForeignKey('users.id'))

class Category(Base):
    __tablename__ = 'categories'
    id = Column(Integer,primary_key=True)
    name = Column(String(64) ,nullable=False,index = True )
    articles = relationship('Article',backref='category')

# 中间表 自己创建。不需要手动管理,orm自动维护
article_tag = Table(
    'article_tag',Base.metadata,
    Column('article_id',Integer,ForeignKey('articles.id')),
    Column('tag_id',Integer,ForeignKey('tags.id'))
)

class Tag(Base):
    __tablename__='tags'
    id = Column(Integer,primary_key=True)
    name = Column( String(64) , nullable=False , index = True)

def CreateData(session):
    #创建了10个用户
    faker_users = [User(
        username=faker.name(),
        password=faker.word(),
        email=faker.email(),
        ) for i in range(10)]
    session.add_all(faker_users)

    # 5个分类
    faker_categories = [Category(name=faker.word()) for i in range(5) ]
    session.add_all( faker_categories )
    # 20个标签
    faker_tags = [Tag(name=faker.word()) for i in range(20)]
    session.add_all(faker_tags)
    # 100篇文章
    for i in range(100):
        article = Article(
            title = faker.sentence(),
            content = ' '.join( faker.sentences(nb=random.randint(10,20))),
            author = random.choice(faker_users),
            category = random.choice(faker_categories),
            )
        for tag in random.sample(faker_tags,random.randint(2,5)):
            article.tags.append(tag)
        session.add(article)

    session.commit()

# 更新一个字段
def Update(session):
    a = session.query(Article).get(20)
    a.title = 'test'
    session.add(a)
    session.commit() 

# 添加一个标签,在  tags表中添加了新的字段
def Add(session):
    a = session.query(Article).get(20)
    a.tags.append( Tag(name='Python test'))
    session.add(a)
    session.commit()

def Delete(session):
    a = session.query(Article).get(10)
    session.delete(a)
    session.commit()

def Query_Iteration(session):
    # 此处得到Query对象返回的是一组可迭代的User实例表
    for user in session.query(User).order_by(User.id):
        print( "( username : %s ) , ( password : %s ),( email : %s )" 
            % (user.username,user.password , user.email) )

#当然如果只需要“用户名”和“密码”,也可以直接查询它们(类的属性名称)
def Query_Iteration1(session):
    for name,password in session.query(User.username , User.password):
        print( " %s , %s " % ( name , password) ) 

def Query_Iteration2(session):
    for user in session.query(User).order_by(User.id)[1:3]:
        print(" %s , %s " % ( user.username,user.password) )

def Query_Iteration3(session):
    for user in session.query(User).filter_by( username='Cheryl Jackson'):
        print(user.password)

def Query_NonIterator(session):
        # all 则是获取所有
    print(session.query(User).all() )

def Query_SQL(session):
    users = session.execute('select * from users')
    for u in users:
        print(u)

# 插入新用户,
def Add_User(session):
    new_user = User( username = 'wang' , password='23',email='111@abc.com')
    session.add( new_user )
    session.commit()

if __name__ == '__main__':
    #Base.metadata.create_all(engine)

    faker = Factory.create()
    Session_class = sessionmaker(bind=engine)
    session = Session_class()

    #CreateData(session)

    #Update(session)
    #Add(session)
    Add_User(session)

    #Delete(session)
    session.close()