python使用資料庫
- 2020 年 1 月 14 日
- 筆記
使用python-DB-API,只需要搞清楚Connection、Cursor對象,打開後一定得關閉。
Cursor對象執行insert、update、delete,結果由rowcount返回影響的行數,拿到執行結果。
Cursor對象執行select,通過featchall()拿到list的結果集,每個元素都是一個tuple,對應一行記錄
SQL帶有參數,需要把參數按照位置傳遞給execute()方法。
安裝mysql官方提供的python驅動
pip install mysql-connection-python –allow-external mysql-connection-python
操作mysql的步驟:
import mysql.connector
conn=mysql.connector.connect(user='root', password='password', database='test')
cursor=conn.cursor()
cursor.execute('create table user(id varchar(20) primary key, name varchar(20))')
cursor.execute('insert into user(id, name) values(%s, %s)', ['1', 'Michael'])
cursor.rowcount
conn.commit()
cursor.close()
cursor = conn.cursor()
cursor.execute('select * from user where id = %s', ['1'])
values = cursor.fetchall()
values
cursor.close()
conn.close()
ORM技術 — Object-Relational Mapping,把關係資料庫的表接受映射到對象上。
ORM框架 — SQLAlchemy
SQLAlchemy的應用, 初始化DBSession
from sqlalchemy import Column, String, create_engine from sqlalchemy.orm import sessionmaker from sqlalchemy.ext.declarative import declarative_base Base = declarative_base() class User(Base): __tablename__ = 'user' id = Column(String(20), primary_key = True) name = Column(String(20)) engine = create_engine('mysql+mysqlconnector://root:password@localhost:3306/test') DBSession = sessionmaker(bind=engine)
一個表對應一個class。使用ORM添加記錄,可以看成為添加一個User對象。
session = DBSession() new_user=User(id='5', name='bob') session.add(new_user) session.commit() session.close()
添加數據的關鍵是獲取session,然後把對象添加到session,最後提交並關閉。DBSession對象可視為當前資料庫連接。
使用ORM從資料庫中查詢數據:結果是一個user對象,而不是tuple。
session = DBSession() user = session.query(User).filter(User.id=='5').one() print('type:', type(user) print('name:', user.name) session.close()
多個表通過外鍵實現一對多,多對多的關聯。ORM框架也可以實現一對多,多對多功能。
class User(Base): __tablename__ = 'user' id = Column(String(20), primary_key = True) name = Column(String(20)) books = relationship('Book') class Book(Base): __tablename__ = 'book' id = Column(String(20), primary_key=True) name = Column(String(20)) user_id = Column(String(20), ForeignKey('user,id'))
查詢一個User對象時,該對象的books屬性將返回一個包含若干個Book對象的list。
ORM的作用就是把資料庫表的一行記錄與一個對象做相互轉換, 使用ORM的前提是了解關係資料庫的原理。