SQLAlchemy：替換對象與一個新的，以下默認值

Question

我想創建一個SQLAlchemy對象的新實例，以便字段充滿默認值，但我想提交到數據庫生成一行UPDATE已經存在相同的主鍵，有效地將其重置爲默認值。有沒有簡單的方法來做到這一點？

Answer 1

我試圖做到這一點，並失敗，因爲SQLAlchemy會話跟蹤對象的狀態。所以沒有簡單的方法讓會話將新對象作爲持久對象來跟蹤。

但是你想將對象重置爲默認值，是嗎？有一個簡單的方法可以做到這一點：

from sqlalchemy.ext.declarative import declarative_base 

class Base(object): 

    def reset(self): 
     for name, column in self.__class__.__table__.columns.items(): 
      if column.default is not None: 
       setattr(self, name, column.default.execute()) 

Base = declarative_base(bind=engine, cls=Base) 

這將reset方法添加到所有模型類。

下面是完整的工作示例亂動：

import os 
from datetime import datetime 


from sqlalchemy import create_engine 
from sqlalchemy import Column, Integer, String, DateTime 
from sqlalchemy.orm import sessionmaker 
from sqlalchemy.ext.declarative import declarative_base 
from sqlalchemy.sql import functions 


here = os.path.abspath(os.path.dirname(__file__)) 
engine = create_engine('sqlite:///%s/db.sqlite' % here, echo=True) 
Session = sessionmaker(bind=engine) 


class Base(object): 

    def reset(self): 
     for name, column in self.__class__.__table__.columns.items(): 
      if column.default is not None: 
       setattr(self, name, column.default.execute()) 

Base = declarative_base(bind=engine, cls=Base) 


class Thing(Base): 
    __tablename__ = 'things' 

    id = Column(Integer, primary_key=True) 
    value = Column(String(255), default='default') 
    ts1 = Column(DateTime, default=datetime.now) 
    ts2 = Column(DateTime, default=functions.now()) 

    def __repr__(self): 
     return '<Thing(id={0.id!r}, value={0.value!r}, ' \ 
       'ts1={0.ts1!r}, ts2={0.ts2!r})>'.format(self) 


if __name__ == '__main__': 
    Base.metadata.drop_all() 
    Base.metadata.create_all() 

    print("---------------------------------------") 
    print("Create a new thing") 
    print("---------------------------------------") 
    session = Session() 
    thing = Thing(
     value='some value', 
     ts1=datetime(2014, 1, 1), 
     ts2=datetime(2014, 2, 2), 
    ) 
    session.add(thing) 
    session.commit() 
    session.close() 

    print("---------------------------------------") 
    print("Quering it from DB") 
    print("---------------------------------------") 
    session = Session() 
    thing = session.query(Thing).filter(Thing.id == 1).one() 
    print(thing) 
    session.close() 

    print("---------------------------------------") 
    print("Reset it to default") 
    print("---------------------------------------") 
    session = Session() 
    thing = session.query(Thing).filter(Thing.id == 1).one() 
    thing.reset() 
    session.commit() 
    session.close() 

    print("---------------------------------------") 
    print("Quering it from DB") 
    print("---------------------------------------") 
    session = Session() 
    thing = session.query(Thing).filter(Thing.id == 1).one() 
    print(thing) 
    session.close() 

Answer 2

有沒有什麼簡單的方法來做到這一點？

經過進一步的考慮，並不是真的。最簡潔的方法是在__init__中定義默認值。從DB獲取對象時從不調用構造函數，所以它非常安全。您還可以使用後端功能，例如current_timestamp()。

class MyObject(Base): 
    id = Column(sa.Integer, primary_key=True) 
    column1 = Column(sa.String) 
    column2 = Column(sa.Integer) 
    columnN = Column(sa.String) 
    updated = Column(sa.DateTime) 

    def __init__(self, **kwargs): 
     kwargs.setdefault('column1', 'default value') 
     kwargs.setdefault('column2', 123) 
     kwargs.setdefault('columnN', None) 
     kwargs.setdefault('updated', sa.func.current_timestamp()) 
     super(MyObject, self).__init__(**kwargs) 

default_obj = MyObject() 
default_obj.id = old_id 
session.merge(default_obj) 
session.commit()