Create Profile model with is_pro_user attribute

Author: tdunn891

accounts/admin.py

@@ -1,3 +1,6 @@
 from django.contrib import admin
+from .models import Profile
 
 # Register your models here.
+
+admin.site.register(Profile)

accounts/models.py

@@ -1,3 +1,29 @@
 from django.db import models
+from django.contrib.auth.models import User
+from django.db.models.signals import post_save
+from django.dispatch import receiver
 
 # Create your models here.
+
+
+class Profile(models.Model):
+    user = models.OneToOneField(User, on_delete=models.CASCADE)
+    is_pro_user = models.BooleanField(default=False)
+
+    def __str__(self):
+        return self.user.username
+
+    # bio = models.TextField(max_length=500, blank=True)
+    # location = models.CharField(max_length=30, blank=True)
+    # birth_date = models.DateField(null=True, blank=True)
+
+
+@receiver(post_save, sender=User)
+def create_user_profile(sender, instance, created, **kwargs):
+    if created:
+        Profile.objects.create(user=instance)
+
+
+@receiver(post_save, sender=User)
+def save_user_profile(sender, instance, **kwargs):
+    instance.profile.save()

accounts/templates/profile.html

@@ -1,10 +1,84 @@
 {% extends 'base.html' %}
 {% block title %}{{ user }}'s Profile{% endblock %}
-{% block page_heading %}
-<h2>Profile</h2>
-<hr>
-{% endblock %}
+{% block page_heading %} <i class="material-icons">account_circle</i> My Account{% endblock %}
 {% block content %}
-{{ user }}
-<p>{{ profile.email }}</p>
-{% endblock %}
+<div class="row">
+   <div class="col-sm-2">
+      <p><i class="material-icons">account_circle</i></p>
+      <p><a href="#">Add Avatar</a></p>
+   </div>
+   <div class="col-sm-10">
+      <br>
+      <table class="table table-sm">
+         <tbody>
+            <tr>
+               <th>Username</th>
+               <td>{{ user.username }} </td>
+            </tr>
+            <tr>
+               <th>First Name</th>
+               <td>{{ user.first_name }} </td>
+            </tr>
+            <tr>
+               <th>Last Name</th>
+               <td>{{ user.last_name }}</td>
+            </tr>
+            <tr>
+               <th>Email</th>
+               <td>{{ user.email }} <a href="#" class="float-right">Update</a></td>
+            </tr>
+            <tr>
+               <th>Member Since</th>
+               <td>{{ user.date_joined.date }}</td>
+            </tr>
+            <tr>
+               <th>Staff</th>
+               <td>{{ user.is_staff }}</td>
+            </tr>
+            <tr>
+               <th>Current Plan</th>
+               {% if user.profile.is_pro_user %}
+               <td><span class="badge badge-primary">PRO</span></td>
+               {% else %}
+               <td>Basic</td>
+               {% endif %}
+            </tr>
+         </tbody>
+      </table>
+   </div>
+</div>
+<div class="row">
+   <div class="col-sm-12 text-center">
+      <h3>Plans</h3>
+      <br>
+   </div>
+</div>
+<div class="row">
+   <div class="col-sm-6 text-center">
+      <div class="jumbotron">
+         {% if user.profile.is_pro_user %}
+         <h5>BASIC</h5>
+         <p>Max Tickets/Month: 10/month</p>
+         {% else %}
+         <h5>BASIC <span class="badge badge-secondary">Current</span></h5>
+         <p>Max Tickets/Month: 10/month</p>
+         <p>Tickets Remaining This Month: placeholder</p>
+         {% endif %}
+      </div>
+   </div>
+   <div class="col-sm-6 text-center">
+      <div class="jumbotron">
+         {% if user.profile.is_pro_user %}
+         <h5>PRO <span class="badge badge-primary align-top">Current Plan</span></h5>
+         <p>Max Tickets/Month: Unlimited</p>
+         <p>Kanban View: Available</p>
+         {% else %}
+         <h5>PRO</h5>
+         Unlock <a href="#">Kanban View</a></p>
+         <p>9.90 EUR / month</p>
+         <p><a href="{% url 'checkout' %}" class="btn btn-primary">Go PRO</a></p>
+         {% endif %}
+      </div>
+   </div>
+</div>
+{% endblock %}

accounts/views.py

@@ -2,6 +2,7 @@
 from django.contrib import auth, messages
 from django.contrib.auth.decorators import login_required
 from django.contrib.auth.models import User
+from .models import Profile
 from accounts.forms import UserLoginForm, UserRegistrationForm
 
 # Create your views here.
@@ -67,13 +68,23 @@ def registration(request):
         "registration_form": registration_form})
 
 
+@login_required()
 def user_profile(request):
     """User's profile page"""
     user = User.objects.get(email=request.user.email)
-    return render(request, 'profile.html', {"profile": user})
+    return render(request, 'profile.html', {"user": user})
 
 
+@login_required()
 def user_list(request):
-    """Shows all users"""
-    users = User.objects.filter()
-    return render(request, 'user_list.html', {"users": users})
+    """Shows list of all users"""
+    # users = User.objects.filter()
+    sort_field = request.GET['sort_by'] if 'sort_by' in request.GET else 'id'
+    users = User.objects.filter().order_by(sort_field)
+    # Staff
+    staff = User.objects.filter(is_staff=1).order_by(sort_field)
+    # Submitters
+    submitters = User.objects.filter(is_staff=0).order_by(sort_field)
+    return render(request, 'user_list.html', {"users": users,
+                                              "staff": staff,
+                                              "submitters": submitters})