File manager - Edit - /usr/lib/python3/dist-packages/landscape/lib/apt/package/__pycache__/store.cpython-310.pyc
Back
o ���c^B � @ s� d Z ddlZzddlZW n ey ddlmZ Y nw ddlmZmZ ddl m Z ddl m Z G dd� de �ZG d d � d e �ZG d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de�Zdd� Zdd� Zdd� ZdS )zEProvide access to the persistent data used by L{PackageTaskHandler}s.� N)�dbapi2)� iteritems�long)�bpickle)� with_cursorc @ � e Zd ZdZdS )�UnknownHashIDRequestz$Raised for unknown hash id requests.N��__name__� __module__� __qualname__�__doc__� r r �A/usr/lib/python3/dist-packages/landscape/lib/apt/package/store.pyr � r c @ r )�InvalidHashIdDbzARaised when trying to add an invalid hash=>id lookaside database.Nr r r r r r r r c @ sl e Zd ZdZdZdd� Zdd� Zedd� �Zed d � �Z ed d � �Z ed d� �Z edd� �Z edd� �Z dS )� HashIdStorea C{HashIdStore} stores package hash=>id mappings in a file. The file is a SQLite database that contains a single table called "hash". The table schema is defined in L{ensure_hash_id_schema}. @param filename: The file where the mappings are persisted to. Nc C s || _ d S �N)� _filename��self�filenamer r r �__init__! s zHashIdStore.__init__c C s t | j� d S r )�ensure_hash_id_schema�_db�r r r r �_ensure_schema$ s zHashIdStore._ensure_schemac C s, t |�D ]\}}|�d|t�|�f� qdS )zbSet the ids of a set of hashes. @param hash_ids: a C{dict} of hash=>id mappings. z REPLACE INTO hash VALUES (?, ?)N)r �execute�sqlite3�Binary)r �cursor�hash_ids�hash�idr r r � set_hash_ids' s ��zHashIdStore.set_hash_idsc C s, |� dt�|�f� |�� }|r|d S dS )z�Return the id associated to C{hash}, or C{None} if not available. @param hash: a C{bytes} representing a hash. � SELECT id FROM hash WHERE hash=?r N)r r r �fetchone)r r r" �valuer r r � get_hash_id1 s �zHashIdStore.get_hash_idc C � |� d� dd� |�� D �S )z=Return a C{dict} holding all the available hash=>id mappings.zSELECT hash, id FROM hashc S s i | ] }t |d �|d �qS �r � ��bytes��.0�rowr r r � <dictcomp>B s z,HashIdStore.get_hash_ids.<locals>.<dictcomp>�r �fetchall�r r r r r � get_hash_ids> � zHashIdStore.get_hash_idsc C s< t |ttf�s J �|�d|f� |�� }|r t|d �S dS )zAReturn the hash associated to C{id}, or C{None} if not available.z SELECT hash FROM hash WHERE id=?r N)� isinstance�intr r r& r- )r r r# r' r r r � get_id_hashD s zHashIdStore.get_id_hashc C � |� d� dS )z Delete all hash=>id mappings.zDELETE FROM hashN�r r4 r r r �clear_hash_idsN � zHashIdStore.clear_hash_idsc C s. z |� dd� W dS tjy t| j��w )z�Check database integrity. @raise: L{InvalidHashIdDb} if the filenme passed to the constructor is not a SQLite database or does not have a table called "hash" with a compatible schema. r% )� N)r r � DatabaseErrorr r r4 r r r � check_sanityS s �zHashIdStore.check_sanity)r r r r r r r r r$ r( r5 r9 r< r@ r r r r r s" r c s� e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd � Zd d � Zd d � Z e dd� �Z e dd� �Z e dd� �Z e dd� �Ze dd� �Ze dd� �Ze dd� �Ze d d � �Ze d d � �Ze d d!� �Ze d"d#� �Ze d$d%� �Ze d&d'� �Ze d(d)� �Ze d*d+� �Ze d,d-� �Ze d.d/� �Ze d0d1� �Z e d2d3� �Z e d4d5� �Z e d6d7� �Z e d8d9� �Z e d:d;� �Z!e d<d=� �Z"e d>d?� �Z#e d@dA� �Z$e dBdC� �Z%e dDdE� �Z&e dFdG� �Z'e dHdI� �Z(e dMdKdL��Z)� Z*S )N� PackageStorea� Persist data about system packages and L{PackageTaskHandler}'s tasks. This class extends L{HashIdStore} by adding tables to the SQLite database backend for storing information about the status of the system packages and about the tasks to be performed by L{PackageTaskHandler}s. The additional tables and schemas are defined in L{ensure_package_schema}. @param filename: The file where data is persisted to. c s t t| ��|� g | _d S r )�superrA r �_hash_id_storesr �� __class__r r r m s zPackageStore.__init__c � t t| ��� t| j� d S r )rB rA r �ensure_package_schemar r rD r r r q � zPackageStore._ensure_schemac C sB t |�}z|�� W n ty } z|�d}~ww | j�|� dS )a Attach an additional "lookaside" hash=>id database. This method can be called more than once to attach several hash=>id databases, which will be queried *before* the main database, in the same the order they were added. If C{filename} is not a SQLite database or does not have a table called "hash" with a compatible schema, L{InvalidHashIdDb} is raised. @param filename: a secondary SQLite databases to look for pre-canned hash=>id mappings. N)r r@ r rC �append)r r � hash_id_store�er r r �add_hash_id_dbu s ��zPackageStore.add_hash_id_dbc C s t | j�dkS )z?Return C{True} if one or more lookaside databases are attached.r )�lenrC r r r r �has_hash_id_db� � zPackageStore.has_hash_id_dbc C s< t |t�sJ �| jD ] }|�|�}|r| S q t�| |�S )a Return the id associated to C{hash}, or C{None} if not available. This method composes the L{HashIdStore.get_hash_id} methods of all the attached lookaside databases, falling back to the main one, as described in L{add_hash_id_db}. )r7 r- rC r( r )r r" �storer# r r r r( � s � zPackageStore.get_hash_idc C s2 | j D ]}|�|�}|dur| S qt�| |�S )a' Return the hash associated to C{id}, or C{None} if not available. This method composes the L{HashIdStore.get_id_hash} methods of all the attached lookaside databases, falling back to the main one in case the hash associated to C{id} is not found in any of them. N)rC r9 r )r r# rP r" r r r r9 � s � zPackageStore.get_id_hashc C � |D ] }|� d|f� qd S )Nz!REPLACE INTO available VALUES (?)r; �r r �idsr# r r r � add_available� � �zPackageStore.add_availablec C �& d� dd� |D ��}|�d| � d S )N�,c s � � | ] }t t|��V qd S r ��strr8 �r/ r# r r r � <genexpr>� � � z0PackageStore.remove_available.<locals>.<genexpr>z&DELETE FROM available WHERE id IN (%s)��joinr �r r rS �id_listr r r �remove_available� � z PackageStore.remove_availablec C � |� d� d S )NzDELETE FROM availabler; r4 r r r �clear_available� rO z PackageStore.clear_availablec C r) )NzSELECT id FROM availablec S � g | ]}|d �qS �r r r. r r r � <listcomp>� � z.PackageStore.get_available.<locals>.<listcomp>r2 r4 r r r � get_available� � zPackageStore.get_availablec C rQ )Nz)REPLACE INTO available_upgrade VALUES (?)r; rR r r r �add_available_upgrades� rU z#PackageStore.add_available_upgradesc C rV )NrW c s rX r rY r[ r r r r\ � r] z9PackageStore.remove_available_upgrades.<locals>.<genexpr>z.DELETE FROM available_upgrade WHERE id IN (%s)r^ r` r r r �remove_available_upgrades� s �z&PackageStore.remove_available_upgradesc C rd )Nz DELETE FROM available_upgrader; r4 r r r �clear_available_upgrades� rO z%PackageStore.clear_available_upgradesc C r) )Nz SELECT id FROM available_upgradec S rf rg r r. r r r rh � ri z7PackageStore.get_available_upgrades.<locals>.<listcomp>r2 r4 r r r �get_available_upgrades� rk z#PackageStore.get_available_upgradesc C rQ )Nz%REPLACE INTO autoremovable VALUES (?)r; rR r r r �add_autoremovable� rU z PackageStore.add_autoremovablec C rV )NrW c s rX r rY r[ r r r r\ � r] z4PackageStore.remove_autoremovable.<locals>.<genexpr>z*DELETE FROM autoremovable WHERE id IN (%s)r^ r` r r r �remove_autoremovable� rc z!PackageStore.remove_autoremovablec C rd )NzDELETE FROM autoremovabler; r4 r r r �clear_autoremovable� rO z PackageStore.clear_autoremovablec C r) )Nz SELECT id FROM autoremovablec S rf rg r r. r r r rh � ri z2PackageStore.get_autoremovable.<locals>.<listcomp>r2 r4 r r r �get_autoremovable� rk z PackageStore.get_autoremovablec C rQ )Nz REPLACE INTO security VALUES (?)r; rR r r r � add_security� rU zPackageStore.add_securityc C rV )NrW c s rX r rY r[ r r r r\ � r] z/PackageStore.remove_security.<locals>.<genexpr>z%DELETE FROM security WHERE id IN (%s)r^ r` r r r �remove_security� rc z PackageStore.remove_securityc C rd )NzDELETE FROM securityr; r4 r r r �clear_security� rO zPackageStore.clear_securityc C r) )NzSELECT id FROM securityc S rf rg r r. r r r rh � ri z-PackageStore.get_security.<locals>.<listcomp>r2 r4 r r r � get_security� rk zPackageStore.get_securityc C rQ )Nz!REPLACE INTO installed VALUES (?)r; rR r r r � add_installed� rU zPackageStore.add_installedc C rV )NrW c s rX r rY r[ r r r r\ r] z0PackageStore.remove_installed.<locals>.<genexpr>z&DELETE FROM installed WHERE id IN (%s)r^ r` r r r �remove_installed rc z PackageStore.remove_installedc C rd )NzDELETE FROM installedr; r4 r r r �clear_installed rO z PackageStore.clear_installedc C r) )NzSELECT id FROM installedc S rf rg r r. r r r rh ri z.PackageStore.get_installed.<locals>.<listcomp>r2 r4 r r r � get_installed rk zPackageStore.get_installedc C r) )z+Get the package ids of all locked packages.zSELECT id FROM lockedc S rf rg r r. r r r rh ri z+PackageStore.get_locked.<locals>.<listcomp>r2 r4 r r r � get_locked r6 zPackageStore.get_lockedc C s |D ] }|� d|f� qdS )z9Add the given package ids to the list of locked packages.z REPLACE INTO locked VALUES (?)Nr; rR r r r � add_locked s �zPackageStore.add_lockedc C rV )NrW c s rX r rY r[ r r r r\ r] z-PackageStore.remove_locked.<locals>.<genexpr>z#DELETE FROM locked WHERE id IN (%s)r^ r` r r r � remove_locked rc zPackageStore.remove_lockedc C r: )z/Remove all the package ids in the locked table.zDELETE FROM lockedNr; r4 r r r � clear_locked" r= zPackageStore.clear_lockedc C s6 t |�}|�dt�t�|��t�� f� t| j|j �S )Nz<INSERT INTO hash_id_request (hashes, timestamp) VALUES (?,?)) �listr r r r �dumps�time� HashIDRequestr � lastrowid)r r �hashesr r r �add_hash_id_request' s �z PackageStore.add_hash_id_requestc C s* |� d|f� |�� st|��t| j|�S )Nz(SELECT 1 FROM hash_id_request WHERE id=?)r r& r r� r )r r � request_idr r r �get_hash_id_request/ s � z PackageStore.get_hash_id_requestc s |� d� � fdd�|�� D �S )Nz SELECT id FROM hash_id_requestc s g | ] }t � j|d ��qS rg )r� r r. r r r rh : s z6PackageStore.iter_hash_id_requests.<locals>.<listcomp>r2 r4 r r r �iter_hash_id_requests7 s z"PackageStore.iter_hash_id_requestsc C rd )NzDELETE FROM hash_id_requestr; r4 r r r �clear_hash_id_requests< rO z#PackageStore.clear_hash_id_requestsc C s4 t �|�}|�d|t�� t�|�f� t| j|j�S )Nz8INSERT INTO task (queue, timestamp, data) VALUES (?,?,?)) r r� r r� r r � PackageTaskr r� )r r �queue�datar r r �add_task@ s �zPackageStore.add_taskc C s. |� d|f� |�� }|rt| j|d �S d S )Nz4SELECT id FROM task WHERE queue=? ORDER BY timestampr )r r& r� r )r r r� r0 r r r � get_next_taskH s �zPackageStore.get_next_taskr c C s" |� dd�dd� |D �� � d S )Nz%DELETE FROM task WHERE id NOT IN (%s)rW c S s g | ]}t |j��qS r )rZ r# )r/ �taskr r r rh T s z,PackageStore.clear_tasks.<locals>.<listcomp>)r r_ )r r � except_tasksr r r � clear_tasksQ s �zPackageStore.clear_tasks)r )+r r r r r r rL rN r( r9 r rT rb re rj rl rm rn ro rp rq rr rs rt ru rv rw rx ry rz r{ r| r} r~ r r� r� r� r� r� r� r� � __classcell__r r rD r rA a s� rA c sP e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Zed d � �Z � Z S ) �FakePackageStorezT A L{PackageStore} with an additional message table to store sent messages. c rF r )rB r� r �ensure_fake_package_schemar r rD r r r \ rH z FakePackageStore._ensure_schemac C s |� dt�t�|��f� d S )Nz%INSERT INTO message (data) VALUES (?))r r r r r� )r r �messager r r � save_message` s �z FakePackageStore.save_messagec C s dd� |� d��� D �S )Nc S rf rg r r. r r r rh g ri z4FakePackageStore.get_message_ids.<locals>.<listcomp>zSELECT id FROM messager2 r4 r r r �get_message_idse s �z FakePackageStore.get_message_idsc C s |� ddd� |D �� d S )Nz#INSERT INTO message (id) VALUES (?)c S s g | ]}|f�qS r r )r/ � message_idr r r rh n s z5FakePackageStore.save_message_ids.<locals>.<listcomp>)� executemany)r r � message_idsr r r �save_message_idsj s �z!FakePackageStore.save_message_idsc C s: d� dgt|� �}|�d| t|���� }dd� |D �S )Nz, �?z9SELECT id, data FROM message WHERE id IN (%s) ORDER BY idc S s g | ] }|d t |d �f�qS r* r, r. r r r rh v s z8FakePackageStore.get_messages_by_ids.<locals>.<listcomp>)r_ rM r �tupler3 )r r r� �params�resultr r r �get_messages_by_idsp s ���z$FakePackageStore.get_messages_by_ids) r r r r r r r� r� r� r� r� r r rD r r� W s r� c @ st e Zd Zdd� Zeedd� ��Zedd� �Zedd� �Zeee�Z ed d � �Z ed d � �Z ee e �Z ed d� �Z dS )r� c C s || _ || _d S r )r r# )r �dbr# r r r r { s zHashIDRequest.__init__c C s& |� d| jf� t�t|�� d ��S )Nz-SELECT hashes FROM hash_id_request WHERE id=?r )r r# r �loadsr- r& r4 r r r r� s �zHashIDRequest.hashesc C � |� d| jf� |�� d S )Nz0SELECT timestamp FROM hash_id_request WHERE id=?r �r r# r& r4 r r r �_get_timestamp� � � z HashIDRequest._get_timestampc C � |� d|| jf� d S )Nz1UPDATE hash_id_request SET timestamp=? WHERE id=?�r r# �r r r' r r r �_set_timestamp� � �z HashIDRequest._set_timestampc C r� )Nz1SELECT message_id FROM hash_id_request WHERE id=?r r� r4 r r r �_get_message_id� r� z HashIDRequest._get_message_idc C r� )Nz2UPDATE hash_id_request SET message_id=? WHERE id=?r� r� r r r �_set_message_id� r� z HashIDRequest._set_message_idc C � |� d| jf� d S )Nz&DELETE FROM hash_id_request WHERE id=?r� r4 r r r �remove� � zHashIDRequest.removeN)r r r r �propertyr r� r� r� � timestampr� r� r� r� r r r r r� y s" r� c @ s e Zd Zdd� Zedd� �ZdS )r� c C sn || _ || _|�� }z|�d|f� |�� }W |�� n|�� w |d | _|d | _t� t |d ��| _ d S )Nz2SELECT queue, timestamp, data FROM task WHERE id=?r r+ � ) r r# r r r& �closer� r� r r� r- r� )r r� r# r r0 r r r r � s � zPackageTask.__init__c C r� )NzDELETE FROM task WHERE id=?r� r4 r r r r� � r� zPackageTask.removeN)r r r r r r� r r r r r� � s r� c C sX | � � }z|�d� W n tjtjfy! |�� | �� Y dS w |�� | �� dS )zeCreate all tables needed by a L{HashIdStore}. @param db: A connection to a SQLite database. z<CREATE TABLE hash (id INTEGER PRIMARY KEY, hash BLOB UNIQUE)N�r r r �OperationalErrorr? r� �rollback�commit�r� r r r r r � s � r c C s� | � � }z*|�d� |�d� |�d� |�d� |�d� |�d� |�d� |�d� W n tjyA |�� | �� Y d S w |�� | �� d S ) zfCreate all tables needed by a L{PackageStore}. @param db: A connection to a SQLite database. z.CREATE TABLE security (id INTEGER PRIMARY KEY)z3CREATE TABLE autoremovable (id INTEGER PRIMARY KEY)z,CREATE TABLE locked (id INTEGER PRIMARY KEY)z/CREATE TABLE available (id INTEGER PRIMARY KEY)z7CREATE TABLE available_upgrade (id INTEGER PRIMARY KEY)z/CREATE TABLE installed (id INTEGER PRIMARY KEY)zkCREATE TABLE hash_id_request (id INTEGER PRIMARY KEY, timestamp TIMESTAMP, message_id INTEGER, hashes BLOB)zVCREATE TABLE task (id INTEGER PRIMARY KEY, queue TEXT, timestamp TIMESTAMP, data BLOB)N)r r r r� r� r� r� r� r r r rG � s � rG c C sX | � � }z|�d� W n tjtjfy! |�� | �� Y d S w |�� | �� d S )Nz8CREATE TABLE message (id INTEGER PRIMARY KEY, data BLOB)r� r� r r r r� � s � r� )r r� r � ImportError� pysqlite2r �twisted.python.compatr r � landscape.libr �landscape.lib.storer � Exceptionr r �objectr rA r� r� r� r rG r� r r r r �<module> s* � J w", %
| ver. 1.4 |
Github
|
.
| PHP 8.2.28 | Generation time: 0.02 |
proxy
|
phpinfo
|
Settings