[英]Pass an argument to a class which becomes a keyword for a function
将WTForms与Flask和SQLAlchemy结合使用。 从用户名和电子邮件字段中获取数据,并确保它不在数据库中。 这就是我现在要做的。
class IsUnique(object):
def __init__(self, db_field=None):
self.db_field = db_field
def __call__(self, form, field):
data = field.data
if self.db_field=='name':
if User.query.filter_by(name=data).first() != None:
raise ValidationError('Sorry, that username is taken.')
if self.db_field=='email':
if User.query.filter_by(email=data).first() != None:
raise ValidationError(
'Sorry, that email address has already been registered.'
)
我想做的是将db_field参数作为字符串传递给类实例,然后将其传递到User.query.filter_by(db_field=data
。不幸的是,我所知道的全部操作是使用if语句,该语句可以工作,但是有点麻烦,必须有一种正确的方法,但是我不知道怎么做。
您可以将关键字参数作为dict传递,如下所示:
def __call__(self, form, field):
key = self.db_field # 'name' or 'email'
params = { key: field.data }
if User.query.filter_by(**params).first() != None:
raise ValidationError('Sorry, that {} is taken.'.format(key))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.