Question

我是Flask的初学者，我开始玩网络应用程序了。

我很难弄清楚如何强制使用唯一的用户名。我正在考虑如何在SQL中执行此操作，可能使用类似user_name text unique on conflict fail的内容，但是如何在Python中捕获错误？

或者，有没有办法管理内置于Flask的内容？

Answer 1

这完全取决于您的数据库层。虽然建议使用SQL Alchemy，但Flask非常具体地 not 与特定的ORM系统捆绑在一起。好消息是SQL Alchemy有一个独特的约束。

以下是它的工作原理：

from sqlalchemy.ext.declarative import declarative_base, InvalidRequestError
engine = #my engine
session = Session() # created by sessionmaker(bind=engine)
Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String, unique=True)

# then later...

user = User()
user.name = 'Frank'
session.add(user)

try:
    session.commit()
    print 'welcome to the club Frank'
except InvalidRequestError:
    print 'You are not Frank. Impostor!!!'

在“之后”两次之后运行该部件。第一次收到欢迎信息，第二次你不会。

附录：Flask与default authentication framework simply stores users in a dict by username最接近的事情。检查以强制执行唯一性的方法是手动测试，例如。

if username in digest_db:
    raise Exception('HEY! "{}" already exists! \
                     You can\'t do that'.format(username))
digest_db.add_user(username, password)

或覆盖RealmDigestDB以确保在添加之前检查：

class FlaskRealmDigestDB(authdigest.RealmDigestDB):
    def add_user(self, user, password):
        if user in self:
            raise AttributeError('HEY! "{}" already exists! \
                                  You can\'t do that'.format(user))
        super(FlaskRealmDigestDB, self).add_user(user, password)

    def requires_auth(self, f):
        # yada yada

或覆盖RealmDigestDB，并使其返回不允许重复分配的内容。例如

class ClosedDict(dict):
    def __setitem__(self, name, val):
       if name in self and val != self[name]:
          raise AttributeError('Cannot reassign {} to {}'.format(name, val))
    super(ClosedDict, self).__setitem__(name,val)


class FlaskRealmDigestDB(authdigest.RealmDigestDB):
    def newDB():
        return ClosedDict()

    def requires_auth(self, f):
        # yada yada

我把这个作为附录，因为该类不以任何方式保留数据，如果你计划扩展authdigest.RealmDigestDB，你应该使用像上面的SQLAlchemy这样的东西。

Answer 2

您可以使用SQLAlchemy.It是一个插件

如何在Flask中强制使用唯一的用户名？

2 个答案: