Your IP : 216.73.217.13
U
bL% � @ s< d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZe� � ddddddd�Z
d Z G d
d � d �Z d d
� Z
G dd� d�Zedk�r8ejejd� eej�dk r�edejd � edejd � edejd � e�d� eejdd� �Ze�ejd � ej�� D ] \ZZe� deef � �qdS )� N�x �d �P �< �( � )�mainZuniverseZcontribZ
restrictedznon-freeZ
multiversea�
CREATE TABLE IF NOT EXISTS "commands"
(
[cmdID] INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
[pkgID] INTEGER NOT NULL,
[command] TEXT,
FOREIGN KEY ([pkgID]) REFERENCES "pkgs" ([pkgID])
);
CREATE TABLE IF NOT EXISTS "packages"
(
[pkgID] INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
[name] TEXT,
[version] TEXT,
[component] TEXT,
[priority] INTEGER
);
CREATE INDEX IF NOT EXISTS idx_commands_command ON commands (command);
CREATE INDEX IF NOT EXISTS idx_packages_name ON packages (name);
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�measurec C s || _ || _d S �N)�what�stats)�selfr r � r �</usr/lib/python3/dist-packages/CommandNotFound/db/creator.py�__init__= s zmeasure.__init__c C s t � � | _d S r
)�time�now)r
r r r � __enter__@ s zmeasure.__enter__c G s: | j | jkrd| j| j < | j| j t�� | j 7 < d S )Nr )r r r r )r
�argsr r r �__exit__B s zmeasure.__exit__N)�__name__�
__module__� __qualname__r r r r r r r r < s r c
C sD zt �| � W n0 tk
r> } z|jtjkr.� W 5 d }~X Y nX d S r
)�os�remove�OSError�errnoZENOENT)�path�er r r �rm_fH s
r c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )� DbCreatorc C s( || _ t�� d | _dt�� d�| _d S )Nr )�total�
total_time)�files�apt_pkgZget_architectures� primary_archr r )r
r# r r r r Q s zDbCreator.__init__c C s� |d }| � |�s"t�d|� d S |d }t�|��l}z|�t� | �|� W nL tjk
r� } z,t |�dkr�t�
d|� t � d� n|�W 5 d }~X Y nX W 5 Q R X t
|� t�||� t|d��}t�| �� |� W 5 Q R X d S )Nz .metadataz0%s does not require an update (inputs unchanged)z.tmpzdatabase is lockedz*%s is locked by another process. Ignoring.r �w)�_db_update_needed�logging�info�sqlite3ZconnectZ
executescript�
create_db_sql�_fill_commandsZOperationalError�str�warning�sys�exitr r �rename�open�json�dump�_calc_input_metadata)r
Zdbname�
metadata_fileZtmpdb�conr �fpr r r �createU s0
�
� zDbCreator.createc
C sx t j�|�sdS z,t|��}t�|�}W 5 Q R X || �� kW S tk
rr } zt� d||� W Y �dS d }~X Y nX d S )NTzcannot read %s: %s)
r r �existsr2 r3 �loadr5 � Exceptionr( r. )r
r6 r8 �metar r r r r' q s
zDbCreator._db_update_neededc C s@ i }| j D ]0}t�|�}|j|j|j|j|j|jd�||< q
|S )N)�st_ino�st_dev�st_uid�st_gid�st_size�st_mtime) r# r �statr> r? r@ rA rB rC )r
r= �p�str r r r5 { s
� z DbCreator._calc_input_metadatac
C s� | j D ]V}tjdd|gtjd��6}| �||j� |�� dkrRtj|jd� |�d��W 5 Q R X qt
�
� | j d | j d< t �
d| j d | j d f � d S )
Nz/usr/lib/apt/apt-helperzcat-file)�stdoutr z#/usr/lib/apt/apt-helper cat-file {})�
returncode�cmdr" z processed %i packages in %.2fsr! )r# �
subprocess�Popen�PIPE�_parse_single_commands_filerG �waitZCalledProcessErrorrH �formatr r r( r) )r
r7 �f�subr r r r, � s
� �zDbCreator._fill_commandsc C s |� d||f��� }|S )Nz�
SELECT packages.pkgID, name, version
FROM commands
INNER JOIN packages on packages.pkgID = commands.pkgID
WHERE commands.command=? AND packages.name=?;
��executeZfetchone)r
r7 �command�pkgname�
already_in_dbr r r �_in_db� s
�zDbCreator._in_dbc C s |� d|f� |� d|f� d S )Nz"DELETE FROM packages WHERE pkgID=?z"DELETE FROM commands WHERE pkgID=?�rS )r
r7 Zpkgidr r r �
_delete_pkgid� s zDbCreator._delete_pkgidc C s" |� d|f��� }|r |d S d S )Nz'SELECT pkgID from packages WHERE name=?r rR )r
r7 rU Zhave_pkgr r r �
_get_pkgid� s �zDbCreator._get_pkgidc C s |� d||||f�}|jS )Nzt
INSERT INTO packages (name, version, component, priority)
VALUES (?, ?, ?, ?);
)rS Z lastrowid)r
r7 rU �version� component�priorityZcurr r r �_insert_package� s
�zDbCreator._insert_packagec C s |� d||f� d S )NzF
INSERT INTO commands (command, pkgID) VALUES (?, ?);
rX )r
r7 rT �pkg_idr r r �_insert_command� s �zDbCreator._insert_commandc
C s2 t �|�}|�� sd S |jd }|�d�r.d S |jd }|jd }|dkrX|| jkrXd S |�� �r.| jd d7 < |jd }|j�d �r�|jd }|j�d
d �}t� } |j�d d �r�t|j�d d �� d
��} |jd � d
�D �]J}
|
| kr�q�t
d| j�� | � ||
|�} W 5 Q R X | �r|t � || d �dk�rht
�d|
� t
d| j�� | �|| d � W 5 Q R X nt
�d|
|f � q�t
�d|
|||f � t
d| j�� | �||�} W 5 Q R X | �st| }
|
t|j�dd��7 }
t
d| j�� | �|||||
�} W 5 Q R X t
d| j�� | �||
| � W 5 Q R X q�qXd S )N�suitez
-backportsr\ �arch�allr! � �namezvisible-pkgnamer[ � zignore-commands�,ZcommandsZsql_already_db� r z+replacing exiting %s in DB (higher version)Zsql_delete_already_in_dbz(skipping %s from %s (lower/same version)zadding %s from %s/%s (%s)Z sql_have_pkgzpriority-bonus�0Zsql_insert_pkgZsql_insert_cmd)r$ ZTagFile�stepZsection�endswithr% r �get�set�splitr rW Zversion_comparer( �debugrY rZ �component_priorities�intr^ r` )r
r7 r8 Ztagfra r\ rb rU r[ Zignore_commandsrT rV r_ r] r r r rM � s\
� z%DbCreator._parse_single_commands_fileN)r r r r r9 r' r5 r, rW rY rZ r^ r` rM r r r r r P s
r �__main__)�level� z%usage: %s <output-db-path> <files...>z. e.g.: %s commands.db ./dists/*/*/*/Commands-*zP e.g.: %s /var/lib/command-not-found/commands.db /var/lib/apt/lists/*Commands-*rd rh z%s: %s) r r3 r( r r* rJ r/ r r$ Zinitrp r+ r r r r Z basicConfig�INFO�len�argv�printr0 �colr9 r �itemsrD Zamountro r r r r �<module> s@ �