""" WARNING: This file about to undergo major refactoring by @pydanny per Issue #99. For wont of a better name, this module is called 'models'. It's role is synonymous with the django.contrib.admin.sites model. """ from django.core.urlresolvers import reverse from django.conf.urls import patterns, url from django.contrib.auth import models as auth_app from django.db.models import get_models, signals from djadmin2 import apiviews from djadmin2 import views MODEL_ADMIN_ATTRS = ( 'list_display', 'list_display_links', 'list_filter', 'admin', 'has_permission', 'has_add_permission', 'has_edit_permission', 'has_delete_permission', ) class BaseAdmin2(object): """ Warning: This class will likely merged with ModelAdmin2 """ search_fields = [] # Show the fields to be displayed as columns # TODO: Confirm that this is what the Django admin uses list_fields = [] #This shows up on the DocumentListView of the Posts list_actions = [] # This shows up in the DocumentDetailView of the Posts. document_actions = [] # shows up on a particular field field_actions = {} fields = None exclude = None fieldsets = None form_class = None filter_vertical = () filter_horizontal = () radio_fields = {} prepopulated_fields = {} formfield_overrides = {} readonly_fields = () ordering = None def __init__(self, model, admin): super(BaseAdmin2, self).__init__() self.model = model self.admin = admin def _user_has_permission(self, user, permission_type, obj=None): """ Generic method for checking whether the user has permission of specified type for the model. Type can be one of view, add, change, delete. You can also specify instance of the model for object-specific permission check. """ if not user.is_authenticated() or not user.is_staff: return False opts = self.model._meta full_permission_name = '%s.%s_%s' % (opts.app_label, permission_type, opts.object_name.lower()) return user.has_perm(full_permission_name, obj) def has_permission(self, request, permission_type, obj=None): return self._user_has_permission(request.user, permission_type, obj) def has_view_permission(self, request, obj=None): """ Can view this object """ return self.has_permission(request, 'view', obj) def has_edit_permission(self, request, obj=None): """ Can edit this object """ return self.has_permission(request, 'change', obj) def has_add_permission(self, request, obj=None): """ Can add this object """ return self.has_permission(request, 'add', obj) def has_delete_permission(self, request, obj=None): """ Can delete this object """ return self.has_permission(request, 'delete', obj) class ModelAdmin2(BaseAdmin2): """ Warning: This class is targeted for reduction. It's bloated and ugly. """ list_display = ('__str__',) list_display_links = () list_filter = () list_select_related = False list_per_page = 100 list_max_show_all = 200 list_editable = () search_fields = () save_as = False save_on_top = False verbose_name = None verbose_name_plural = None model_admin_attributes = MODEL_ADMIN_ATTRS create_form_class = None update_form_class = None # Views index_view = views.ModelListView create_view = views.ModelAddFormView update_view = views.ModelEditFormView detail_view = views.ModelDetailView delete_view = views.ModelDeleteView # API configuration api_serializer_class = None # API Views api_list_view = apiviews.ListCreateAPIView api_detail_view = apiviews.RetrieveUpdateDestroyAPIView def __init__(self, model, admin, **kwargs): self.model = model self.admin = admin self.app_label = model._meta.app_label self.model_name = model._meta.object_name.lower() if self.verbose_name is None: self.verbose_name = self.model._meta.verbose_name if self.verbose_name_plural is None: self.verbose_name_plural = self.model._meta.verbose_name_plural def get_default_view_kwargs(self): return { 'app_label': self.app_label, 'model': self.model, 'model_name': self.model_name, 'model_admin': ImmutableAdmin(self), } def get_default_api_view_kwargs(self): kwargs = self.get_default_view_kwargs() kwargs.update({ 'serializer_class': self.api_serializer_class, }) return kwargs def get_prefixed_view_name(self, view_name): return '{}_{}_{}'.format(self.app_label, self.model_name, view_name) def get_index_kwargs(self): return self.get_default_view_kwargs() def get_create_kwargs(self): kwargs = self.get_default_view_kwargs() kwargs.update({ 'form_class': self.create_form_class if self.create_form_class else self.form_class, }) return kwargs def get_update_kwargs(self): kwargs = self.get_default_view_kwargs() kwargs.update({ 'form_class': self.update_form_class if self.update_form_class else self.form_class, }) return kwargs def get_detail_kwargs(self): return self.get_default_view_kwargs() def get_delete_kwargs(self): return self.get_default_view_kwargs() def get_index_url(self): return reverse('admin2:{}'.format(self.get_prefixed_view_name('index'))) def get_api_list_kwargs(self): kwargs = self.get_default_api_view_kwargs() kwargs.update({ 'paginate_by': self.list_per_page, }) return kwargs def get_api_detail_kwargs(self): return self.get_default_api_view_kwargs() def get_urls(self): return patterns('', url( regex=r'^$', view=self.index_view.as_view(**self.get_index_kwargs()), name=self.get_prefixed_view_name('index') ), url( regex=r'^create/$', view=self.create_view.as_view(**self.get_create_kwargs()), name=self.get_prefixed_view_name('create') ), url( regex=r'^(?P[0-9]+)/$', view=self.detail_view.as_view(**self.get_detail_kwargs()), name=self.get_prefixed_view_name('detail') ), url( regex=r'^(?P[0-9]+)/update/$', view=self.update_view.as_view(**self.get_update_kwargs()), name=self.get_prefixed_view_name('update') ), url( regex=r'^(?P[0-9]+)/delete/$', view=self.delete_view.as_view(**self.get_delete_kwargs()), name=self.get_prefixed_view_name('delete') ), ) def get_api_urls(self): return patterns('', url( regex=r'^$', view=self.api_list_view.as_view(**self.get_api_list_kwargs()), name=self.get_prefixed_view_name('api-list'), ), url( regex=r'^(?P[0-9]+)/$', view=self.api_detail_view.as_view(**self.get_api_detail_kwargs()), name=self.get_prefixed_view_name('api-detail'), ), ) @property def urls(self): # We set the application and instance namespace here return self.get_urls(), None, None @property def api_urls(self): return self.get_api_urls(), None, None class ImmutableAdmin(object): """ Only __init__ allows setting of attributes """ def __init__(self, model_admin): """ The __init__ is the only method where the ImmutableModelAdmin allows for setting of values. """ for attr_name in model_admin.model_admin_attributes: setattr(self, attr_name, getattr(model_admin, attr_name)) self.__delattr__ = self._immutable self.__setattr__ = self._immutable def _immutable(self, name, value): raise TypeError("Can't modify immutable model admin") def create_extra_permissions(app, created_models, verbosity, **kwargs): """ Creates 'view' permissions for all models. django.contrib.auth only creates add, change and delete permissions. Since we also support read-only views, we need to add our own extra permission. Copied from django.contrib.auth.management.create_permissions """ from django.contrib.contenttypes.models import ContentType app_models = get_models(app) # This will hold the permissions we're looking for as # (content_type, (codename, name)) searched_perms = list() # The codenames and ctypes that should exist. ctypes = set() for klass in app_models: ctype = ContentType.objects.get_for_model(klass) ctypes.add(ctype) opts = klass._meta perm = ('view_%s' % opts.object_name.lower(), u'Can view %s' % opts.verbose_name_raw) searched_perms.append((ctype, perm)) # Find all the Permissions that have a content_type for a model we're # looking for. We don't need to check for codenames since we already have # a list of the ones we're going to create. all_perms = set(auth_app.Permission.objects.filter( content_type__in=ctypes, ).values_list( "content_type", "codename" )) perms = [ auth_app.Permission(codename=codename, name=name, content_type=ctype) for ctype, (codename, name) in searched_perms if (ctype.pk, codename) not in all_perms ] auth_app.Permission.objects.bulk_create(perms) if verbosity >= 2: for perm in perms: print "Adding permission '%s'" % perm signals.post_syncdb.connect(create_extra_permissions, dispatch_uid="django-admin2.djadmin2.models.create_extra_permissions")