Current File : /opt/alt/python311/lib/python3.11/site-packages/pydig/__pycache__//resolver.cpython-311.pyc
�
��;f � �d � d dl Z d dlZd dlZddlmZ ej e� � Z G d� d� � ZdS )� N� )� QueryTypec �V � e Zd Z d d�Zed� � � Zed� � � Zd� Zd� Z dS )
�Resolver�digN�utf-8c �F � || _ |pg | _ |pg | _ || _ dS )zN
Stores some customisable options into this resolver instance
N)�
executable� nameservers�additional_args�encoding)�selfr
r r r
s �A/opt/alt/python311/lib/python3.11/site-packages/pydig/resolver.py�__init__zResolver.__init__ s0 � � %���&�,�"��� .� 4�"�� � ��
�
�
� c �4 � t j | j � � S )zE
Returns a random nameserver we should query against
)�random�choicer )r s r �
nameserverzResolver.nameserver s � �
�}�T� -�.�.�.r c � � t � d� t | � � � � � � t j | � � S )zr
Calls out to subprocess with the passed in args
This method is normally mocked in tests
z%Executing subprocess.check_output({}))�logger�info�format�repr�
subprocess� check_output)�argss r �_executezResolver._execute" s< � � � � �;�B�B�4��:�:�N�N�O�O�O��&�t�,�,�,r c # � K � | j V � | j r d� | j � � V � |V � |j V � dV � | j E d{V �� dS )zr
Builds up the final arguments to pass into subprocess
dig @1.1.1.1 example.com A +short
z@{}z+shortN)r
r r r �namer )r �domain�
query_types r �_argszResolver._args, s� � � � � �o� � � � � � 0��,�,�t��/�/� /� /� /� � � � � �o� � � � ���� �'�'�'�'�'�'�'�'�'�'r c �H � |� � � }t j |� � }t | � ||� � � � }| � |� � � | j � � � � � }|r|� d� � ng S )zn
Queries the resolver for a specific domain and query type
Returns a list of records
�
)
�lowerr �get�listr# r �decoder
�strip�split)r r! r" r �outputs r �queryzResolver.queryD s� � � � � � � �� �]�:�.�.�
��D�J�J�v�z�2�2�3�3��� � �t�$�$�+�+�D�M�:�:�@�@�B�B��%+�3�v�|�|�D�!�!�!��3r )r NNr )
�__name__�
__module__� __qualname__r �propertyr � staticmethodr r# r- � r r r r
s� � � � � � � �� ��
!�
!�
!�
!� �/� /� �X�/� �-� -� �\�-�(� (� (�0 4� 4� 4� 4� 4r r ) r r �loggingr" r � getLoggerr. r r r3 r r �<module>r6 s� ��
�
�
�
� � � � � ���� !� !� !� !� !� !� �� �8� $� $��F4� F4� F4� F4� F4� F4� F4� F4� F4� F4r 