Ref #365 Added Clear DB page

silk/templates/silk/clear_db.html

@@ -0,0 +1,96 @@
+{% extends 'silk/base/root_base.html' %}
+{% load static %}
+{% load silk_inclusion %}
+
+{% block menu %}
+    {% if silk_request %}
+        {% request_menu request silk_request %}
+    {% else %}
+        {% root_menu request %}
+    {% endif %}
+{% endblock %}
+
+{% block style %}
+    {{ block.super }}
+    <style>
+        .container {
+            padding: 0 1em;
+        }
+
+        h2 {
+            margin-bottom: 10px;
+        }
+
+        .cleardb-form .cleardb-form-wrapper{
+            margin-bottom: 20px;
+        }
+
+        .cleardb-form label {
+            display: block;
+            margin-bottom: 8px;
+        }
+
+        .cleardb-form .btn {
+            background: #333344;
+            color: #fff;
+            padding: 10px 20px;
+            border-radius: 2px;
+            cursor: pointer;
+            box-shadow: none;
+            font-size: 16px;
+            line-height: 20px;
+            border: 0;
+            min-width: 150px;
+            text-align: center;
+        }
+        .cleardb-form label :last-child {
+            margin-bottom: 0;
+        }
+        .msg {
+            margin-top: 20px;
+            color: #bac54b;
+        }
+    </style>
+
+{% endblock %}
+
+{% block js %}
+    {{ block.super }}
+    <script>
+
+        $(document).ready(function () {
+            initFilters();
+            initFilterButton();
+        });
+    </script>
+{% endblock %}
+
+
+{% block data %}
+    <div class="container">
+        <h2>Silk Clear DB</h2>
+        <form class="cleardb-form" action="." method="post">
+            {% csrf_token %}
+            <div class="cleardb-form-wrapper">
+                <label>
+                    <input type="checkbox" name="clear_requests" disabled="disabled"/>
+                    Requests
+                </label>
+                <label>
+                    <input type="checkbox" name="clear_profiling" disabled="disabled"/>
+                    Profiling
+                </label>
+                <label>
+                    <input type="checkbox" name="clear_all"/>
+                    All
+                </label>
+            </div>
+            <button class="btn">Clear</button>
+        </form>
+        <div class="msg">{{ msg }}</div>
+    </div>
+{% endblock %}
+
+
+{% block filter %}{% endblock %}
+{% block filters %}{% endblock %}

silk/templates/silk/inclusion/root_menu.html

@@ -20,4 +20,11 @@
         </div>
     </a>
 </div>
+<div class="menu-item selectable-menu-item {% navactive request 'silk:cleardb' %}">
+    <a href="{% url "silk:cleardb" %}">
+        <div class="menu-item-outer">
+            <div class="menu-item-inner">Clear DB</div>
+        </div>
+    </a>
+</div>
 

silk/urls.py

@@ -1,5 +1,6 @@
 from django.conf.urls import url
 
+from silk.views.clear_db import ClearDBView
 from silk.views.profile_detail import ProfilingDetailView
 from silk.views.profile_download import ProfileDownloadView
 from silk.views.profile_dot import ProfileDotView
@@ -82,6 +83,7 @@
         name='profile_sql_detail'
     ),
     url(r'^profiling/$', ProfilingView.as_view(), name='profiling'),
+    url(r'^cleardb/$', ClearDBView.as_view(), name='cleardb'),
     url(
         r'^request/(?P<request_id>[a-zA-Z0-9\-]+)/cprofile/$',
         CProfileView.as_view(),

silk/views/clear_db.py

@@ -0,0 +1,33 @@
+from django.db import connection
+from django.shortcuts import render
+from django.utils.decorators import method_decorator
+from django.views.generic import View
+
+from silk.auth import login_possibly_required, permissions_possibly_required
+from silk.models import Request, Response, SQLQuery, Profile
+
+
+class ClearDBView(View):
+
+    def _truncate_tables(self, models):
+        raw_query = 'TRUNCATE TABLE {0};'
+        truncate_query = [raw_query.format(m._meta.db_table) for m in models]
+        truncate_query = ' '.join(truncate_query)
+        query = 'SET FOREIGN_KEY_CHECKS = 0; {0} SET FOREIGN_KEY_CHECKS = 1;'.format(truncate_query)
+        cursor = connection.cursor()
+        cursor.execute(query)
+
+    @method_decorator(login_possibly_required)
+    @method_decorator(permissions_possibly_required)
+    def get(self, request, *_, **kwargs):
+        return render(request, 'silk/clear_db.html')
+
+    @method_decorator(login_possibly_required)
+    @method_decorator(permissions_possibly_required)
+    def post(self, request, *_, **kwargs):
+        context = {}
+        if 'clear_all' in request.POST:
+            self._truncate_tables([Response, SQLQuery, Profile, Request])
+            tables = ['Response', 'SQLQuery', 'Profile', 'Request']
+            context['msg'] = 'Cleared data for following silk tables: {0}'.format(', '.join(tables))
+        return render(request, 'silk/clear_db.html', context=context)