对象关系映射

数据库中的记录可以与一个 Python 对象对应。

例如对于上一节中的数据库：

Order	Date	Stock	Quantity	Price
A0001	2013-12-01	AAPL	1000	203.4
A0002	2013-12-01	MSFT	1500	167.5
A0003	2013-12-02	GOOG	1500	167.5

可以用一个类来描述：

Attr.	Method
Order id	Cost
Date
Stock
Quant.
Price

可以使用 sqlalchemy 来实现这种对应：

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Date, Float, Integer, String

Base = declarative_base()

class Order(Base):
    __tablename__ = 'orders'
    
    order_id = Column(String, primary_key=True)
    date = Column(Date)
    symbol = Column(String)
    quantity = Column(Integer)
    price = Column(Float)
    
    def get_cost(self):
        return self.quantity*self.price

生成一个 Order 对象：

import datetime
order = Order(order_id='A0006', date=datetime.date.today(), symbol='MSFT', quantity=-1000, price=187.54)

调用方法：

order.get_cost()

-187540.0

使用上一节生成的数据库产生一个 session：

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

engine = create_engine("sqlite:///my_database.sqlite")   # 相当于 connection
Session = sessionmaker(bind=engine) # 相当于 cursor
session = Session()

使用这个 session 向数据库中添加刚才生成的对象：

session.add(order)
session.commit()

显示是否添加成功：

for row in engine.execute("SELECT * FROM orders"):
    print (row)

('A0001', '2013-12-01', 'AAPL', 1000, 203.4)
('A0002', '2013-12-01', 'MSFT', 1500, 167.5)
('A0003', '2013-12-02', 'GOOG', 1500, 167.5)
('A0004', '2015-09-10', 'MSFT', -1000, 187.54)
('A0007', '2018-12-01', 'AAPLG', 2000, 203.4)
('A0008', '2013-12-01', 'MSFT', 1500, 167.5)
('A0009', '2013-12-02', 'GOOG', 1500, 167.5)
('A0006', '2018-12-24', 'MSFT', -1000, 187.54)

使用 filter 进行查询，返回的是 Order 对象的列表：

for order in session.query(Order).filter(Order.symbol=="AAPL"):
    print (order.order_id, order.date, order.get_cost())

A0001 2013-12-01 203400.0

返回列表的第一个：

order_2 = session.query(Order).filter(Order.order_id=='A0002').first()

order_2.symbol

'MSFT'