| Server IP : 209.38.156.173 / Your IP : 216.73.216.122 [ Web Server : Apache/2.4.52 (Ubuntu) System : Linux lakekumayuhotel 5.15.0-136-generic #147-Ubuntu SMP Sat Mar 15 15:53:30 UTC 2025 x86_64 User : root ( 0) PHP Version : 8.1.2-1ubuntu2.22 Disable Function : NONE Domains : 2 Domains MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/landscape/lib/apt/package/__pycache__/ |
Upload File : |
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", %