U
�zk^� � @ s~ d dl Z d dlmZ d dlmZmZmZ d dlmZm Z d dl
m Z e � e
�Zddd�Zdd d
�Zd d � Zd
d� Zdd� ZdS )� N)�walk)�abspath�isfile�join)�Popen�PIPE)�
PUBLIC_DIR_RE�z.pyc c s� t |�}t| tt f�r6| D ]}t|�D ]
}|V q&qnRt| �rP| �|�rP| V n8tt| ��D ]*\}}}|D ]}|�|�rjt||�V qjq\dS )z6Generate *.py file names available in given directory.N) �tuple�
isinstance�list�from_directoryr �endswithr r r )�dname�
extensions�item�fn�root�dirsZ
file_names� r �%/usr/share/python3/debpython/files.pyr
s
r
c c sh t |�}td| dtd�}|�� \}}|jdkr<td| ��t|d�}|�� D ]}|�|�rN|V qNdS )z4Generate *.py file names available in given package.z/usr/bin/dpkg -L %sT)�shell�stdoutr zcannot get content of %szutf-8N) r
r r Z communicate�
returncode� Exception�str�
splitlinesr )Z package_namer Zprocessr �stderr�liner r r � from_package. s �
r c c s | D ]}|� |�r|V qdS )z4Generate *.py file names that match given directory.N)�
startswith)�filesr r r r r �filter_directory< s
r" c c s` t dd� |D ��}t dd� |D ��}| D ]2}t�|�}|r(|�d�}||ksT||kr(|V q(dS )z3Generate *.py file names that match given versions.c s s | ]}d | V qdS )z%d.%dNr ��.0�ir r r � <genexpr>E s z filter_public.<locals>.<genexpr>c s s | ]}t |d �V qdS )r N)r r# r r r r& F s � N)�setr �match�group)r! ZversionsZvstrZ shared_vstrr Z
public_dirZversr r r �
filter_publicC s
r+ c c s& t |�}| D ]}|�|�s |V q dS )z<Removes files with matching extensions from given generator.N)r
r )r! r r r r r �filter_out_extO s
r, )r )r )Zlogging�osr �os.pathr r r �
subprocessr r Z debpythonr Z getLogger�__name__�logr
r r" r+ r, r r r r �<module> s
|