Enable custom actions.

flask_peewee/admin.py

@@ -169,12 +169,19 @@ def get_query(self):
     def get_object(self, pk):
         return self.get_query().where(self.pk==pk).get()
 
+    def get_actions(self):
+        return (
+            ('delete', {'url': '/delete/', 'title': 'Delete'}),
+            ('export', {'url': '/export/', 'title': 'Export'}),
+        )
+
     def get_urls(self):
+        actions_urls = [(v['url'], getattr(self, k))
+                        for k, v in self.get_actions()]
         return (
             ('/', self.index),
             ('/add/', self.add),
-            ('/delete/', self.delete),
-            ('/export/', self.export),
+        ) + tuple(actions_urls) + (
             ('/<pk>/', self.edit),
             ('/_ajax/', self.ajax_list),
         )
@@ -219,12 +226,12 @@ def index(self):
         # create a paginated query out of our filtered results
         pq = PaginatedQuery(query, self.paginate_by)
 
+        # actions
         if request.method == 'POST':
-            id_list = request.form.getlist('id')
-            if request.form['action'] == 'delete':
-                return redirect(url_for(self.get_url_name('delete'), id=id_list))
-            else:
-                return redirect(url_for(self.get_url_name('export'), id=id_list))
+            action = request.form['action']
+            if action in [k for k, v in self.get_actions()]:
+                id_list = request.form.getlist('id')
+                return redirect(url_for(self.get_url_name(action), id=id_list))
 
         return render_template(self.templates['index'],
             model_admin=self,

flask_peewee/templates/admin/models/index.html

@@ -3,13 +3,17 @@
 {% block tab_index_class %}active{% endblock %}
 
 {% block extra_tabs %}
+    {% set model_admin_actions = model_admin.get_actions() %}
+    {% if model_admin_actions %}
     <li class="dropdown" style="float:right">
       <a class="dropdown-toggle" data-toggle="dropdown" href="#">With selected... <span class="caret" /></a>
       <ul class="dropdown-menu pull-right">
-        <li><a href="#" onclick="Admin.index_submit('export');">Export</a></li>
-        <li><a href="#" onclick="Admin.index_submit('delete');">Delete</a></li>
+        {% for action, info in model_admin_actions %}
+        <li><a href="#" onclick="Admin.index_submit('{{ action }}');">{{ info.title }}</a></li>
+        {% endfor %}
       </ul>
     </li>
+    {% endif %}
     {% include "admin/includes/filter_dropdown.html" %}
 {% endblock %}