# 对象关系映射

数据库中的记录可以与一个 `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` 来实现这种对应：

In [1]:

```
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` 对象：

In [2]:

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

```

调用方法：

In [3]:

```
order.get_cost()

```

Out[3]:

```
-187540.0
```

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

In [4]:

```
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` 向数据库中添加刚才生成的对象：

In [5]:

```
session.add(order)
session.commit()

```

显示是否添加成功：

In [6]:

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

```

```
(u'A0001', u'2013-12-01', u'AAPL', 1000, 203.4)
(u'A0002', u'2013-12-01', u'MSFT', 1500, 167.5)
(u'A0003', u'2013-12-02', u'GOOG', 1500, 167.5)
(u'A0004', u'2015-09-10', u'MSFT', -1000, 187.54)

```

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

In [7]:

```
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

```

返回列表的第一个：

In [8]:

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

```

In [9]:

```
order_2.symbol

```

Out[9]:

```
u'MSFT'
```