3 š(ƒh;ã@sJddlmZddlmZGdd„deƒZeƒZdd„Zdd„Zd d „Z d S) é)ÚImproperlyConfigured)Úsettingsc@s<eZdZdZiZdd„Zdd„Zdd„Zdd „Zd d „Z d S) ÚLookupChannelRegistryzï Registry for LookupChannels activated for your django project. This includes any installed apps that contain lookup.py modules (django 1.7+) and any lookups that are explicitly declared in `settings.AJAX_LOOKUP_CHANNELS` c CsNi|_yddlm}Wntk r*Yn X|dƒttdƒrJ|jtjƒdS)z§ Called when loading the application. Cannot be called a second time, (eg. for testing) as Django will not re-import and re-register anything. r)Úautodiscover_modulesZlookupsÚAJAX_LOOKUP_CHANNELSN)Ú _registryZdjango.utils.module_loadingrÚ ImportErrorÚhasattrrÚregisterr)Úselfr©r úE/tmp/pip-install-q3hcpn_q/django-ajax-selects/ajax_select/registry.pyÚ load_channelss z#LookupChannelRegistry.load_channelscCs@x:|jƒD].\}}|dkr.||jkr8|j|=q ||j|<q WdS)akRegister a set of lookup definitions. Args: lookup_specs (dict): One or more LookupChannel specifications - `{'channel': LookupChannelSubclass}` - `{'channel': ('module.of.lookups', 'MyLookupClass')}` - `{'channel': {'model': 'MyModelToBeLookedUp', 'search_field': 'field_to_search'}}` N)Úitemsr)r Z lookup_specsÚchannelÚspecr r r r s   zLookupChannelRegistry.registerc Cs¼ddlm}y|j|}Wn$tk r>tdd|iƒ‚YnXt|ƒtkr\t||ƒr\|ƒSt|tƒrz|j |d|dƒSt|t ƒr¬t |diidgƒ}t ||dƒ}|ƒSt d |ƒ‚d S) aiFind the LookupChannel class for the named channel and instantiate it. Args: channel (string): - name that the lookup channel was registered at Returns: LookupChannel Raises: ImproperlyConfigured - if channel is not found. Exception - invalid lookup_spec was stored in registery r)Ú LookupChannelz=No ajax_select LookupChannel named %(channel)r is registered.rÚmodelÚ search_fieldÚézInvalid lookup spec: %sN)Ú ajax_selectrrÚKeyErrorrÚtypeÚ issubclassÚ isinstanceÚdictÚ make_channelÚtupleÚ __import__ÚgetattrÚ Exception)r rrZ lookup_specZ lookup_moduleÚ lookup_classr r r Úget/s   zLookupChannelRegistry.getcCs ||jkS)N)r)r rr r r Ú is_registeredWsz#LookupChannelRegistry.is_registeredcs8ddlm}|jdƒ\‰‰G‡‡‡fdd„d|ƒ}|ƒS)zúAutomatically make a LookupChannel. Args: app_model (str): app_name.ModelName arg_search_field (str): the field to search against and to display in search results Returns: LookupChannel r)rÚ.cseZdZe””ƒZ”ZdS)z=LookupChannelRegistry.make_channel..MadeLookupChannelN)Ú__name__Ú __module__Ú __qualname__Ú get_modelrrr )Ú app_labelÚarg_search_fieldÚ model_namer r ÚMadeLookupChannelfs r-)rrÚsplit)r Z app_modelr+rr-r )r*r+r,r rZs z"LookupChannelRegistry.make_channelN) r&r'r(Ú__doc__rrr r#r$rr r r r rs(rc CsJyddlm}Wn(tk r8ddlm}|j||ƒSX|j||ƒSdS)z0Loads the model given an 'app_label' 'ModelName'r)Úapps)ÚmodelsN)Ú django.appsr0rZ django.dbr1r))r*r,r0r1r r r r)qs  r)c Cs*yddlm}Wntk r$dSXdS)Nr)Ú AppConfigFT)r2r3r)r3r r r Úcan_autodiscover~s r4cs‡fdd„}|S)a"Decorator to register a LookupClass. Example:: from ajax_select import LookupChannel, register @register('agent') class AgentLookup(LookupClass): def get_query(self): ... def format_item(self): ... csˆs tdƒ‚tjˆ|iƒ|S)Nz'Lookup Channel must have a channel name)Ú ValueErrorÚregistryr )r")rr r Ú_wrapper–szregister.._wrapperr )rr7r )rr r †s r N) Zdjango.core.exceptionsrZ django.confrÚobjectrr6r)r4r r r r r Ús  i