From 490bac968fd09a9d29eb88e7a56c8946383e38b2 Mon Sep 17 00:00:00 2001
From: Max Yakovenko <mxyakovenko9@gmail.com>
Date: Thu, 26 Jul 2018 00:39:55 +0300
Subject: [PATCH] fix bunch model deleting. Add missing required fields to
 admin user creation form

---
 accounts_ext/admin.py                         | 12 ++--
 .../admin/delete_selected_confirmation.html   | 55 +++++++++++++++++++
 2 files changed, 62 insertions(+), 5 deletions(-)
 create mode 100644 templates/admin/delete_selected_confirmation.html

diff --git a/accounts_ext/admin.py b/accounts_ext/admin.py
index 9fbafaf..a612a06 100644
--- a/accounts_ext/admin.py
+++ b/accounts_ext/admin.py
@@ -6,9 +6,11 @@ from django.utils.translation import ugettext_lazy as _
 from import_export import resources
 from jet.filters import RelatedFieldAjaxListFilter, DateRangeFilter
 
-from accounts_ext.forms import UserChangeForm, UserCreationForm
 from core.admin import SafeModelAdmin
-from .models import Profile, Company, UserPoints, Group
+from referral.admin import RefarralAdminInline
+
+from .forms import UserChangeForm, UserCreationForm
+from .models import Profile, Company, Group
 
 
 class CustomUserResource(resources.ModelResource):
@@ -23,11 +25,11 @@ class CustomUserResource(resources.ModelResource):
 @admin.register(get_user_model())
 class UserAdmin(SafeModelAdmin, BaseUserAdmin):
     inlines = [
-        # @TODO: ADD INLINE ADMIN FOR REFERRALS
+        RefarralAdminInline
     ]
 
     fieldsets = (
-        (None, {'fields': ('email', 'username', 'password')}),
+        (None, {'fields': ('email', 'username', 'password','referral_user')}),
         (_('Permissions'), {'fields': ('status', 'is_superuser',
                                        'groups', 'user_permissions')}),
         (_('Important dates'), {'fields': ('last_login', 'confirmed_at')}),
@@ -35,7 +37,7 @@ class UserAdmin(SafeModelAdmin, BaseUserAdmin):
     add_fieldsets = (
         (None, {
             'classes': ('wide',),
-            'fields': ('username', 'password1', 'password2'),
+            'fields': ('username', 'email', 'password1', 'password2', 'referral_user', 'status', 'is_superuser'),
         }),
     )
     list_display = ('email', 'username', 'status')
diff --git a/templates/admin/delete_selected_confirmation.html b/templates/admin/delete_selected_confirmation.html
new file mode 100644
index 0000000..dae2cbe
--- /dev/null
+++ b/templates/admin/delete_selected_confirmation.html
@@ -0,0 +1,55 @@
+{% extends "admin/base_site.html" %}
+{% load i18n l10n admin_urls static %}
+
+{% block extrahead %}
+    {{ block.super }}
+    {{ media }}
+    <script type="text/javascript" src="{% static 'admin/js/cancel.js' %}"></script>
+{% endblock %}
+
+{% block bodyclass %}{{ block.super }} app-{{ opts.app_label }} model-{{ opts.model_name }} delete-confirmation delete-selected-confirmation{% endblock %}
+
+{% block breadcrumbs %}
+<div class="breadcrumbs">
+<a href="{% url 'admin:index' %}">{% trans 'Home' %}</a>
+&rsaquo; <a href="{% url 'admin:app_list' app_label=opts.app_label %}">{{ opts.app_config.verbose_name }}</a>
+&rsaquo; <a href="{% url opts|admin_urlname:'changelist' %}">{{ opts.verbose_name_plural|capfirst }}</a>
+&rsaquo; {% trans 'Delete multiple objects' %}
+</div>
+{% endblock %}
+
+{% block content %}
+{% if perms_lacking %}
+    <p>{% blocktrans %}Deleting the selected {{ objects_name }} would result in deleting related objects, but your account doesn't have permission to delete the following types of objects:{% endblocktrans %}</p>
+    <ul>
+    {% for obj in perms_lacking %}
+        <li>{{ obj }}</li>
+    {% endfor %}
+    </ul>
+{% elif protected %}
+    <p>{% blocktrans %}Deleting the selected {{ objects_name }} would require deleting the following protected related objects:{% endblocktrans %}</p>
+    <ul>
+    {% for obj in protected %}
+        <li>{{ obj }}</li>
+    {% endfor %}
+    </ul>
+{% else %}
+    <p>{% blocktrans %}Are you sure you want to delete the selected {{ objects_name }}? All of the following objects and their related items will be deleted:{% endblocktrans %}</p>
+    {% include "admin/includes/object_delete_summary.html" %}
+    <h2>{% trans "Objects" %}</h2>
+    {% for deletable_object in deletable_objects %}
+        <ul>{{ deletable_object|unordered_list }}</ul>
+    {% endfor %}
+    <form method="post">{% csrf_token %}
+    <div>
+    {% for obj in queryset %}
+    <input type="hidden" name="{{ action_checkbox_name }}" value="{{ obj.pk|unlocalize }}" />
+    {% endfor %}
+    <input type="hidden" name="action" value="delete_selected_safe" />
+    <input type="hidden" name="post" value="yes" />
+    <input type="submit" value="{% trans "Yes, I'm sure" %}" />
+    <a href="#" class="button cancel-link">{% trans "No, take me back" %}</a>
+    </div>
+    </form>
+{% endif %}
+{% endblock %}