在开发过程中不可避免会遇到存储，数据库是存储方式中的一种。
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)
上面是一些简单的操作，查看更多可以去看看官网。
