+d%ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ej dZ GddZ e Z dZdd Zd Zd ZGd d eZdS)NconfintakeceZdZdZddZedZejdZdZdZ fdZ fd Z ed Z d Z d Zd ZddZdZddZdZxZS) DriverSouceszJ Handles the various ways in which drivers can be known to Intake Nch|pt|_||_d|_d|_d|_d|_dS)a Parameters ---------- config: intake.config.Config oinstance, optional If not given, will use ``intake.config.conf`` singleton do_scan: bool or None Whether to scan packages with names ``intake_*`` for valid drivers. This is for backward compatibility only. If not given, value comes from the config key "package_scan", default False. N)rdo_scan_scanned _entrypoints _registered _disabled)selfconfigr s 7lib/python3.11/site-packages/intake/source/discovery.py__init__zDriverSouces.__init__s8Nd    cT|j|jddn|jS)N package_scanF)r rgetrs rrzDriverSouces.package_scan/s(7;|]ty}}^U333QUQ]]rc||jd<dS)Nrr)rvals rrzDriverSouces.package_scan3s$' .!!!rcj|j&ttjd|_|jS)Nzintake.drivers)r list entrypoints get_group_allrs rfrom_entrypointszDriverSouces.from_entrypoints7s3   R $[%>?O%P%P Q QD   rcpd|jdiDS)Nc .g|]\}}|tj|d|vr|dddn|dddd|vr|dddn|dddS):r.)name module_name object_name)r EntryPointrsplit.0kvs r z*DriverSouces.from_conf..>s   1   "3!8$\AHHS!$4$4Q$7$7RUWXIYIYZ[I\BEIJBJkcjkjrjrsvxyjzjz{|j}j}PQPXPXY\^_P`P`abPc      rdrivers)rritemsrs r from_confzDriverSouces.from_conf=sA   i44::<<     rctt||||dSN)superr __setitem__save)rkeyvalue __class__s rr4zDriverSouces.__setitem__Fs4 lD!!--c5999 rctt|||dSr2)r3r __delitem__r5)rr6r8s rr:zDriverSouces.__delitem__Js2 lD!!--c222 rcD|jt|_|jSr2)r _package_scanrs rscannedzDriverSouces.scannedNs! = ,)OODM}rc|jz(DriverSouces.disabled..Xs(___DAqTUY^T^_a___rr.)rrrr/rs rdisabledzDriverSouces.disabledVsF > `__DIMM)R,H,H,N,N,P,P___DN~rc|j`i}|jr||j||zD] }|||j< ||_|jSr2)r rupdater=rr0r$)routeps r registeredzDriverSouces.registered\sy   #C  ) 4<(((++--0@0@@ " "!BG "D rchfdDS)NcHi|]\}}|v||Sr@)rB)r*r+r,rs r z0DriverSouces.enabled_plugins..ks1WWWAat}}>VW1WWWr)rGr/rs`renabled_pluginszDriverSouces.enabled_pluginsjs3WWWW!2!2!8!8!:!:WWWWrFcBt|}||vr|std|d||vr7|r|||nt d|d|||<dS)aAdd runtime driver definition to list of registered drivers (drivers in global scope with corresponding ``intake.open_*`` function) Parameters ---------- name: str Name of the driver value: str, entrypoint or class Pointer to the implementation clobber: bool If True, perform the operation even if the driver exists do_enable: bool If True, unset the disabled flag for this driver zDriver z already enabledzAdding driver z, but it is disabledN) _normalizerG ValueErrorrBenableloggerwarning)rr$r7clobber do_enables rregister_driverzDriverSouces.register_driverms$ 4??$$ $ ?W ?=t===>> > 4==?? " L L D%((((JJJJKKK"'$rcrt|}||dS)z Remove runtime registered driverN)rMrGpop)rr$s runregister_driverzDriverSouces.unregister_drivers3$ d#####rc0|j}d|vri|d<|r ||d|<n&|d|dur |d|=||vr'|||dS)a Explicitly assign a driver to a name, or remove ban Updates the associated config, which will be persisted Parameters ---------- name : string As in ``'zarr'`` driver : string Dotted object name, as in ``'intake_xarray.xzarr.ZarrSource'``. If None, simply remove driver disable flag, if it is found r.FN)rrrBremover5)rr$driverrs rrOzDriverSouces.enables F " # "F9   (&,F9 d # # I  " "4 ( (E 1 (y!$' 4==?? " ) MMOO " "4 ( ( ( rct|}|j}d|vri|d<d|d|<|||dS)zDisable a driver by name. Updates the associated config, which will be persisted Parameters ---------- name : string As in ``'zarr'`` r.FN)rMrr5rBadd)rr$rs rdisablezDriverSouces.disablesm$ F " # "F9 "'y$  D!!!!!r)NN)FFr2)__name__ __module__ __qualname____doc__rpropertyrsetterrr0r4r:r=rBrGrKrTrWrOr] __classcell__)r8s@rrrs[&^^X^(((!!!    X    XXX((((2$$$ 2"""""""rrc  |S#t$r&}td|jd|jd|d}~wt $r.}td|jd|jd|jd|d}~wwxYw)zQ Call entrypoint.load() and, if it fails, raise context-specific errors. zFailed to load z driver because module z could not be imported.Nz driver because no object named z could be found in the module r#)load ImportErrorConfigurationErrorr$r%AttributeErrorr&) entrypointerrs r_load_entrypointrls    YYY "O:?"O"O_i_u"O"O"OPPVY Y   )jo ) )+ ) )% ) ) )    s  A<!A A<)A77A<intake_c ,tjdti}tj|D]\}}}||rt j}t|}|D][\}} ||vrMtj ||} tj | } tjd|d| d| dV| ||<\t d|t j|z fz|S) arScan for intake drivers and return a dict of plugins. This searches path (or sys.path) for packages with names that start with plugin_prefix. Those modules will be imported and scanned for subclasses of intake.source.base.Plugin. Any subclasses found will be instantiated and returned in a dictionary, with the plugin's name attribute as the key. zPackage scanning may be removed)category)pathzPlugin name collision for "z " from z and z$ Keeping plugin from first location.zImport %s took: %7.2f s) warningswarnPendingDeprecationWarningpkgutil iter_modules startswithtimeload_plugins_from_moduler/inspectgetfilerPdebug) rp plugin_prefixpluginsimporterr$ispkgt new_plugins plugin_nameplugin orig_pathnew_paths rr<r<s= M3>WXXXXG!(!54!@!@!@NN$ ??= ) ) N A2488K'2'8'8':': 2 2# V') 2 ' 0D E EI&v66HMM DO;;PYPYPY[c[c[c e,2GK(( LL2dDIKK!O5LL M M M Nrc|stjdd|}|stjd|d|S)Nz [-=~^&|@+]_z&Invalid Intake plugin name "%s" found.r) stacklevel) isidentifierresubrqrr)r$s rrMrMs\     /vlC..     T >QRSSSS Krcddlm}ddlm}i} t j|}n8#t $r+}|drt d|d}~wwxYwnG#t$r:}t d ||icYd}~Sd}~wwxYwtj |tjD]!\}}t|||fr |||j<"|S)zImports a module and returns dictionary of discovered Intake plugins. Plugin classes are instantiated and added to the dictionary, keyed by the name attribute of the plugin object. r)Catalog) DataSourcez.pyzIntake formerly supported executing arbitrary Python files not on the sys.path. This is no longer supported. Drivers must be specific with a module path like 'package_name.module_name, not a Python filename like 'module.py'.NzImport module <{}> failed: {})intake.catalogr intake.sourcer importlib import_modulergendswith ExceptionrPr{formatry getmembersisclass issubclassr$) r%rrr}moderrorerclss rrxrxsF '&&&&&((((((G )+66CC   ##E** "#      4;;KKKLLL $S'/::$$3 cGZ0 1 1 $ #GCH  Ns7&A A&AAA B#)/BB#B#ceZdZdS)rhN)r^r_r`r@rrrhrhsDrrh)Nrm)rryloggingrtrrwrqrrr getLoggerrPrr.rlr<rMrxrrhr@rrrs.   8 $ $]"]"]"]"]"]"]"]"@ ,..    D###L          r