ログインユーザーの役割を判別し、新規登録できるようにしたい。

Question

実現したいこと

生産者とお客を結ぶSNSを作りたいです。
生産者とお客のモデルは別々に作成しています。

＜機能＞
生産者はログイン後、新規投稿、更新、削除ができる。

発生している問題・分からないこと

生産者としてログインしているのにもかかわらず、生産者としてログインしていることが判断できていません。

エラーメッセージ

新規投稿ボタンを押すと「生産者としてログインしてください」のメッセージが表示されます

該当のソースコード

# お客のモデルクラス
class Customer(db.Model, UserMixin):
    __tablename__ = 'customer'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(100), nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    password_hash = db.Column(db.String(128), nullable=False)
    buyer_profile = db.Column(db.Text, nullable=True)
    created_at = db.Column(db.DateTime, nullable=False, default=datetime.now(tokyo_timezone))
    updated_at = db.Column(db.DateTime, nullable=False, default=datetime.now(tokyo_timezone), onupdate=datetime.now(tokyo_timezone))

    def set_password(self, password):
        self.password_hash = generate_password_hash(password)

    def check_password(self, password):
        return check_password_hash(self.password_hash, password)

    @property
    def is_farmer(self):
        return False  # Customer には生産者のフラグは False に設定


# 生産者のモデルクラス
class Farmer(db.Model, UserMixin):
    __tablename__ = 'farmer'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(100), nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    password_hash = db.Column(db.String(128), nullable=False)
    farmer_profile = db.Column(db.String(1000), nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, default=datetime.now(tokyo_timezone))
    updated_at = db.Column(db.DateTime, nullable=False, default=datetime.now(tokyo_timezone), onupdate=datetime.now(tokyo_timezone))

    products = db.relationship('Product', back_populates='farmer', cascade='all, delete-orphan')

    def set_password(self, password):
        self.password_hash = generate_password_hash(password)

    def check_password(self, password):
        return check_password_hash(self.password_hash, password)

    @property
    def is_farmer(self):
        return True  # Farmer には生産者のフラグを True に設定

@marche_bp.route('/create', methods=['GET', 'POST'])
@login_required
def create():
    if not current_user.is_farmer:  # current_userが生産者でない場合、エラーメッセージを表示
        flash("生産者のアカウントでログインしてください", "danger")
        return redirect(url_for('marche.index'))

    if request.method == 'POST':
        product_name = request.form.get('product_name')
        detail = request.form.get('detail')
        file = request.files.get('img_name')

        if file:
            filename = file.filename
            save_path = os.path.join('static', 'images', filename)
            file.save(save_path)
        else:
            filename = 'default.jpg'

        post = Product(product_name=product_name, detail=detail, img_name=filename, farmer_id=current_user.id)
        db.session.add(post)
        db.session.commit()

        flash("登録しました", "success")
        return redirect(url_for('marche.lists'))

    return render_template("marche/create.html")

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

ChatGPTも使って修正しましたが、解決できませんでした。

補足

Flask==3.1.0

生産者とお客のモデルをそれぞれ別に作っていますが、Userのモデルは１つにした方がいいのでしょうか？

Answer 1

自己解決しました。
Login関数で生産者とお客の判定ができていませんでした。
Login関数を修正後、無事に動くようになりました。