Djnagoでログインした後の遷移先を変えたい

Question

前提

Djangoでスケジュール調整Webアプリを作っています。
https://qiita.com/knakajima3027/items/34b2a105da7cdb411736
ここを参考にログイン機能を作りました。

実現したいこと

Djangoでログインした際に、データベースにあるユーザー情報を取得して、取得したデータによって
遷移先を変えたいです。

まずWebアプリにアクセスするとログイン画面が表示され、メアドとパスワードを入力しログインします。

そこで、ログインしたユーザーのユーザー情報を参照し、役職レコードがあるのでそこを見て、
役員（０）だったら、役員用のページ　平社員（１）だったら　平社員用のページ　という感じです。

ログイン用アカウントのデータベースの中身

メールアドレス:[e-mail]
Code:　　　　1000(プライマリキー)
Name:　　　役員１　（その人の名前）
Post:　　　　0　（0:役員、1:社員）
Belong:　　　0000 (その人の部署)

setting.py

LANGUAGE_CODE = "ja"

TIME_ZONE = "Asia/Tokyo"

LOGIN_URL = "account:login"

LOGIN_REDIRECT_URL = "account:home"

LOGOUT_REDIRECT_URL = "account:top"

USE_I18N = True

USE_TZ = True


# Static files (CSS, JavaScript, Images)
# https://docs.djangoproject.com/en/4.1/howto/static-files/

STATIC_URL = "static/"

STATICFILES_DIRS = (os.path.join(BASE_DIR,'static'),)

# Default primary key field type
# https://docs.djangoproject.com/en/4.1/ref/settings/#default-auto-field

DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField"

AUTH_USER_MODEL = 'account.SyainMaster'

login.html

{% extends "account/base.html" %}
{% block title %}ログイン{% endblock %}

{% block content %}
<div class="container">
    <div class="head">
        <header>
            <a class="headmoji">スケジュールアプリ</a>
        </header>
    </div>
    <p>{{message}}</p>
    <form action="{% url 'account:login'%}" method="post">
    {% csrf_token %}
        {{ form.as_p }}
        <p><input type="hidden" name="next" value="{{next}}"></p>
        <p><input type="submit" value="ログイン"></p>
    </form>
</div>
{% endblock %}

form.py

class LoginForm(AuthenticationForm):
    class Meta:
        model = SyainMaster
        fields = ['email', 'password']

views.py

class HomeView(LoginRequiredMixin, TemplateView):
    
    def get(self, request, *args, **kwargs):
        if request.POST['post'] == 0:
            template_name = "account/Yakuinhome.html"                 <--ここが問題？？？
        else:
            template_name = "account/Syainhome.html"
    
    
    def get_context_data(self, **kwargs):
        content = super().get_context_data(**kwargs)
        content.update({
                'MendanMaster':MendanMaster.objects.all(),
                })
        
        return content
    

class LoginView(LoginView):
    """ログインページ"""
    form_class = forms.LoginForm
    template_name = "account/login.html"

urls.py

from django.urls import path

from . import views


app_name="account"
urlpatterns = [
   
    path("", views.LoginView.as_view(), name="login"),
    path("", views.LogoutView.as_view(), name="logout"),
    path("home/", views.HomeView.as_view(), name="home"),
    path("sakusei/", views.CreateMendanView.as_view(), name="sakusei"),
    path("itiran/", views.sakusei_view, name="itiran"),
    path("Ysyousai/",views.SyousaiMendanView.as_view(), name="Ysyousai"),
    path("Ysyousai_K/",views.Syousai_K_MendanView.as_view(), name="Ysyousai_K" ),
    path("Syozoku/",views.SyozokuView.as_view(), name="Syozoku"),
    path("Select/",views.SelectTimeView.as_view(), name="Select"),
    path("Ssyousai/",views.SyainSyousaiMendanView.as_view(), name="Ssyousai"),
    path("Ssyousai_K/",views.SyainSyousai_K_MendanView.as_view(), name="Ssyousai_K" ),
    
]

Comments

[m.ts10806]

＞Djangoでログインした際に

そもそもログインってDBにあるユーザ情報と照合するので
「ログイン」時点でユーザ情報取得できてるのでは。

[Colton]

できてないんですよね.....
どうやったら遷移先をわけられるんでしょうか...

[m.ts10806]

では、まずログイン画面で入力された入力内容からDBを検索するところから。
それができれば検索結果のユーザ情報を見てリダイレクト
もしくは、同じテンプレートファイルでも分岐により出力内容を変える。セッションも必要です。

Answer 1

def get(self)で、ユーザーの情報を取得し、 return [template_name]　で　テンプレートを更新できるようです。

views.py

class HomeView(LoginRequiredMixin, TemplateView):
        
    template_name = "account/Yakuinhome.html"
    
    def get_template_names(self):
        
        if self.request.user.post == 0:　　#ログインしたユーザーの「post」が「0」だった場合
            template_name = "account/Yakuinhome.html"
        else:　　　　　　　　　　　　　　#そうじゃない場合
            template_name = "account/Syainhome.html"
            
        return [template_name]