
    ogfL3                     
   d Z dZdZddlZddlZddlZddlmZ ej                  dk\  rddl	m
ZmZmZmZmZ  G d	 d
e      Zn'ddl	mZmZmZmZ ej"                  e_        d Zee_        d Z ee      ZdZdgZ G d de      Zy)z0Yaroslav Halchenko, Serg G. Brester (aka sebres)zHCopyright (c) 2007 Yaroslav Halchenko, 2015 Serg G. Brester (aka sebres)GPL    N   )	getLogger)   )ConfigParserBasicInterpolationInterpolationMissingOptionErrorNoOptionErrorNoSectionErrorc                   "     e Zd ZdZ fdZ xZS )BasicInterpolationWithNamea  Decorator to bring __name__ interpolation back.

		Original handling of __name__ was removed because of
		functional deficiencies: http://bugs.python.org/issue10489

		commit v3.2a4-105-g61f2761
		Author: Lukasz Langa <lukasz@langa.pl>
		Date:	Sun Nov 21 13:41:35 2010 +0000

		Issue #10489: removed broken `__name__` support from configparser

		But should be fine to reincarnate for our use case
		c                     |rNt         |vrE|j                         }||d<   |j                  ||||       t        t        |   ||||||g|i |S y y N__name__)r   copy_map_section_optionssuperr   _interpolate_some)
selfparseroptionaccumrestsectionmapargskwargs	__class__s
            A/usr/lib/python3/dist-packages/fail2ban/client/configparserinc.pyr   z,BasicInterpolationWithName._interpolate_some6   sq    x3
((*CC
O
s;+TDVUD'3A15A9?A A 'g    )r   
__module____qualname____doc__r   __classcell__r   s   @r   r   r   (   s    A Ar    r   )SafeConfigParserr	   r
   r   c                 ^    | j                  ||||        | j                  |||||g|i |S N)r   _cp_interpolate_some)r   r   r   r   r   r   r   r   s           r   r   r   F   s;    GVT37	"	"	"65$	Vt	Vv	VVr    c                     g }| D ]n  }|j                  |       t        j                  j                  |      d   dz   }|| vs>t        j                  j	                  |      s^|j                  |       p |S )z,Expands config files with local extension.
	r   z.local)appendospathsplitextisfile)	filenamesnewFilenamesfilename	localnames       r   _expandConfFilesWithLocalr4   L   sq      "Xhggx(+h6)iBGGNN9$=y!	"
 	r       SafeConfigParserWithIncludesc                       e Zd ZdZdZ ej                  d      Z ej                  d      Z ej                  d      Z	e
j                  dk\  rd fd	ZnddZd	i fd
Zd Zed        Zd Zg fdZg fdZd Zd ZddZddZddZ xZS )r6   a  
	Class adds functionality to SafeConfigParser to handle included
	other configuration files (or may be urls, whatever in the future)

	File should have section [includes] and only 2 options implemented
	are 'files_before' and 'files_after' where files are listed 1 per
	line.

	Example:

[INCLUDES]
before = 1.conf
         3.conf

after = 1.conf

	It is a simple implementation, so just basic care is taken about
	recursion. Includes preserve right order, ie new files are
	inserted to the list of read configs before original, and their
	includes correspondingly so the list should follow the leaves of
	the tree.

	I wasn't sure what would be the right way to implement generic (aka c++
	template) so we could base at any *configparser class... so I will
	leave it for the future

	INCLUDESz^([\w\-]+)/([^\s>]+)$z%\(([\w\-]+/([^\)]+))\)sz^(\w+)(\?.+)$r   r   c                 ~    |j                         }t               |d<   d|d<   t        t        |   |i | || _        y )Ninterpolation;inline_comment_prefixes)r   r   r   r6   __init__
_cfg_share)r   share_configr   r   r   s       r   r>   z%SafeConfigParserWithIncludes.__init__   sH    KKM6796/'*6
#$	%t5	!4?r    c                 D    t        j                  | g|i | || _        y r(   )r&   r>   r?   )r   r@   r   r   s       r   r>   z%SafeConfigParserWithIncludes.__init__   s!    T3D3F3!4?r    Fc                    d}d|v rt         j                  j                  |      }|rh|j                  d      }|j                  d      }|j	                         }|dk(  rd|z   |f}n
|dk7  r|fnd}|D ]  }	 | j                  |||	      }	|	c S  | j                  ||||
      }	|	S # t        t        f$ r
}
Y d}
~
Hd}
~
ww xY w)zGet an option value for a given section.
		
		In opposite to `get`, it differentiate session-related option name like `sec/opt`.
		N/   r   knownKNOWN/default)DEFAULT)raw)rI   vars)r6   SECTION_OPTNAME_CREsearchgrouplowergetr   r
   )r   r   r   rI   rJ   soptsecoptseclwrves              r   get_exz#SafeConfigParserWithIncludes.get_ex   s    
 
$F]
&
:
:
A
A&
I4		A3	A3IIK6Wg&Dy(C6lD 
s
	#s	$QX
 hhwCdh3!	
(	 M* 
	
s   7B''C ;C c                 J   d|vsd|vryd}t         j                  j                  |      }|sy|D ]  \  }}||vs|dt        |        }	|	j	                         }
|
dk7  r5d}|
dk(  r	 | j
                  d|z      |   }n	 	 | j
                  |	   }	|	|   }nd}|r	 | j                  |   }d}	 ||<    |S # t        $ r d}Y +w xY w# t        $ r Y w xY w# t        $ r d}Y Kw xY w# t        $ r Y w xY w#  	 |j                  d   |<   n#  | j                  |<   Y nxY wY xY w)	z
		Interpolates values of the section options (name syntax `%(section/option)s`).

		Fallback: try to wrap missing default options as "default/options" resp. "known/options"
		rC   z%(r   NrG   rE   rF   rD   )	r6   SECTION_OPTSUBST_CREfindalllenrN   	_sectionsKeyError	_defaults_maps)r   r   r   r   defaultsrplcmntsoptreprP   rR   rQ   rS   usedefrT   s                r   r   z1SafeConfigParserWithIncludes._map_section_options   s    	_D(
'(==EEdK'	
 *idC
(
zS	
CYY[FV'>>(7*+C0q	nnS! s8q
 V
..
a GXd^G*V 
.C  v    v   !hnnQdnnTsx   'B4?CCC%+C44CC	CCCCC"!C"%	C10C14D"7D
	D"
DD"c                     | j                   S r(   )r?   r   s    r   r@   z)SafeConfigParserWithIncludes.share_config   s    	r    c                    t         }| j                  rdt        |t              s|ndj	                  |      z   }| j                  j                  |d      \  }}|; || j                        }|j                  |d      }||f| j                  |<   ||fS t        j                         t        k  rt        j                  t        d|       ||fS  |       }|j                  |d      }||fS )Nzinc:)NN)r@   F)get_includesz    Shared file: %s)r6   r?   
isinstancelistjoinrO   readlogSysgetEffectiveLevellogLevellog)r   r2   SCPWIhashvcfgis         r   _getSharedSCPWIz,SafeConfigParserWithIncludes._getSharedSCPWI   s    
&%	__:h#=86;;xCXY5OO|463	k
T__
-C.A!1XDOOE q/ 	  "h.
JJx.9
 q/ 
3
xxux-1
q/r    c                 V   t        |t              s|g}t        |      }| j                  r`ddj	                  |      z   }| j                  j                  |      }|-g }|D ]  }|| j                  ||      z  } || j                  |<   |S g }|D ]  }|| j                  ||      z  } |S )Nz	inc-path:rf   )rh   ri   r4   r?   rj   rO   2_SafeConfigParserWithIncludes__getIncludesUncached)r   r0   seenrq   fileNamesFullr2   s         r   _getIncludesz)SafeConfigParserWithIncludes._getIncludes   s    	It	$}9'	2)	__I./5??&&u-=M Ad004@@]A*DOOE
- ?hD..x>>=?	r    c                    t         }	 | j                  |      \  }}|sg S 	 t
        j                  j                  |      }dg fdg fg}|j                  |j                         v r|D ]  \  }	}
|	|j                  |j                        v s$|j                  |j                  |	      }|j                  d      D ]d  }t
        j                  j                  |      r|}n t
        j                  j                  ||      }||v rJ||gz   }|
| j                  ||      z  }
f  |d   d   |gz   |d   d   z   S # t        $ r'}t        j	                  d|d|       g cY d}~S d}~ww xY w)	z
		Given 1 config resource returns list of included files
		(recursively) with the original one as well
		Simple loops are taken care about
		zError decoding config file 'z': Nbeforeafter
r   rD   )r6   rt   UnicodeDecodeErrorrl   errorr,   r-   dirnameSECTION_NAMEsectionsoptionsrO   splitisabsrj   ry   )r   resourcerw   rp   r   rs   rU   resourceDirnewFilesoption_nameoption_listnewResourcesnewResourcerss                  r   __getIncludesUncachedz2SafeConfigParserWithIncludes.__getIncludesUncached  sg    '%##H-961
I  )+"~}.(
6??,,#+ -{KfnnU%7%788JJu11;?\$**40 -	{	#q77<<[1q	
d
(
aT&&q!,,k-- 
!Q8*	$x{1~	55+ 
 	<<1EF
9s   D5 5	E%>E E% E%c                     | j                   S r(   )r]   rd   s    r   get_defaultsz)SafeConfigParserWithIncludes.get_defaults7      	r    c                     | j                   S r(   )r[   rd   s    r   get_sectionsz)SafeConfigParserWithIncludes.get_sections:  r   r    c                     	 | j                   |   }|r/t        |j	                               t        | j
                        z  S t        |j	                               S # t        $ r t        |      w xY w)zReturn a list of option names for the given section name.

		Parameter `withDefault` controls the include of names from section `[DEFAULT]`
		)r[   r\   r   setkeysr]   ri   )r   r   withDefaultoptss       r   r   z$SafeConfigParserWithIncludes.options=  sf    
!
..
!4 
diik
S0
00	diik	 
 !		  !s   A A0c                    t        |t              s|g}g }|r|| j                  |      z  }n|}|sg S t        j	                  d|       |st        |      dkD  rg }| j                         }| j                         }|D ]x  }| j                  |      \  }}	|	s||	z  }|j                  |j                                |j                         j                         D ]  \  }
}t        j                  j                  |
      }|r^|j                         \  }
}|j                         }	 |d= t        |j#                               D ]  }|j%                  |      }||||z   <    |j'                  |
      }t        |t(              r_| j+                  d|
z   t)        t-        |j                               D 	cg c]  }	|	d   |v s|	 c}	      d       |j                  |       |j                         ||
<    { |S t        j/                         t0        k  rt        j3                  t0        d|d          t4        j6                  dk\  rt9        j:                  | |d	
      S t9        j:                  | |      S # t         $ r Y Xw xY wc c}	w )Nz  Loading files: %srD   r   rF   r    z    Reading file: %sr9   zutf-8)encoding)rh   ri   ry   rl   inforZ   r   r   rt   updateitemsr6   CONDITIONAL_REmatchgroupsr   r\   r   poprO   dictmerge_sectioniterrm   rn   ro   sysversion_infor&   rk   )r   r0   rg   rx   retalldallsr2   rr   rs   nr   condkrT   s2s                   r   rk   z!SafeConfigParserWithIncludes.readL  sJ   	It	$}9-D%%i00==	
9++#]3S'!+	3



4



4  x!!(+FCAXS	[[!!#$!!#))+ A)88>>qAd	wq$668qjM QVVX 1EE!H!d( 88A;b	B	(1*bhhj)7AQqTQYa78"> 
yy|tAw+< : 8+	::h.a0@A


mg
FF


m
441   8s   'I61J>J6	JJc                 *   | j                         }	 ||   }|s|j                  |       y i }|j	                         D ]%  \  }}|j                  |      r|dk7  s||||z   <   ' |j                  |       y # t        $ r t               x||<   }Y zw xY wr   )r   r\   r   r   r   
startswith)	r   r   r   prefr   rQ   skr   rT   s	            r   r   z*SafeConfigParserWithIncludes.merge_section  s    				$ 	g3 
::g		"mmo da
,,t
jBtAvJ **R. 
  4=3 s   A7 7BBr(   )T)r   r!   r"   r#   r   recompilerK   rX   r   r   r   r>   rV   r   propertyr@   rt   ry   rv   r   r   r   rk   r   r$   r%   s   @r   r6   r6   _   s    8 !rzz":;"

#>?"**-."" (-2 :7r  & )+ * 13  6D;5zr    )
__author____copyright____license__r,   r   r   helpersr   r   configparserr   r&   r   r	   r
   r   r   r   r)   r4   r   rl   rn   __all__r6    r    r   <module>r      s   . @
Z 	 	 
 tA A
A"4 A2A A *:)K)K&W '8#	 
8	 *
*w#3 wr    