3 ���f! � @ s: d dl mZ ddlmZ d dlZd dlZG dd� d�ZdS )� )� defaultdict� )�dnskeyNc @ sn e Zd ZdZedd� �ZdZg Zddd�Zdd� Z d d � Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� ZdS )�keydictz> A dictionary of keys, indexed by name, algorithm, and key id c C s t t�S )N)r �dict� r r � /usr/lib/python3.6/keydict.py�<lambda> s zkeydict.<lambda>Nc K s� |j dd �| _|j dd �}|s:|j dd �p,d}| j|� nXxV|D ]N}d|krb|d d k rb|d }n|rr|j|�jptd}| j||�s@| jj|� q@W d S )NZkeyttl�zones�path�.)�get�_defttl�readallZpolicyZ directory�readone�_missing�append)�selfZdp�kwargsr r �zoner r r �__init__ s zkeydict.__init__c C sL t j tjj|d��}x2|D ]*}t||| j�}|| j|j |j |j < qW d S )Nz *.private) �glob�osr �joinr r �_keydict�name�alg�keyid)r r �files�infile�keyr r r r , s zkeydict.readallc C s� |j d�s|d7 }d| d }tjtjj||��}d}xR|D ]J}t||| j�}|j|krZq<|dkrh|jnd}|| j | |j |j < d}q<W |S )Nr �Kz +*.privateFT) �endswithr r r r r r �fullnamer r r r ) r r r �matchr �foundr r Zkeynamer r r r 3 s zkeydict.readonec c sJ xD| j j� D ]6\}}x,|j� D ] \}}x|j� D ] }|V q0W q W q W d S )N)r �items�values)r r � algorithmsr �keysr r r r �__iter__D s zkeydict.__iter__c C s | j | S )N)r )r r r r r � __getitem__J s zkeydict.__getitem__c C s | j j� S )N)r r) )r r r r r M s z keydict.zonesc C s | j | j� S )N)r r) )r r r r r r( P s zkeydict.algorithmsc C s | j | | j� S )N)r r) )r r r r r r r) S s z keydict.keysc C s | j S )N)r )r r r r �missingV s zkeydict.missing)N)�__name__� __module__� __qualname__�__doc__r r r r r r r r* r+ r r( r) r, r r r r r s r )� collectionsr � r r r r r r r r �<module> s