OwlCyberSecurity - MANAGER
Edit File: creator.cpython-38.pyc
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@ �