在開發(fā)過程中不可避免會遇到存儲，數(shù)據(jù)庫是存儲方式中的一種。
Python 操作mysql有幾個模塊:
- pymyql
- sqlalchemy
既然要操作數(shù)據(jù)庫，那么就得先有數(shù)據(jù)庫。
創(chuàng)建數(shù)據(jù)庫
使用命令行連接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語句只會一點(diǎn)點(diǎn)，下面就寫了一個小demo，具體可以網(wǎng)上找下mysql命令的學(xué)習(xí)，pymysql并不是本文的重點(diǎn)。
#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')
# 游標(biāo)
curosr = con.cursor(cursor=pymysql.cursors.DictCursor)
data = [
('小姐姐', 18),
('小姐姐2', 19)
]
try:
# 執(zhí)行sql命令
curosr.executemany('INSERT INTO meizi(name,age) VALUES (%s,%s)', data)
except Exception as e:
# 回滾
con.rollback()
print('事務(wù)出錯', e)
else:
# 提交事務(wù)
con.commit()
# 查詢
curosr.execute('SELECT * FROM meizi')
print(curosr.fetchall())
# 關(guān)閉游標(biāo)
curosr.close()
# 關(guān)閉連接
con.close()
if __name__ == '__main__':
func()
執(zhí)行結(jié)果：
結(jié)果.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將數(shù)據(jù)庫中的數(shù)據(jù)轉(zhuǎn)化成對象，當(dāng)然，也可以將對象持久化成數(shù)據(jù)庫中的數(shù)據(jù)。
安裝也是通過pip來進(jìn)行安裝：
pip install SQLAlchemy
我使用家里的網(wǎng)絡(luò)沒有安裝上，鏈接超時，然后我使用了國內(nèi)的鏡像：
https://pypi.douban.com/simple/ 豆瓣
http://mirrors.aliyun.com/pypi/simple/ 阿里
http://pypi.hustunique.com/simple/ 華中理工大學(xué)
http://pypi.sdutlinux.org/simple/ 山東理工大學(xué)
http://pypi.mirrors.ustc.edu.cn/simple/ 中國科學(xué)技術(shù)大學(xué)
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
# 創(chuàng)建基類
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)
# 連接數(shù)據(jù)庫 '數(shù)據(jù)庫類型+數(shù)據(jù)庫驅(qū)動名稱://用戶名:口令@機(jī)器地址:端口號/數(shù)據(jù)庫名'
engine = create_engine('mysql+pymysql://root:root@localhost:3307/mysql_test?charset=utf8')
# 創(chuàng)建所有繼承Base類的表結(jié)構(gòu)
Base.metadata.create_all(engine)
# 生成session類
session_class = sessionmaker(bind=engine)
# 實(shí)例session
session = session_class()
# 添加數(shù)據(jù)
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)])
# 提交數(shù)據(jù)
session.commit()
# Class表數(shù)據(jù)
print(session.query(Class).all())
# Student表數(shù)據(jù)
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())
# 關(guān)閉連接
session.close()
# 刪除表結(jié)構(gòu)
Base.metadata.drop_all(engine)
上面是一些簡單的操作，查看更多可以去看看官網(wǎng)。
