/usr/lib/python3/dist-packages/landscape/lib/apt/package/__pycache__
Edit: /usr/lib/python3/dist-packages/landscape/lib/apt/package/__pycache__/store.cpython-310.pyc (20923B)
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 eZG d
d deZG dd deZG dd deZG dd deZdd Zdd Zdd ZdS )zEProvide access to the persistent data used by L{PackageTaskHandler}s. N)dbapi2) iteritemslong)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) _filenameselffilenamer r r __init__! s
zHashIdStore.__init__c C s t | j d S r )ensure_hash_id_schema_dbr 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 executesqlite3Binary)r cursorhash_idshashidr r r set_hash_ids' s
zHashIdStore.set_hash_idsc C s, | dt|f | }|r|d S dS )zReturn 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.0rowr r r
B s z,HashIdStore.get_hash_ids..r fetchallr r r r r get_hash_ids>
zHashIdStore.get_hash_idsc C s< t |ttfs 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)
isinstanceintr 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 hashNr r4 r r r clear_hash_idsN zHashIdStore.clear_hash_idsc C s. z | dd W dS tjy t| jw )zCheck 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 fddZ fddZdd 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
dd? Z#e
d@dA Z$e
dBdC Z%e
dDdE Z&e
dFdG Z'e
dHdI Z(e
dMdKdLZ) 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_storeer r r add_hash_id_dbu s zPackageStore.add_hash_id_dbc C s t | jdkS )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 |tsJ | 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 z0PackageStore.remove_available..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
z.PackageStore.get_available..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..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..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..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..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..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..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..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..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..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..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 |}|dtt|t f t| j|j S )Nz.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|jS )Nz8INSERT INTO task (queue, timestamp, data) VALUES (?,?,?)) r r r r r r PackageTaskr r )r r queuedatar 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" | dddd |D d S )Nz%DELETE FROM task WHERE id NOT IN (%s)rW c S s g | ]}t |jqS r )rZ r# )r/ taskr r r rh T s z,PackageStore.clear_tasks..)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 fddZedd 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 | dtt|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..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 | ]}|fqS r r )r/
message_idr r r rh n s z5FakePackageStore.save_message_ids..) 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 fqS r* r, r. r r r rh v s z8FakePackageStore.get_messages_by_ids..)r_ rM r tupler3 )r r r paramsresultr 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 ZeeeZ 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 tt| 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 s* J w", %