flask - flask 中不同类型的用户注册

Question

我是 Flask 的新手，我只是想用它创建一个简单的市场。我的网站上有三种类型的用户 [卖家、买家、管理员]。我认为每种类型的用户都应该有自己的注册和登录，因为他们的角色应该提供不同类型的信息。另一方面，每个用户在网站中可能有多个角色，因此我认为不可能为所有用户创建一个表并为他们分配角色。因此，我认为每种类型的用户都应该有三个表。

由于只有用户加载程序可供用户使用，我该如何实现我的目标。

我不确定它是否正确，我想我可以创建一个包含所有必需列的完整用户，并根据用户用于注册的地址，将正确的信息添加到数据库中。并根据用户用于登录的地址找到正确的角色并显示正确的 View 。

Answer 1

您可以为所有类型的用户使用一种模型:

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    usertype = db.Column(db.String(120))
    password_hash = db.Column(db.String(128))

用户加载程序将只有 ID :

@login.user_loader
def load_user(id):
    return User.query.get(int(id))

确保在注册期间您应该询问用户类型:

class RegistrationForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired()])
    email = StringField('Email', validators=[DataRequired(), Email()])
    usertype = SelectField('User Type', choices = [('sellers', 'sellers'),('buyers', 'buyers')], validators=[DataRequired()])
    password = PasswordField('Password', validators=[DataRequired()])
    password2 = PasswordField(
        'Repeat Password', validators=[DataRequired(), EqualTo('password')])
    submit = SubmitField('Register')

并且无需在登录时提供用户类型:

class LoginForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired()])
    password = PasswordField('Password', validators=[DataRequired()])
    remember_me = BooleanField('Remember Me')
    submit = SubmitField('Sign In')

在 HTMl 文件中，您可以相应地进行编码。以来自base.html 的这段代码为例。

    {% if current_user.usertype == 'buyers' %}
      <ul class="nav navbar-nav">
        <li class="dropdown">
          <a class="dropdown-toggle" data-toggle="dropdown" href="#">List
          <span class="caret"></span></a> 
          <ul class="dropdown-menu">
             <li><a href="{{ url_for('buyerslist')}}">List of buyers</a></li>
          </ul>
        </li>
      </ul>
    {% endif %}