File: //usr/lib/python3.6/site-packages/ipaclient/plugins/__pycache__/dns.cpython-36.opt-1.pyc
3
�d[e^H � @ s d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
m Z m Z m
Z
mZ d dlmZ d dlmZmZ d dlmZ d d lmZmZ d d
lmZ d d lmZ ejr�e Z e� Z d2Z d3Z d4Z!d5dd�Z"d6dd�Z#d7dd�Z$G dd� de�Z%e ddd �G d d � d e%��Z&e ddd �G d d � d e%��Z'e dd!�G d"d#� d#e��Z(e ddd �G d$d%� d%e��Z)e ddd �G d&d'� d'e��Z*e ddd �G d(d)� d)e��Z+e ddd �G d*d+� d+e��Z,e ddd �G d,d-� d-e��Z-e ddd �G d.d/� d/e��Z.e ddd �G d0d1� d1e��Z/dS )8� )�print_functionN)�MethodOverride)�errors)�get_record_rrtype�has_cli_options�iterate_rrparams_by_parts�part_name_format�record_name_format)�Command)�Bool�Str)�Registry)�_�ngettext)�util)�DNSName�A�AAAA�PTR�NS�MX�LOCc C sB t | j� |jf }t|j| j�}|j }|j|||d�||< d S )N)�optional�label)r �lower�name�unicode�paramsr �requiredZ prompt_param)�rrtype�cmd�partZ output_kw�defaultr r r � r# �/usr/lib/python3.6/dns.py�__get_part_param4 s r% c
C s� d }|d k r0t t| j� � }|jjj||�d }i }y|jjdj| j� � }W n tk
rb |S X xXt |j
� �D ]H\}}t | j� |j f }||j
kr�qr|r�|| } nd } t
| |||| � qrW |S )N�resultz dns{}record)r r r �apir
�dnsrecord_split_parts�Object�format�KeyError� enumerater r r r% )
r r � mod_dnsvalueZ mod_partsr � user_options�rrobjZpart_idr! r" r# r# r$ � prompt_parts> s&
r0 Fc
C s� i }y|j jdj| j� � }W n tk
r2 |S X xh|j� D ]\}t| j� |jf }||jkr`q>||krjq>|j }|r~| r~q>|j f |�} t
| |||| � q>W |S )Nz dns{}record) r' r) r* r r+ r r r r Z get_defaultr% )
r r �kw�prompt_optionalr. r/ r! r r r" r# r# r$ �prompt_missing_partsZ s"
r3 c s e Zd Z� fdd�Z� ZS )�DNSZoneMethodOverridec # s: x4t t| �j� D ]"}|jdkr,|j|jt�}|V qW d S )NZidnsallowdynupdate)�superr4 � get_optionsr Z clone_retyper )�selfZoption)� __class__r# r$ r6 t s
z!DNSZoneMethodOverride.get_options)�__name__�
__module__� __qualname__r6 �
__classcell__r# r# )r8 r$ r4 s s r4 T)�override�no_failc @ s e Zd ZdS )� dnszone_addN)r9 r: r; r# r# r# r$ r? { s r? c @ s e Zd ZdS )� dnszone_modN)r9 r: r; r# r# r# r$ r@ � s r@ )r> c @ s0 e Zd Zed�ZdZed�ed�fZdd� ZdS )r( zSplit DNS record to partsTr �valuec s� � fdd�}f }t |�}|d kr*|d �}n�|d kr<|d
�}n�|d!krN|d�}n�|dkr`|d�}nt|d"krp� f}nd|dkr�tjd�}|j� �} | d k r�tdd� | j� D ��}n(|d krԈ jd d
�}
t|
�d kr�t|
�}t|d �S )#Nc s � j � }t|�| krd S t|�S )N)�split�len�tuple)�count�values)rA r# r$ �
split_exactly� s z4dnsrecord_split_parts.execute.<locals>.split_exactlyr r �CNAME�DNAMEr r � �AFSDB�KXr � �CERT�DLV�DS�SRV�TLSA� ZNAPTR� �A6�TXTr aR (?P<d1>\d{1,2}\s+)(?:(?P<m1>\d{1,2}\s+)(?P<s1>\d{1,2}(?:\.\d{1,3})?\s+)?)?(?P<dir1>[NS])\s+(?P<d2>\d{1,3}\s+)(?:(?P<m2>\d{1,2}\s+)(?P<s2>\d{1,2}(?:\.\d{1,3})?\s+)?)?(?P<dir2>[WE])\s+(?P<alt>-?\d{1,8}(?:\.\d{1,2})?)m?(?:\s+(?P<siz>\d{1,8}(?:\.\d{1,2})?)m?(?:\s+(?P<hp>\d{1,8}(?:\.\d{1,2})?)m?(?:\s+(?P<vp>\d{1,8}(?:\.\d{1,2})?)m?\s*)?)?)?$c s s" | ]}|d k r|j � n|V qd S )N)�strip)�.0�xr# r# r$ � <genexpr>� s z0dnsrecord_split_parts.execute.<locals>.<genexpr>ZSSHFP� )r& )r r rH rI r r )rK rL r )rN rO rP rQ rR )rU rV ) r �re�compile�matchrD �groupsrB rC �dict) r7 r rA �args�optionsrG r& r Zregex�mrF r# )rA r$ �execute� s2
z dnsrecord_split_parts.executeN) r9 r: r; r �__doc__ZNO_CLIr Z
takes_argsrd r# r# r# r$ r( � s
r( c @ s e Zd ZdZdd� ZdS )�
dnsrecord_addzkNo options to add a specific record provided.
Command help may be consulted for all supported record types.c 0 C s0 y\t | || j� i }x:t| |dd�D ](}t|j�}t|| |dd�}|j|� q$W |j|� d S tjk
rr Y nX yt |d �}W n4 t
k
r� } ztj dt |�d��W Y d d }~X nX yt |d �}W n6 t
k
�r } ztj dt |�d��W Y d d }~X nX |j
� �rdjt�} n |j� �r.djt�} n
djt�} | jjjtd �� | jjjtd
�| � d}
x�|
�s| jjjtd ��}|d k�r�d S |j� }y,t|j� } | j| } d | j k�r�t � �W nL t t fk
�r djd
d� | j� D ��}
| jjjtd�|
� �wfY nX d}
�qfW t || �}|j|� d S )NT)Z
skip_extraF)r2 �idnsname)r �error�dnszoneidnsnamez, z7Please choose a type of DNS resource record to be addedz4The most common types for this type of zone are: %s
zDNS resource record type� no_optionc s s, | ]$}t |j�rd |jkrt |j�V qdS )rj N)r r �flags)rX �pr# r# r$ rZ s
z<dnsrecord_add.interactive_prompt_callback.<locals>.<genexpr>z3Invalid or unsupported type. Allowed values are: %s) r �
no_option_msgr r r r3 �updater � OptionErrorr � ExceptionZValidationErrorr Zis_empty�join�_zone_top_record_typesZ
is_reverse�_rev_top_record_types�_top_record_types�Backend�textui� print_plainr �prompt�upperr r r rk �
ValueErrorr+ r0 )r7 r1 Znew_kwZrrparamr r. rg �eZzonenameZ common_types�okr �paramZ all_typesr# r# r$ �interactive_prompt_callback� s`
$$
z)dnsrecord_add.interactive_prompt_callbackN)r9 r: r; rm r~ r# r# r# r$ rf � s rf c @ s e Zd ZdZdd� ZdS )�
dnsrecord_modz0No options to modify a specific record provided.c C s� yt | || jd� W n tjk
r* Y nX d S | jjd |d |d �d }| jjjt d�� | jjjt d�� g }xx|D ]p}y| j
| }W n t k
r� w|Y nX t |j
�}|s�q||j||f� dj||j
�}| jjjd |j|f � q|W | jjjd
� x�|D ]�\}}t||j
�}x�||j
D ]�} |j| � | jjjt d �t|j| d � d
d�}
|
dk�r&t|| | d�} | g||j
< |j| � |�r&| jjjtddd�tt|�|d� � P �q&W �qW d S )NT�dnsrecord_showri rg r& z-No option to modify specific record provided.z Current DNS record contents:
z, z%s: %s� z Modify %(name)s '%(value)s'?)r rA F)r" )r- zb%(count)d %(type)s record skipped. Only one value per DNS record type can be modified at one time.zc%(count)d %(type)s records skipped. Only one value per DNS record type can be modified at one time.r )rE �type)r rm r ro r' r
ru rv rw r r r+ r r �appendrq r �list�remove� prompt_yesnor` r0 rn r rC ) r7 r1 �
dns_recordZ
record_params�attrr} r �rec_type_contentZ
rec_values� rec_valueZ mod_valuer. r# r# r$ r~ sL
z)dnsrecord_mod.interactive_prompt_callbackN)r9 r: r; rm r~ r# r# r# r$ r
s r c @ s e Zd Zed�Zdd� ZdS )�
dnsrecord_delz�Neither --del-all nor options to delete a specific record provided.
Command help may be consulted for all supported record types.c C s� |j dd�rd S yt| || j� W n tjk
r8 Y nX d S | jjd |d |d �d }| jjj t
d�� | jjj t
d�dd �}|d
kr�d
|d< d S | jjj t
d �� g }xp|D ]h}y| j | }W n t
k
r� w�Y nX t|j�s�q�|j|� d j||j �}| jjj d
|j|f � q�W | jjj d� xt|D ]l}g }xL||j D ]>} | jjj t
d�t|j| d� dd �}
|
d
k�rH|j| � �qHW |�r4t|�||j< �q4W d S )NZdel_allFr� ri rg r& z-No option to delete specific record provided.z Delete all?)r" Tz Current DNS record contents:
z, z%s: %sr� z Delete %(name)s '%(value)s'?)r rA )�getr rm r ro r' r
ru rv rw r r� r r+ r r r� rq r r` rD ) r7 r1 r� Z user_del_allZpresent_paramsr� r} r� Zdeleted_valuesr� Zuser_del_valuer# r# r$ r~ L sH
z)dnsrecord_del.interactive_prompt_callbackN)r9 r: r; r rm r~ r# r# r# r$ r� G s r� c @ s e Zd Zdd� ZdS )�
dnsconfig_modc C s4 |j dd�r0| jjjtd�� | jjjtd�� d S )N�idnsforwardersFz#Server will check DNS forwarder(s).z(This may take some time, please wait ...)r� ru rv rw r )r7 r1 r# r# r$ r~ � s
z)dnsconfig_mod.interactive_prompt_callbackN)r9 r: r; r~ r# r# r# r$ r� � s r� c @ s e Zd Zdd� ZdS )�dnsforwardzone_addc C s` d|kr,|j d�dkr,| jjjtd��|d< |j dd�r\| jjjtd�� | jjjtd�� d S )Nr� ZidnsforwardpolicyZnonez
DNS forwarderFz#Server will check DNS forwarder(s).z(This may take some time, please wait ...)r� ru rv rx r rw )r7 r1 r# r# r$ r~ � s
z.dnsforwardzone_add.interactive_prompt_callbackN)r9 r: r; r~ r# r# r# r$ r� � s r� c @ s e Zd Zdd� ZdS )�dnsforwardzone_modc C s4 |j dd�r0| jjjtd�� | jjjtd�� d S )Nr� Fz#Server will check DNS forwarder(s).z(This may take some time, please wait ...)r� ru rv rw r )r7 r1 r# r# r$ r~ � s
z.dnsforwardzone_mod.interactive_prompt_callbackN)r9 r: r; r~ r# r# r# r$ r� � s r� c sP e Zd ZdZedded�d�fZdd� Zd d
� Z� fd d �Z � fd
d�Z
� Z S )�dns_update_system_records� ipa_records�location_recordszout?Zcliz,file to store DNS records in nsupdate format)�include�docc C sd x^| j D ]T}|j|�r|jdj|| �dd� x"t|| �D ]}|j|dd� q<W |jd� qW dS )z?Print output in standard format common across the other pluginsz{}:rJ )�indentrM r� N)�
record_groupsr� Zprint_indentedr* �sortedZ
print_line)r7 rv r& �labels�key�valr# r# r$ �_standard_output� s
z*dns_update_system_records._standard_outputc C s� dd� }dd� | j � D �}t� }x�| jD ]�}|j|�r(|jdj|| �� xZt|| �D ]J}||�}||kr�|j|� |jdj|d |d d �� |jd
j|�� qXW |jd � q(W d S )
z%Store data in nsupdate format in filec S s | j dd�}|d |d fS )z9Get rname and rtype from textual representation of record� rS r r[ )rB )�record�lr# r# r$ �parse_rname_rtype� s zJdns_update_system_records._nsupdate_output_file.<locals>.parse_rname_rtypec S s i | ]}t |j�|j�qS r# )r r r )rX rl r# r# r$ �
<dictcomp>� s zCdns_update_system_records._nsupdate_output_file.<locals>.<dictcomp>z; {}
z update delete {rname} {rtype}
r rJ )ZrnameZrtypezupdate add {}
zsend
N)�
output_params�setr� r� �writer* r� �add) r7 Zout_fr& r� r� Zalready_removedr� r� Z r_name_typer# r# r$ �_nsupdate_output_file� s
z/dns_update_system_records._nsupdate_output_filec s� |j dd �}|rtj|� tt| �j||�}|r�d|kr�y*t|d��}| j||d � W d Q R X W n6 tt fk
r� } zt
j t |�d��W Y d d }~X nX |S )N�outr& �w)�reason)
�popr Zcheck_writable_filer5 r� �forward�openr� �OSError�IOErrorr Z FileErrorr )r7 �keysrb r� �res�fr{ )r8 r# r$ r� � s
"z!dns_update_system_records.forwardc s� t j|�}|jdi �}|jdd � |jdd � tt| �j||f|�|� dd� | j� D �}|jdi �}| j|||� t |d �S )Nr& r� r� c S s i | ]}t |j�|j�qS r# )r r r )rX rl r# r# r$ r� � s z<dns_update_system_records.output_for_cli.<locals>.<dictcomp>rA )
�copy�deepcopyr� r� r5 r� �output_for_clir� r� �int) r7 rv �outputra rb Z output_superZ super_resr� r& )r8 r# r$ r� � s
z(dns_update_system_records.output_for_cli)r� r� ) r9 r: r; r� r r Z
takes_optionsr� r� r� r� r< r# r# )r8 r$ r� � s r� )r r )r )r r r )N)N)F)0Z
__future__r Zsixr� r\ Zipaclient.frontendr Zipalibr Z
ipalib.dnsr r r r r Zipalib.frontendr
Zipalib.parametersr r Zipalib.plugabler
r r r Zipapython.dnsutilr ZPY3�strr �registerrt rs rr r% r0 r3 r4 r? r@ r( rf r r� r� r� r� r� r# r# r# r$ �<module> sT
8L98