SQLAlchemy实例的更简洁的初始化

时间:2016-09-21 02:42:44

标签: python sqlalchemy

这是我第一次尝试sqlalchemy。我有一个带有我的usr信息的json文件,我想把它们放在一个sqlite3数据库文件中。它可以工作,但我发现实例初始化很详细,因为表中有很多列,如下所示。

是否可以使用字典作为输入来初始化User()?像a = User(usr)

这样的东西
import json                                                            
from sqlalchemy import *                                               
from sqlalchemy.ext.declarative import declarative_base                

engine = create_engine('sqlite:///tutorial.db', echo=True)             

Base = declarative_base()                                              

class User(Base):                                                      
    __tablename__ = 'users'                                            

    id = Column(Integer, primary_key=True)                             
    bbs_id = Column(String)                                            
    name = Column(String)                                              
    sex = Column(String, nullable=False)                               
    city = Column(String)                                              
    state = Column(String)                                             
    class_type = Column(String, nullable=False)                        
    class_id = Column(String, nullable=False)                          
    latitude = Column(Float)                                           
    longitude = Column(Float)                                          

    def __repr__(self):                                                
        return "<User(bbs_id='%s', name='%s'>" % (self.bbs_id, self.name)

Base.metadata.create_all(engine)                                       


with open('mydata.json') as fin:                                        
    usrs = json.load(fin)                                              

usr = usrs[0]                                                          

a = User(id=usr['id'], bbs_id=usr['bbs_id'], name=usr['name'])  

1 个答案:

答案 0 :(得分:2)

如果您知道JSON对象中的属性名称与Python模型的列名称匹配,则只需更改:

a = User(id=usr['id'], bbs_id=usr['bbs_id'], name=usr['name'])

为:

a = User(**usr)

Double-star/dict unpacking传递dict的每个键/值对,就像它是由关键字传递的参数一样。由于您没有覆盖模型的__init__,因此它已经允许并期望参数是可选的并通过关键字传递,因此这完全排列。