File: //usr/lib/python3.9/site-packages/ipaclient/plugins/__pycache__/dns.cpython-39.pyc
a
}�f^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 d Z d
Z dZ!d/dd�Z"d0dd�Z#d1dd�Z$G dd� de�Z%e ddd�G dd� de%��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 d-d.� d.e��Z/dS )2� )�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 | �� |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# �9/usr/lib/python3.9/site-packages/ipaclient/plugins/dns.py�__get_part_param4 s �r% c
C s� d }|d ur0t t| �� � }|jj�||�d }i }z|jjd�| �� � }W n tyd | Y S 0 t |�
� �D ]H\}}t | �� |j f }||j
vr�qr|r�|| } nd } t
| |||| � qr|S )N�result� 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, ��
r1 Fc
C s� i }z|j jd�| �� � }W n ty4 | Y S 0 |�� D ]^}t| �� |jf }||jvr`q>||v rjq>|j }|r||s|q>|j f i |��} t
| |||| � q>|S )Nr' ) r( r* r+ r r, r r r r � get_defaultr% )
r r �kw�prompt_optionalr/ r0 r! r r r" r# r# r$ �prompt_missing_partsZ s"
r5 c s e Zd Z� fdd�Z� ZS )�DNSZoneMethodOverridec # s6 t t| ��� D ]"}|jdkr*|�|jt�}|V qd S )NZidnsallowdynupdate)�superr6 � get_optionsr Z clone_retyper )�self�option�� __class__r# r$ r8 t s
z!DNSZoneMethodOverride.get_options)�__name__�
__module__� __qualname__r8 �
__classcell__r# r# r; r$ r6 s s r6 T)�override�no_failc @ s e Zd ZdS )� dnszone_addN�r= r> r? r# r# r# r$ rC { s rC c @ s e Zd ZdS )� dnszone_modNrD r# r# r# r$ rE � s rE )rB c @ s0 e Zd Zed�ZdZed�ed�fZdd� ZdS )r) zSplit DNS record to partsTr �valuec s� � fdd�}d}t |�}|dv r*|d�}n�|dv r<|d�}n�|dv rN|d �}n�|d
v r`|d �}nt|d v rp� f}nd|d
kr�t�d�}|�� �} | d ur�tdd� | �� D ��}n(|dkrԈ �d d�}
t|
�dkr�t|
�}t|d�S )Nc s � � � }t|�| krd S t|�S �N)�split�len�tuple)�count�values�rF r# r$ �
split_exactly� s z4dnsrecord_split_parts.execute.<locals>.split_exactlyr# )r r ZCNAMEZDNAMEr r � )ZAFSDBZKXr � )ZCERTZDLVZDSZSRVZTLSA� ZNAPTR� )ZA6ZTXTr 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 ur|� � n|V qd S rG )�strip)�.0�xr# r# r$ � <genexpr>� s z0dnsrecord_split_parts.execute.<locals>.<genexpr>ZSSHFP� )r& ) r �re�compile�matchrJ �groupsrH rI �dict) r9 r rF �args�optionsrN r& r Zregex�mrL r# rM r$ �execute� s6
�
�
z dnsrecord_split_parts.executeN) r= r> r? r �__doc__ZNO_CLIr Z
takes_argsr` 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
C s, zZt | || j� i }t| |dd�D ](}t|j�}t|| |dd�}|�|� q"|�|� W d S tjyn Y n0 zt |d �}W n6 t
y� } z tj dt |�d��W Y d }~n
d }~0 0 zt |d �}W n8 t
�y } z tj dt |�d��W Y d }~n
d }~0 0 |�
� �rd�t�} n |�� �r.d�t�} n
d�t�} | jj�td �� | jj�td
�| � d}
|
�s| jj�td ��}|d u �r�d S |�� }z,t|�� } | j| } d | j v �r�t � �W nL t t f�y
d�d
d� | �� D ��}
| jj�td�|
� Y �qdY n0 d}
�qdt || �}|�|� d S )NT)Z
skip_extraF)r4 �idnsname)r �error�dnszoneidnsname�, 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 |jvrt |j�V qdS )rg N)r r �flags�rT �pr# r# r$ rV s
�z<dnsrecord_add.interactive_prompt_callback.<locals>.<genexpr>z3Invalid or unsupported type. Allowed values are: %s) r �
no_option_msgr r r r5 �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 rh �
ValueErrorr, r1 )r9 r3 Znew_kwZrrparamr r/ rc �eZzonenameZ common_types�okr �paramZ all_typesr# r# r$ �interactive_prompt_callback� sh �
�
((
�
�
z)dnsrecord_add.interactive_prompt_callbackN�r= r> r? rk r| r# r# r# r$ rb � s rb c @ s e Zd ZdZdd� ZdS )�
dnsrecord_modz0No options to modify a specific record provided.c C s� zt | || jd� W n tjy( Y n0 d S | jjd |d |d �d }| jj�t d�� | jj�t d�� g }|D ]p}z| j
| }W n t y� Y qxY n0 t |j
�}|s�qx|�||f� d�||j
�}| jj�d |j|f � qx| jj�d
� |D ]�\}}t||j
�}||j
D ]�} |�| � | jjjt d �t|j| d � d
d�}
|
du �r t|| | d�} | g||j
< |�| � |�r | jj�tddd�tt|�|d� � q��q q�d S )NT�dnsrecord_showre rc r& z-No option to modify specific record provided.� Current DNS record contents:
rf �%s: %s� z Modify %(name)s '%(value)s'?�r rF 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 )rK �type)r rk r rm r( r
rs rt ru r r r, r r �appendro r �list�remove� prompt_yesnor\ r1 rl r rI ) r9 r3 �
dns_recordZ
record_params�attrr{ r �rec_type_contentZ
rec_values� rec_valueZ mod_valuer/ r# r# r$ r| sV
�
�
��z)dnsrecord_mod.interactive_prompt_callbackNr} 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� |� dd�rd S zt| || j� W n tjy6 Y n0 d S | jjd |d |d �d }| jj� t
d�� | jjj t
d�dd �}|d
u r�d
|d< d S | jj� t
d �� g }|D ]h}z| j | }W n t
y� Y q�Y n0 t|j�s�q�|�|� d �||j �}| jj� d
|j|f � q�| jj� d� |D ]h}g }||j D ]>} | jjj t
d�t|j| d� dd �}
|
d
u �r>|�| � �q>|�r,t|�||j< �q,d S )NZdel_allFr re rc r& z-No option to delete specific record provided.z Delete all?r� Tr� rf r� r� z Delete %(name)s '%(value)s'?r� )�getr rk r rm r( r
rs rt ru r r� r r, r r r� ro r r\ rJ ) r9 r3 r� Z user_del_allZpresent_paramsr� r{ r� Zdeleted_valuesr� Zuser_del_valuer# r# r$ r| L sN
��
z)dnsrecord_del.interactive_prompt_callbackN)r= r> r? r rk r| r# r# r# r$ r� G s r� c @ s e Zd Zdd� ZdS )�
dnsconfig_modc C s4 |� dd�r0| jj�td�� | jj�td�� d S �N�idnsforwardersF�#Server will check DNS forwarder(s).�(This may take some time, please wait ...�r� rs rt ru r �r9 r3 r# r# r$ r| � s ��z)dnsconfig_mod.interactive_prompt_callbackN�r= r> r? r| r# r# r# r$ r� � s r� c @ s e Zd Zdd� ZdS )�dnsforwardzone_addc C s` d|vr,|� d�dkr,| jj�td��|d< |� dd�r\| jj�td�� | jj�td�� d S )Nr� ZidnsforwardpolicyZnonez
DNS forwarderFr� r� )r� rs rt rv r ru r� r# r# r$ r| � s �� ��z.dnsforwardzone_add.interactive_prompt_callbackNr� r# r# r# r$ r� � s r� c @ s e Zd Zdd� ZdS )�dnsforwardzone_modc C s4 |� dd�r0| jj�td�� | jj�td�� d S r� r� r� r# r# r$ r| � s ��z.dnsforwardzone_mod.interactive_prompt_callbackNr� r# r# r# r$ r� � s r� c sP e Zd ZdZedded�d�fZdd� Zdd � 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)Zinclude�docc C s\ | j D ]P}|�|�r|jd�|| �dd� t|| �D ]}|j|dd� q8|�d� qdS )z?Print output in standard format common across the other pluginsz{}:rO )�indentrP r� N)�
record_groupsr� Zprint_indentedr+ �sortedZ
print_line)r9 rt r& �labels�key�valr# r# r$ �_standard_output� s
z*dns_update_system_records._standard_outputc C s� dd� }dd� | � � D �}t� }| jD ]�}|�|�r&|�d�|| �� t|| �D ]J}||�}||vr�|�|� |�dj|d |d d �� |�d
�|�� qT|�d � q&d S )
z%Store data in nsupdate format in filec S s | � dd�}|d |d fS )z9Get rname and rtype from textual representation of record� rQ r rW )rH )�record�lr# r# r$ �parse_rname_rtype� s zJdns_update_system_records._nsupdate_output_file.<locals>.parse_rname_rtypec S s i | ]}|j t|j��qS r# �r r r ri r# r# r$ �
<dictcomp>� s zCdns_update_system_records._nsupdate_output_file.<locals>.<dictcomp>z; {}
z update delete {rname} {rtype}
r rO )ZrnameZrtypezupdate add {}
zsend
N)�
output_params�setr� r� �writer+ r� �add) r9 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� |� dd �}|rt�|� tt| �j|i |��}|r�d|v r�z>t|d�� }| �||d � W d � n1 sn0 Y W n8 tt fy� } z t
j t |�d��W Y d }~n
d }~0 0 |S )N�outr& �w)�reason)
�popr Zcheck_writable_filer7 r� �forward�openr� �OSError�IOErrorr Z FileErrorr )r9 �keysr^ r� �res�fry r; r# r$ r� � s
2&z!dns_update_system_records.forwardc s� t �|�}|�di �}|�dd � |�dd � tt| �j||g|�R i |�� dd� | �� D �}|�di �}| �|||� t |d �S )Nr& r� r� c S s i | ]}|j t|j��qS r# r� ri r# r# r$ r� � s z<dns_update_system_records.output_for_cli.<locals>.<dictcomp>rF )
�copy�deepcopyr� r� r7 r� �output_for_clir� r� �int) r9 rt �outputr] r^ Z output_superZ super_resr� r& r; r# r$ r� � s"
���� z(dns_update_system_records.output_for_cli) r= r> r? r� r r Z
takes_optionsr� r� r� r� r@ r# r# r; r$ r� � s �� r� )N)N)F)0Z
__future__r Zsixr� rX Zipaclient.frontendr Zipalibr Z
ipalib.dnsr r r r r Zipalib.frontendr
Zipalib.parametersr r Zipalib.plugabler
r r r Zipapython.dnsutilr ZPY3�strr �registerrr rq rp r% r1 r5 r6 rC rE r) rb r~ r� r� r� r� r� r# r# r# r$ �<module> sT
8
L
9
8