ó ßüÚ\c@s®ddlZddlZddlZddlZddlZddlZddlmZddlZddl m Z ej e ƒZ d„Zdd„Zdddd„ZdS(iÿÿÿÿN(t import_modulei(t import_filec Cs©| r!|jjdd ƒ r!|S|rÂ|jjdd ƒ rÂg|D]}|jdƒrD|^qD}x|D]}tj|ƒ‚qlWtjdt|ƒdkr¦dnddj|ƒfƒ‚nt |jjdƒƒ}g|j ƒD]&}t |dtj ƒrç|d^qç}t|ƒdkrGtjd t |jƒƒƒ‚n|rf| rftjd ƒ‚n|sp|S|d } tj| d tƒ}| j|t |ƒƒ|S(sKClick option callback providing validation of preload subcommand arguments.tpreloadt-s%Got unexpected extra argument%s: (%s)itstt t dask_setups<Multiple --preload modules with click-configurable setup: %ssPUnknown argument specified: %r Was click-configurable --preload target provided?itallow_extra_argsN(tparamstgettNonet startswithtclickt NoSuchOptiont UsageErrortlentjoint_import_modulestvaluest isinstancetCommandtlisttkeystContexttFalset parse_args( tctxtparamtvaluetvtunexpected_argstatpreload_modulestmtpreload_commandstpreload_command((s5lib/python2.7/site-packages/distributed/preloading.pytvalidate_preload_argvs4( 4&   csi}x|D]û}|jdƒrÅ|dk r²tjj|ƒ}tjj||ƒ}tjj|ƒrtj||ƒst j d|ƒqnt j ||ƒt |ƒd‰qît |ƒd‰n)|tjkrát|ƒntj|‰‡fd†dDƒ||hs Rt dask_teardownN(RR*(tendswithR tostpathtbasenameRtexiststfilecmptcmptloggerterrortshutiltcopyRtsystmodulesR(tnamestfile_dirtresult_modulestnameR.tcopy_dst((R)s5lib/python2.7/site-packages/distributed/preloading.pyR=s$     c Cs×t|d|ƒ}x¾|jƒD]°\}}|jddƒ}|jddƒ}|r®t|tjƒr¡|jdt|ƒdt ƒ} |j || j | j Žq®||ƒn|drt j|d|ƒqqWdS(s Imports modules, handles `dask_setup` and `dask_teardown`. Parameters ---------- names: list of strings Module names or file paths parameter: object Parameter passed to `dask_setup` and `dask_teardown` argv: [string] List of string arguments passed to click-configurable `dask_setup`. file_dir: string Path of a directory where files should be copied R9RR*RN(RtitemsR R RR Rt make_contextRRtcallbacktargsR tatexittregister( R8t parameterR9targvtimported_modulesR;t interfaceRR*tcontext((s5lib/python2.7/site-packages/distributed/preloading.pyR!os  (RAtloggingR,R4R6R0t importlibRR tutilsRt getLoggert__name__R2R%R RR!(((s5lib/python2.7/site-packages/distributed/preloading.pyts        - 2