Question

Python noob，

目前我正在使用SQLAlchemy，我有这个：

from __init__ import Base
from sqlalchemy.schema import Column, ForeignKey
from sqlalchemy.types import Integer, String
from sqlalchemy.orm import relationship

class User(Base):
    __tablename__ = "users"
    id = Column(Integer, primary_key=True)
    username = Column(String, unique=True)
    email = Column(String)
    password = Column(String)
    salt = Column(String)
    openids = relationship("OpenID", backref="users")

User.__table__.create(checkfirst=True)

#snip definition of OpenID class

def create(**kwargs):
    user = User()
    if "username" in kwargs.keys():
        user.username = kwargs['username']
    if "email" in kwargs.keys():
        user.username = kwargs['email']
    if "password" in kwargs.keys():
        user.password = kwargs['password']

    return user

这是/db/users.py，所以它将被用作：

from db import users
new_user = users.create(username="Carson", password="1234")
new_user.email = "email@address.com"
users.add(new_user) #this function obviously not defined yet

但是create()中的代码有点愚蠢，我想知道是否有更好的方法可以做到这一点，不需要if梯形图，如果添加任何键都会失败已经在User对象中了。像：

for attribute in kwargs.keys():
    if attribute in User:
        setattr(user, attribute, kwargs[attribute])
    else:
        raise Exception("blah")

这样我就可以将它放在自己的函数中（除非希望已经存在一个？）所以我不必一次又一次地执行if梯子，所以我可以在不修改代码的情况下更改表结构。 / p>

有什么建议吗？

Answer 1

我的建议是不要进一步简化它。如果您指定任意属性，则可能会踩到重要的对象结构。

我要做的一个简化是当你在字典上使用它时放弃.keys();包容检查和迭代都已使用密钥。

...

第二个想法，你可以有一个包含已知安全属性的class属性，然后在函数中检查这个属性，并在实例上使用setattr()。

Answer 2

实际上，声明性基类已经插入了您正在寻找的确切构造函数，如the declarative modules docs中所述。因此，只需执行User(username="Carson", password="1234")即可执行您想要的操作，User(something_not_an_attribute='foo')将引发异常。

Answer 3

如果您不需要涵盖继承的属性，

def create(**kwargs):
    keys_ok = set(User.__dict__)
    user = User()
    for k in kwargs:
        if k in keys_ok:
            setattr(user, k, kwargs[k])

如果您做需要涵盖继承的属性，inspect.getmembers可以提供帮助（使用自定义谓词来避免名称以下划线开头的成员，或者您想要确保的其他成员不能这样设置。）

如果set(kwargs) - set(keys_ok)不为空，我也会考虑（至少）发出警告 - 即，如果传递给create 的某些命名参数不能作为创建实例中的参数;这不是一件好事......！ - ）

从** kwargs设置构造期间类的属性

3 个答案: