1

我正在尝试在我的应用程序中实现烧瓶登录，它也使用烧瓶蒙戈引擎。这是我的用户原型：

class User(db.Document, UserMixin):
    username = db.StringField(max_length=80)
    email = db.StringField(max_length=255, unique=True)
    password = db.StringField(max_length=255, required=True)
    active = db.BooleanField(default=True)

    def __init__(self, *args, **kwargs):
        super(db.Document, self).__init__(self)
        try:
            self.username = kwargs['username']
            self.email = kwargs['email']
            self.password = kwargs['password']
        except:
            flash('Bad arguments for User')

    @staticmethod
    def salt_password(password):
        return generate_password_hash(password)

    @property
    def is_authenticated(self):
        return True

    @property
    def is_active(self):
        return self.active

    @property
    def is_anonymous(self):
        return False

    def get_id(self):
        return unicode(self._id)

    def __repr__(self):
        return '<User %r>' % (self.username)

    def check_pwd(self, password):
        return check_password_hash(self.password, password)

但是，当我login_user(user)从登录视图调用该函数时，它调用 的get_id方法User，但self._id返回None. 我也尝试self.id过同样的结果。然后，我尝试_id明确添加该字段：

class User(db.Document, UserMixin):
    _id = db.ObjectIdField(default=bson.ObjectId())

但是然后，self._id给了我'User u'username'>而不是用户ID。

关于如何检索_id用户的任何想法？

4

2 回答 2

1

我找到了解决此问题的方法。我的用户原型使用 pymongo 更新如下：

def get_id(self):
    user_queried = self._get_collection().find_one({'username':self.username, 'email':self.email, 'password':self.password})
    if user_queried is not None:
        return unicode(user_queried['_id'])
    else:
        return 'None'
于 2016-04-11T17:43:00.227 回答
1

虽然我不知道是否仍然需要回答您的问题 - 您找到了适合您的解决方案并且您的问题已经过时了。但是，如果其他人出现，我使用以下代码解决了它：

from mongoengine import Document, StringField, FloatField, BooleanField

class User(Document):

name = StringField(primary_key=True, required=True, max_length=50)
mail = StringField(required=True, max_length=255)
password = StringField(required=True, max_length=80)
first_name = StringField(max_length=255)
second_name = StringField(max_length=255)
authenticated = True  # BooleanField(required=True, default=True)
is_active = True  # BooleanField(required=True, default=True)

meta = {'db_alias': 'users'}

@property
def is_authenticated(self):
    return True

@property
def is_active(self):
    return True

@property
def is_anonymous(self):
    return False

def get_id(self):
    return self.name

并添加了以下 user_loader

@login_manager.user_loader
def load_user(user_id):
    user = User.objects(name=user_id)[0]
    return user
于 2017-05-30T10:40:10.587 回答