Wf dZddlZddlZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl m Z ejdZe ej dej"j%edd Zd gd d d d gggd ZdZGddeZdZeZej5ej6Zej:Zy)z+Intake config manipulations and persistenceN) expanduser)make_path_posix) yaml_loadintakeINTAKE_CONF_DIR~z.intakeINFOTF) logging catalog_path allow_importallow_templates allow_pickleimport_on_startup extra_importsimport_block_list reader_avoidc pttjdtjt dS)NINTAKE_CONF_FILEz conf.yaml)rosgetenv posixpathjoinconfdir-lib/python3.12/site-packages/intake/config.pycfiler)s% 299%7Q\9]^ __rceZdZdZd fd ZdZd dZejdZ d dZ fdZ d fd Z d Z d d Zd ZxZS) ConfigzkIntake's dict-like config system Instance ``intake.conf`` is globally used throughout the package c ||n t|_|jd|_t |di|t j|dy)Nr r)rfilename reload_alltempsuper__init__loggersetLevel)selfr!kwargs __class__s rr%zConfig.__init__3sD$,$8eg    "6"Y(rcN|j|jty)z Set conf values back to defaultsN)clearupdatedefaultsr(s rresetz Config.reset:s  Hrc>|xs |j}|dury tjtjj |t|d5}tjt||dddy#t t f$rYFwxYw#1swYyxYw)zgSave current configuration to file as YAML Uses ``self.filename`` for target location FNw) r!rmakedirspathdirnameOSErrorIOErroropenyamldumpdict)r(fnfs rsavez Config.save?s  4== ;   KK+ ,"c] %a IId4j! $ % %!    % %s2A> B>BBBc#VKd|j|j|ywN)r,r-)r(r#s r_unsetz Config._unsetOs   Ds')c tj|}|r|j|ddlm}|j||||j |S)a>Change config values within a context or for the session values: dict This can be deeply nested to set only leaf values See also: ``intake.readers.utils.nested_keys_to_dict`` Examples -------- Value resets after context ends >>> with intake.conf.set(mybval=5): ... ... Set for whole session >>> intake.conf.set(myval=5) Set only a single leaf value within a nested dict >>> intake.conf.set(intake.readers.utils.nested_keys_to_dict({"deep.2.key": True}) r) merge_dicts)copydeepcopyr-intake.readers.utilsrCrA)r( update_dictkwr#rCs rsetz Config.setUsF0}}T"  IIk "4 Kb)*{{4  rcb||vrt||S|tvr t|St|r@)r$ __getitem__r.KeyError)r(itemr*s rrKzConfig.__getitem__us6 4<7&t, , X D> !4. rc.||vrt||S|Sr@)r$rK)r(keydefaultr*s rgetz Config.get}s $;7&s+ +rcd|j|j|jyr@)r0loadload_envr/s rr"zConfig.reload_alls   rcR|xs |j}tjj|r0t |5} |j t |dddyy#t$r0}tjdj||Yd}~>d}~wwxYw#1swYyxYw)zUpdate global config from YAML file If fn is None, looks in global config directory, which is either defined by the INTAKE_CONF_DIR env-var or is ~/.intake/ . z'Failure to load config file "{fn}": {e})r<eN) r!rr4isfiler8r-r Exceptionr&warningformat)r(r<r=rVs rrSz Config.loads 4== 77>>" b dQdKK ! - d d !dNN#O#V#VZ\`a#V#bccd d ds.BA!! B*&BBBBB&ctjjD]E\}}|jds|ddj } t j |}||<Gy#t$rYwxYw)z9Analyse environment variables and update conf accordinglyINTAKE_N)renvironitems startswithlowerast literal_eval ValueError)r(kvk2vals rrTzConfig.load_envsuJJ$$& DAq||I&qrU[[]**1-CR  "s A(( A43A4r@)__name__ __module__ __qualname____doc__r%r0r> contextlibcontextmanagerrArIrKrQr"rSrT __classcell__)r*s@rrr-sQ ) %  !@!  d" rrct|ttfr|Sddl}|j t j dk(rdnd}|j|S)a Return a list of directories from the intake path. If a string, perhaps taken from an environment variable, then the list of paths will be split on the character ":" for posix of ";" for windows. Protocol indicators ("protocol://") will be ignored. rNnt;z(? save_confrS load_condrrrrs1    8"   8 $ )"))$5rww||JsOU^7_` a `uTup x II II r