3 (h )@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZdd lmZdd lmZdd lmZGd ddeZGdddeZGdddeZGdddeeZGdddeeZddZdS))unicode_literals)wraps)chain)url)messages)reverse)QuerySet)Http404HttpResponseRedirect)HttpResponseBase)View)SingleObjectMixin)MultipleObjectMixincspeZdZdZgZgZdZfddZdfdd Zdfdd Z d d Z d d Z ddZ ddZ ddZZS)BaseDjangoObjectActionsa ModelAdmin mixin to add new actions just like adding admin actions. Attributes ---------- model : django.db.models.Model The Django Model these actions work on. This is populated by Django. change_actions : list of str Write the names of the methods of the model admin that can be used as tools in the change view. changelist_actions : list of str Write the names of the methods of the model admin that can be used as tools in the changelist view. tools_view_name : str The name of the Django Object Actions admin view, including the 'admin' namespace. Populated by `_get_action_urls`. Ncstt|j}|j|S)z)Prepend `get_urls` with our own patterns.)superrget_urls_get_action_urls)selfurls) __class__N/tmp/pip-install-q3hcpn_q/django-object-actions/django_object_actions/utils.pyr*sz BaseDjangoObjectActions.get_urlscsH|pi}|jfddj|||Djdttj||||S)Ncsg|]}j|qSr)_get_tool_dict).0action)rrr 3sz7BaseDjangoObjectActions.change_view..) objectactionstools_view_name)updateget_change_actionsrrr change_view)rrequest object_idform_url extra_context)r)rrr!/s   z#BaseDjangoObjectActions.change_viewcs@|pi}|jfddj|Djdttj||S)Ncsg|]}j|qSr)r)rr)rrrr?sz;BaseDjangoObjectActions.changelist_view..)rr)rget_changelist_actionsrrrchangelist_view)rr"r%)r)rrr';s    z'BaseDjangoObjectActions.changelist_viewcCs|jS)a> Override this to customize what actions get to the change view. This takes the same parameters as `change_view`. For example, to restrict actions to superusers, you could do: class ChoiceAdmin(DjangoObjectActions, admin.ModelAdmin): def get_change_actions(self, request, **kwargs): if request.user.is_superuser: return super(ChoiceAdmin, self).get_change_actions( request, **kwargs ) return [] )change_actions)rr"r#r$rrrr Jsz*BaseDjangoObjectActions.get_change_actionscCs|jS)zU Override this to customize what actions get to the changelist view. )changelist_actions)rr"rrrr&\sz.BaseDjangoObjectActions.get_changelist_actionsc Csi}|jjj}d|jjj|f}d|}d||_x$t|j|jD]}t||||<qBWt d|j j t j |j|d||j jd|dt d|j j tj |j|d ||j jd|dgS) z6Get the url patterns that route each action to a view.z%s_%sz %s_actionszadmin:z#^(?P.+)/actions/(?P\w+)/$zadmin:%s_change)modelactionsback current_app)namez^actions/(?P\w+)/$zadmin:%s_changelist)r*_meta model_nameZ app_labelrrr(r)getattrrZ admin_siteZ admin_viewChangeActionViewZas_viewr.ChangeListActionView)rr+r0Z base_url_nameZmodel_actions_url_namerrrrres.    z(BaseDjangoObjectActions._get_action_urlscCs0t||}|j|\}}t|t|d|||dS)z.Represents the tool as a dict with extra meta.label)r.r4standard_attrs custom_attrs)r1_get_button_attrsdict)rZ tool_nametoolr5r6rrrrs  z&BaseDjangoObjectActions._get_tool_dictcCst|di}d|kr|jdd|kr0|jd|jddt|ddd}i}i}x6t|f|jD]"\}}||kr~|||<qd|||<qdW||fS)a= Get the HTML attributes associated with a tool. There are some standard attributes (class and title) that the template will always want. Any number of additional attributes can be specified and passed on. This is kinda awkward and due for a refactor for readability. attrshreftitleclassrZshort_description)r=r<)r1popgetr8items)rr9r:Z default_attrsr5r6kvrrrr7s      z)BaseDjangoObjectActions._get_button_attrs)rN)N)__name__ __module__ __qualname____doc__r(r)rrr!r'r r&rrr7 __classcell__rr)rrrs   + rc@seZdZdZdZdS)DjangoObjectActionsz&django_object_actions/change_form.htmlz&django_object_actions/change_list.htmlN)rCrDrEZchange_form_templateZchange_list_templaterrrrrHsrHc@sLeZdZdZdZdZdZdZeddZ eddZ ddZ e Z d d Z dS) BaseActionViewa The view that runs a change/changelist action callable. Attributes ---------- back : str The urlpattern name to send users back to. This is set in `_get_action_urls` and turned into a url with the `back_url` property. model : django.db.model.Model The model this tool operates on. actions : dict A mapping of action names to callables. NcCstdS)z tuple: The argument(s) to send to the action (excluding `request`). Change actions are called with `(request, obj)` while changelist actions are called with `(request, queryset)`. N)NotImplementedError)rrrr view_argsszBaseActionView.view_argscCstdS)z str: The url path the action should send the user back to. If an action does not return a http response, we automagically send users back to either the change or the changelist page. N)rJ)rrrrback_urlszBaseActionView.back_urlc KsTy|j|}Wntk r*tdYnX||f|j}t|trJ|St|jS)NzAction does not exist)r+KeyErrorr rK isinstancer r rL)rr"r9kwargsviewretrrrr?s zBaseActionView.getcCstj||dS)z Mimic Django admin actions's `message_user`. Like the second example: https://docs.djangoproject.com/en/1.9/ref/contrib/admin/actions/#custom-admin-action N)rinfo)rr"messagerrr message_userszBaseActionView.message_user)rCrDrErFr,r*r+r-propertyrKrLr?postrTrrrrrIs   rIc@s$eZdZeddZeddZdS)r2cCs |jfS)N)Z get_object)rrrrrKszChangeActionView.view_argscCst|j|jdf|jdS)Npk)argsr-)rr,rOr-)rrrrrLszChangeActionView.back_urlN)rCrDrErUrKrLrrrrr2s r2c@s$eZdZeddZeddZdS)r3cCs |jfS)N) get_queryset)rrrrrK szChangeListActionView.view_argscCst|j|jdS)N)r-)rr,r-)rrrrrLszChangeListActionView.back_urlN)rCrDrErUrKrLrrrrr3 s r3cstfdd}|S)z>Decorator that makes standard Django admin actions compatible.cs~t|tsry|j|j|jd}WnNtk rpy |jj}Wntk rZ|jj}YnX|j j|jd}YnX|||S)N)rW) rNrrYfilterrWAttributeErrorr/r*Zconcrete_modelZobjects)rr"Zquerysetr*)funcrrdecorated_functions  z6takes_instance_or_queryset..decorated_function)r)r\r]r)r\rtakes_instance_or_querysetsr^N) __future__r functoolsr itertoolsrZdjango.conf.urlsrZdjango.contribrZdjango.core.urlresolversrZdjango.db.models.queryrZ django.httpr r Zdjango.http.responser Zdjango.views.genericr Zdjango.views.generic.detailr Zdjango.views.generic.listrobjectrrHrIr2r3r^rrrrs$           +@