我的用户模型看起来像这样
class UserModel(Document):
first_name = StringField(required=True, max_length=50)
last_name = StringField(required=True, max_length=50)
username = StringField(required=True)
password = StringField(required=True, min_length=6)
def to_json(self):
return {
"id": str(self.pk),
"first_name": self.first_name,
"last_name": self.last_name,
"username": self.username,
"password": self.password
}
我的身份验证方法如下所示
def authenticate(username, password):
user = UserModel.objects.get(username=username)
if user and checkpw(password.encode('utf-8'), user.password.encode('utf-8')):
return user.to_json()
def identity(payload):
user_id = payload['identity']
u = UserModel.objects.get(pk=user_id)
return u.to_json
jwt = JWT(app, authenticate, identity) # /auth endpoint
我创建了to_json()方法,希望解决这个问题,并序列化主键,它将出现在像这样的普通查询中
"_id" : ObjectId("5aa9613d4fe35c23fca4d601")
instead of nicely like this~
"id": "5aa9613d4fe35c23fca4d601"
我得到的错误是:
"/home/joe/PyRest/FlaskMongo-venv/lib/python3.6/site-packages/flask_jwt/__init__.py", line 53, in _default_jwt_payload_handler
identity = getattr(identity, 'id') or identity['id']
AttributeError: 'dict' object has no attribute 'id'`
答案 0 :(得分:2)
这是Flask-JWT的常见抱怨。一眼就看起来它有7个开放的合并请求来修复它:https://github.com/mattupstate/flask-jwt/pulls?utf8=%E2%9C%93&q=is%3Apr+is%3Aopen+id
不幸的是,flask-jwt已经被遗弃了很长时间。也许请查看flask-jwt-extended或flask-jwt-simple作为仍然积极维护的替代品(和IMO更好的设计,但我是作者所以我当然有偏见);)