@@ -43,38 +43,57 @@
{% endblock %}
+{% block table_container %}
+
+
+
+ | {% trans 'ID' %} |
+ {% trans 'Username' %} |
+ {% trans 'Type' %} |
+ {% trans 'UA' %} |
+ {% trans 'IP' %} |
+ {% trans 'City' %} |
+ {% trans 'MFA' %} |
+ {% trans 'Reason' %} |
+ {% trans 'Status' %} |
+ {% trans 'Date' %} |
+
+
+
+
+ {% for login_log in object_list %}
+
+ | {{ forloop.counter }} |
+ {{ login_log.username }} |
+ {{ login_log.get_type_display }} |
+
+ {{ login_log.user_agent | truncatechars:20 }}
+ |
+ {{ login_log.ip }} |
+ {{ login_log.city }} |
+ {{ login_log.get_mfa_display }} |
+ {{ login_log.get_reason_display }} |
+ {{ login_log.get_status_display }} |
+ {{ login_log.datetime }} |
+
+ {% endfor %}
+
+
+
-{% block table_head %}
-
{% trans 'ID' %} |
-
{% trans 'Username' %} |
-
{% trans 'Type' %} |
-
{% trans 'UA' %} |
-
{% trans 'IP' %} |
-
{% trans 'City' %} |
-
{% trans 'MFA' %} |
-
{% trans 'Reason' %} |
-
{% trans 'Status' %} |
-
{% trans 'Date' %} |
{% endblock %}
-{% block table_body %}
- {% for login_log in object_list %}
-
- | {{ forloop.counter }} |
- {{ login_log.username }} |
- {{ login_log.get_type_display }} |
-
- {{ login_log.user_agent | truncatechars:20 }}
- |
- {{ login_log.ip }} |
- {{ login_log.city }} |
- {{ login_log.get_mfa_display }} |
- {{ login_log.get_reason_display }} |
- {{ login_log.get_status_display }} |
- {{ login_log.datetime }} |
-
- {% endfor %}
-{% endblock %}
{% block custom_foot_js %}
@@ -95,6 +114,29 @@
width: 'auto'
});
})
+ .on('click', '.btn_export', function () {
+ var date_form = $('#id_date_from').val();
+ var date_to = $('#id_date_to').val();
+ var user = $('.select2 option:selected').val();
+ var keyword = $('#search').val();
+ $.ajax({
+ url: "{% url "audits:login-log-export" %}",
+ method: 'POST',
+ data: JSON.stringify({
+ 'date_form':date_form,
+ 'date_to':date_to,
+ 'user':user,
+ 'keyword':keyword
+ }),
+ dataType: "json",
+ success: function (data, textStatus) {
+ window.open(data.redirect)
+ },
+ error: function () {
+ toastr.error('Export failed');
+ }
+ })
+ })
{% endblock %}
diff --git a/apps/audits/urls/view_urls.py b/apps/audits/urls/view_urls.py
index 473a2d83f..ef400cb99 100644
--- a/apps/audits/urls/view_urls.py
+++ b/apps/audits/urls/view_urls.py
@@ -14,4 +14,5 @@ urlpatterns = [
path('operate-log/', views.OperateLogListView.as_view(), name='operate-log-list'),
path('password-change-log/', views.PasswordChangeLogList.as_view(), name='password-change-log-list'),
path('command-execution-log/', views.CommandExecutionListView.as_view(), name='command-execution-log-list'),
+ path('login-log/export/', views.LoginLogExportView.as_view(), name='login-log-export'),
]
diff --git a/apps/audits/utils.py b/apps/audits/utils.py
new file mode 100644
index 000000000..36c54e81b
--- /dev/null
+++ b/apps/audits/utils.py
@@ -0,0 +1,22 @@
+import csv
+import codecs
+from django.http import HttpResponse
+
+
+def get_excel_response(filename):
+ excel_response = HttpResponse(content_type='text/csv')
+ excel_response[
+ 'Content-Disposition'] = 'attachment; filename="%s"' % filename
+ excel_response.write(codecs.BOM_UTF8)
+ return excel_response
+
+
+def write_content_to_excel(response, header=None, login_logs=None, fields=None):
+ writer = csv.writer(response, dialect='excel', quoting=csv.QUOTE_MINIMAL)
+ if header:
+ writer.writerow(header)
+ if login_logs:
+ for log in login_logs:
+ data = [getattr(log, field.name) for field in fields]
+ writer.writerow(data)
+ return response
\ No newline at end of file
diff --git a/apps/audits/views.py b/apps/audits/views.py
index a862ca066..3e2cc3473 100644
--- a/apps/audits/views.py
+++ b/apps/audits/views.py
@@ -1,8 +1,23 @@
+import csv
+import json
+import uuid
+import codecs
+
+
from django.conf import settings
+from django.urls import reverse
+from django.utils import timezone
+from django.core.cache import cache
+from django.http import HttpResponse, JsonResponse
+from django.utils.decorators import method_decorator
+from django.views import View
+from django.views.decorators.csrf import csrf_exempt
from django.views.generic import ListView
from django.utils.translation import ugettext as _
+from django.contrib.auth.mixins import LoginRequiredMixin
from django.db.models import Q
+from audits.utils import get_excel_response, write_content_to_excel
from common.mixins import DatetimeSearchMixin
from common.permissions import AdminUserRequiredMixin
@@ -232,3 +247,38 @@ class CommandExecutionListView(UserCommandExecutionListView):
}
kwargs.update(context)
return super().get_context_data(**kwargs)
+
+
+@method_decorator(csrf_exempt, name='dispatch')
+class LoginLogExportView(LoginRequiredMixin, View):
+
+ def get(self, request):
+ fields = [
+ field for field in UserLoginLog._meta.fields
+ ]
+ filename = 'login-logs-{}.csv'.format(
+ timezone.localtime(timezone.now()).strftime('%Y-%m-%d_%H-%M-%S')
+ )
+ excel_response = get_excel_response(filename)
+ header = [field.verbose_name for field in fields]
+ login_logs = cache.get(request.GET.get('spm', ''), [])
+
+ response = write_content_to_excel(excel_response, login_logs=login_logs,
+ header=header, fields=fields)
+ return response
+
+ def post(self, request):
+ try:
+ date_form = json.loads(request.body).get('date_form', [])
+ date_to = json.loads(request.body).get('date_to', [])
+ user = json.loads(request.body).get('user', [])
+ keyword = json.loads(request.body).get('keyword', [])
+
+ login_logs = UserLoginLog.get_login_logs(
+ date_form=date_form, date_to=date_to, user=user, keyword=keyword)
+ except ValueError:
+ return HttpResponse('Json object not valid', status=400)
+ spm = uuid.uuid4().hex
+ cache.set(spm, login_logs, 300)
+ url = reverse('audits:login-log-export') + '?spm=%s' % spm
+ return JsonResponse({'redirect': url})
\ No newline at end of file