add show_avatar and avatar_field logic

Author: wuyue92tree

adminlteui/admin.py

@@ -69,6 +69,14 @@ class GeneralOptionForm(forms.Form):
         help_text=_("Login page welcome sign.")
     )
 
+    avatar_field = forms.CharField(label=_('Avatar Field'),
+                                   widget=widgets.AdminTextInputWidget(),
+                                   required=False,
+                                   help_text=_(
+                                       "which field is avatar."))
+    show_avatar = forms.BooleanField(
+        label=_('Show Avatar'), required=False)
+
     def save(self):
         try:
             # clear site-logo
@@ -77,6 +85,16 @@ def save(self):
                 obj.delete()
                 self.changed_data.remove('site_logo')
 
+            if not self.data.get('show_avatar'):
+                try:
+                    obj = Options.objects.get(option_name='show_avatar')
+                    obj.option_value = 'off'
+                    obj.save()
+                except Exception:
+                    obj = Options.objects.create(option_name='show_avatar',
+                                                 option_value='off')
+                    obj.save()
+
             for data_item in self.changed_data:
                 try:
                     obj = Options.objects.get(option_name=data_item)
@@ -164,7 +182,11 @@ def general_option_view(self, request):
                 'welcome_sign': get_option(option_name='welcome_sign'),
                 'site_logo': ImageBox(
                     get_option(option_name='site_logo')) if get_option(
-                    option_name='site_logo') else ''
+                    option_name='site_logo') else '',
+                'show_avatar': True if get_option(
+                    option_name='show_avatar') == 'on' else False,
+                'avatar_field': get_option(
+                    option_name='avatar_field') or 'request.user.head_avatar',
             }
             form = GeneralOptionForm(initial=initial_value)
         else:
@@ -183,7 +205,11 @@ def general_option_view(self, request):
                     'welcome_sign': get_option(option_name='welcome_sign'),
                     'site_logo': ImageBox(
                         get_option(option_name='site_logo')) if get_option(
-                        option_name='site_logo') else ''
+                        option_name='site_logo') else '',
+                    'show_avatar': True if get_option(
+                        option_name='show_avatar') == 'on' else False,
+                    'avatar_field': get_option(
+                        option_name='avatar_field') or 'request.user.head_avatar',
                 }
                 form = GeneralOptionForm(initial=initial_value)
                 messages.add_message(request, messages.SUCCESS,

adminlteui/templates/admin/base.html

@@ -76,14 +76,28 @@
                     <!-- User Account: style can be found in dropdown.less -->
                     <li class="dropdown user user-menu">
                         <a href="#" class="dropdown-toggle" data-toggle="dropdown">
+                            {% get_adminlte_option 'show_avatar' as adminlte_show_avatar  %}
+                            {% if adminlte_show_avatar.valid and adminlte_show_avatar.show_avatar == 'on' %}
+                            {% get_adminlte_option 'avatar_field' request as adminlte_avatar_field  %}
+                            {% if adminlte_avatar_field.valid and adminlte_avatar_field.avatar_field %}
+                            <img src="{{ adminlte_avatar_field.avatar_field }}" class="user-image" alt="User Image" style="width: 25px; height:25px;">
+                            {% else %}
                             <img src={% static "admin/dist/img/default.jpg" %} class="user-image" alt="User Image">
+                            {% endif %}
+                            {% endif %}
                             <span class="hidden-xs">{{ request.user.username }}</span>
                         </a>
                         <ul class="dropdown-menu">
                             <!-- User image -->
+                            {% get_adminlte_option 'show_avatar' as adminlte_show_avatar  %}
+                            {% if adminlte_show_avatar.valid and adminlte_show_avatar.show_avatar == 'on' %}
                             <li class="user-header">
+                                {% get_adminlte_option 'avatar_field' request as adminlte_avatar_field  %}
+                                {% if adminlte_avatar_field.valid and adminlte_avatar_field.avatar_field %}
+                                <img src="{{ adminlte_avatar_field.avatar_field }}" class="img-circle" alt="User Image" style="width: 90px; height:90px;">
+                                {% else %}
                                 <img src={% static "admin/dist/img/default.jpg" %} class="img-circle" alt="User Image">
-
+                                {% endif %}
                                 <p>
                                     {{ request.user.username }} -
                                     {% if request.user.is_superuser %}
@@ -98,6 +112,24 @@
                                     <small>{% trans "Register time" %}： {{ request.user.date_joined }}</small>
                                 </p>
                             </li>
+                            {% else %}
+                            <li class="user-header" style="height: auto;">
+                                <p>
+                                    {{ request.user.username }} -
+                                    {% if request.user.is_superuser %}
+                                    {% trans "Super manager" %}
+                                    {% else %}
+                                    {% if request.user.groups.all %}
+                                    {% for i in request.user.groups.all %}{{ i.name }}{% endfor %}
+                                    {% else %}
+                                    {% trans 'Normal' %}
+                                    {% endif %}
+                                    {% endif %}
+                                    <small>{% trans "Register time" %}： {{ request.user.date_joined }}</small>
+                                </p>
+                            </li>
+                            {% endif %}
+
                             <!-- Menu Body -->
                             <!--<li class="user-body">-->
                             <!--<div class="row">-->
@@ -137,15 +169,23 @@
         <!-- sidebar: style can be found in sidebar.less -->
         <section class="sidebar">
             <!-- Sidebar user panel -->
+            {% get_adminlte_option 'show_avatar' as adminlte_show_avatar  %}
+            {% if adminlte_show_avatar.valid and adminlte_show_avatar.show_avatar == 'on' %}
             <div class="user-panel">
                 <div class="pull-left image">
+                    {% get_adminlte_option 'avatar_field' request as adminlte_avatar_field  %}
+                    {% if adminlte_avatar_field.valid and adminlte_avatar_field.avatar_field %}
+                    <img src="{{ adminlte_avatar_field.avatar_field }}" class="img-circle" alt="User Image" style="width: 45px; height:45px;">
+                    {% else %}
                     <img src={% static "admin/dist/img/default.jpg" %} class="img-circle" alt="User Image">
+                    {% endif %}
                 </div>
                 <div class="pull-left info">
                     <p>{{ request.user.username }}</p>
                     <a href="#"><i class="fa fa-circle text-success"></i> {% trans "Online" %}</a>
                 </div>
             </div>
+            {% endif %}
             <!-- search form -->
             <form action="#" method="get" class="sidebar-form">
                 <div class="input-group">
@@ -460,8 +500,8 @@ <h3 class="control-sidebar-heading">Chat Settings</h3>
                 if (menuActions[i].href===window.location.origin + pathname) {
                     continue
                 }
-
                 console.log('menu matched.', window.location.href, menuActions[i].href);
+                menuActions[i].parentNode.setAttribute('class', 'active');
                 if (menuActions[i].parentNode.parentNode.getAttribute('class') !== 'sidebar-menu') {
                     menuActions[i].parentNode.parentNode.setAttribute('class', 'treeview-menu menu-open');
                     menuActions[i].parentNode.parentNode.parentNode.setAttribute('class', 'treeview active');

adminlteui/templatetags/adminlte_options.py

@@ -1,17 +1,30 @@
+import traceback
 from django import template
 from adminlteui.models import Options
+from django.conf import settings
 
 register = template.Library()
 
 
 @register.simple_tag
-def get_adminlte_option(option_name):
+def get_adminlte_option(option_name, request=None):
     config_ = {}
     config_list = Options.objects.filter(valid=True)
 
     if config_list.filter(option_name=option_name):
         config_[option_name] = config_list.get(
             option_name=option_name).option_value
+        if request and option_name=='avatar_field':
+            try:
+                # request.user.head_avatar
+                image_path = eval(config_[option_name]).name
+                if image_path:
+                    config_[option_name] = settings.MEDIA_URL + image_path
+                else:
+                    config_[option_name] = None
+            except Exception as e:
+                traceback.print_exc()
+                config_[option_name] = None
         config_['valid'] = config_list.get(
             option_name=option_name).valid
     return config_