Merge pull request #61 from tonimichel/master

Added support for custom CreateViews and UpdateViews

Author: asifpy

crudbuilder/abstract.py

@@ -1,4 +1,5 @@
 import django
+
 try:
  from django.apps import apps
 except ImportError:
@@ -14,10 +15,10 @@
 
 class BaseBuilder(object):
  def __init__(
- self,
- app,
- model,
- crud
+  self,
+  app,
+  model,
+  crud
  ):
 
  self.model = model
@@ -29,6 +30,9 @@ def __init__(
  self.detailview_excludes = self._has_crud_attr('detailview_excludes')
  self.createupdate_forms = self._has_crud_attr('createupdate_forms')
 
+ self.custom_update_view_mixin = self._has_crud_attr('custom_update_view_mixin')
+ self.custom_create_view_mixin = self._has_crud_attr('custom_create_view_mixin')
+
  # django tables2
  self.custom_table2 = self._has_crud_attr('custom_table2')
  self.tables2_fields = self._has_crud_attr('tables2_fields')

crudbuilder/views.py

@@ -127,9 +127,13 @@ def generate_create_view(self):
  custom_postfix_url=self.custom_postfix_url
  )
 
+ parent_classes = [self.get_createupdate_mixin(), CreateView]
+ if self.custom_create_view_mixin:
+ parent_classes.insert(0, self.custom_create_view_mixin)
+
  create_class = type(
  name,
- (self.get_createupdate_mixin(), CreateView),
+ tuple(parent_classes),
  create_args
  )
 
@@ -172,9 +176,13 @@ def generate_update_view(self):
  custom_postfix_url=self.custom_postfix_url
  )
 
+ parent_classes = [self.get_createupdate_mixin(), UpdateView]
+ if self.custom_update_view_mixin:
+ parent_classes.insert(0, self.custom_update_view_mixin)
+
  update_class = type(
  name,
- (self.get_createupdate_mixin(), UpdateView),
+ tuple(parent_classes),
  update_args
  )
  self.classes[name] = update_class