在开发过程中不可避免会遇到存储，数据库是存储方式中的一种。

Python 操作mysql有几个模块:

• pymyql
• sqlalchemy

既然要操作数据库，那么就得先有数据库。

创建数据库

使用命令行连接mysql,也可以使用一些可视化工具或者插件。

λ mysql -P 3307 -uroot -p
Enter password: ****


mysql> create database mysql_test charset utf8;
Query OK, 1 row affected (0.00 sec)

mysql> show databases;
+--------------------+
| Database           |
+--------------------+
| information_schema |
| a                  |
| amazon             |
| amazon_data        |
| ipproxy            |
| jd                 |
| jd_analysis        |
| mysql              |
| mysql_test         |
| performance_schema |
| proxy              |
| sys                |
| video              |
+--------------------+
13 rows in set (0.00 sec)

mysql> use mysql_test;
Database changed

mysql> create table meizi(
    -> id int not null primary key auto_increment,
    -> name char(20) not null,
    -> age int(4) not null);
Query OK, 0 rows affected (0.04 sec)

mysql> show tables;
+----------------------+
| Tables_in_mysql_test |
+----------------------+
| meizi                |
+----------------------+
1 row in set (0.00 sec)

mysql> desc meizi;
+-------+----------+------+-----+---------+----------------+
| Field | Type     | Null | Key | Default | Extra          |
+-------+----------+------+-----+---------+----------------+
| id    | int(11)  | NO   | PRI | NULL    | auto_increment |
| name  | char(20) | NO   |     | NULL    |                |
| age   | int(4)   | NO   |     | NULL    |                |
+-------+----------+------+-----+---------+----------------+
3 rows in set (0.00 sec)

我使用的是cmder,所以前面有一个λ。

Pymysql

安装直接用pip安装

pip install pymysql

由于本人能力有限，sql语句只会一点点，下面就写了一个小demo，具体可以网上找下mysql命令的学习，pymysql并不是本文的重点。

#pymysql_test.py

import pymysql

def func():
    # 连接
    con = pymysql.connect(host='127.0.0.1', port=3307, user='root', 
                             password='root', db='mysql_test', charset='utf8')
    # 游标
    curosr = con.cursor(cursor=pymysql.cursors.DictCursor)
    data = [
        ('小姐姐', 18),
        ('小姐姐2', 19)
    ]
    try:
        # 执行sql命令
        curosr.executemany('INSERT INTO meizi(name,age) VALUES (%s,%s)', data)
    except Exception as e:
        # 回滚
        con.rollback()
        print('事务出错', e)
    else:
        # 提交事务
        con.commit()
    # 查询
    curosr.execute('SELECT * FROM meizi')
    print(curosr.fetchall())
    # 关闭游标
    curosr.close()
    # 关闭连接
    con.close()


if __name__ == '__main__':
    func()

执行结果：

结果.png

mysql> select * from meizi;
+----+---------+-----+
| id | name    | age |
+----+---------+-----+
|  1 | 小姐姐  |  18 |
|  2 | 小姐姐2 |  19 |
+----+---------+-----+
2 rows in set (0.00 sec)

Sqlalchemy

sqlalchemy是一个orm(Object Relational Mapping)映射框架，简单来说，可以通过orm将数据库中的数据转化成对象，当然，也可以将对象持久化成数据库中的数据。

安装也是通过pip来进行安装：

pip install SQLAlchemy

我使用家里的网络没有安装上，链接超时，然后我使用了国内的镜像：

https://pypi.douban.com/simple/ 豆瓣
http://mirrors.aliyun.com/pypi/simple/ 阿里
http://pypi.hustunique.com/simple/ 华中理工大学
http://pypi.sdutlinux.org/simple/ 山东理工大学
http://pypi.mirrors.ustc.edu.cn/simple/ 中国科学技术大学
https://pypi.tuna.tsinghua.edu.cn/simple 清华

具体使用：

pip install -i https://pypi.douban.com/simple/  SQLAlchemy

一大段代码来袭...

# sqlalchemy_test.py


from sqlalchemy import Column, Integer, String, create_engine, ForeignKey, and_, or_, func
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy.ext.declarative import declarative_base

# 创建基类
Base = declarative_base()


# 映射表
class Class(Base):
    __tablename__ = 'class'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(20), nullable=False)
    student = relationship("Student", backref='class')

    def __repr__(self):
        return '<Class({},{})>'.format(self.id, self.name)


class Student(Base):
    __tablename__ = 'student'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(20), nullable=False)
    age = Column(Integer, nullable=False)
    class_id = Column(Integer, ForeignKey('class.id'))

    def __repr__(self):
        return '<Student({},{},{},{})>'.format(self.id, self.name, self.age, self.class_id)


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

# 创建所有继承Base类的表结构
Base.metadata.create_all(engine)

# 生成session类
session_class = sessionmaker(bind=engine)
# 实例session
session = session_class()

# 添加数据
session.add(Class(name=u'meizi'))
# 查询id
class_id = session.query(Class.id).filter_by(name='meizi').first().id
session.add_all(
    [Student(name=u'xiaojiejie', age=18, class_id=class_id), Student(name=u'xiaojiejie2', age=19, class_id=class_id)])
# 提交数据
session.commit()
# Class表数据
print(session.query(Class).all())
# Student表数据
print(session.query(Student).all())

# 过滤 排序 限制
print(session.query(Student).filter(Student.name.like('%xiao%')).order_by(Student.id.desc()).limit(1).all())
# 通过主键
print(session.query(Student).get(1))
print(session.query(Student).filter(Student.id == 1).all())
# in
print(session.query(Student).filter(Student.name.in_(['xiaojiejie'])).all())
# not in
print(session.query(Student).filter(~Student.name.in_(['sujian'])).all())
# and
print(session.query(Student).filter(and_(~Student.name.in_(['sujian']), Student.age == 18)).all())
# or
print(session.query(Student).filter(or_(~Student.name.in_(['sujian']), Student.age != None)).all())
# count
print(session.query(Student).filter_by(name='').count())
print(session.query(func.count(), Student.name).group_by(Student.name).all())
# 多表
print(session.query(Class, Student).filter(Class.id == Student.class_id).all())
# 修改
session.query(Student).filter(Student.id == 1).update({Student.age: 20})
session.commit()
print(session.query(Student).filter_by(id=1).first().age)
# 修改
session.query(Student).filter_by(id=1).first().age = 21
session.commit()
print(session.query(Student).filter(Student.id == 1).first().age)

# 删除
session.delete(session.query(Student).get(1))
print(session.query(Student).all())
# 回滚
session.rollback()
print(session.query(Student).all())
# 关闭连接
session.close()

# 删除表结构
Base.metadata.drop_all(engine)

上面是一些简单的操作，查看更多可以去看看官网。