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的前提是了解關係資料庫的原理。