File: //usr/lib/python3/dist-packages/awscli/customizations/history/__pycache__/db.cpython-310.pyc
o
.&�a1$ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dlm
Z
d dlm Z e� e
�ZG dd� de�ZG dd � d ej�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de �ZdS )� N)�collections_abc)�BaseHistoryHandler)�sqlite3)� binary_typec @ s^ e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
e dd� �Z e j
dd� �Z dS )�DatabaseConnectionz�
CREATE TABLE IF NOT EXISTS records (
id TEXT,
request_id TEXT,
source TEXT,
event_type TEXT,
timestamp INTEGER,
payload TEXT
)zPRAGMA journal_mode=WALc C s t j|dd d�| _| �� d S )NF)�check_same_thread�isolation_level)r �connect� _connection�_ensure_database_setup)�self� db_filename� r �B/usr/lib/python3/dist-packages/awscli/customizations/history/db.py�__init__* s � zDatabaseConnection.__init__c C � | j �� d S �N�r
�close�r r r r r / � zDatabaseConnection.closec G s | j j|g|�R � S r )r
�execute)r �query�
parametersr r r r 2 s zDatabaseConnection.executec C s | � � | �� d S r )�_create_record_table�_try_to_enable_walr r r r r 5 s z)DatabaseConnection._ensure_database_setupc C s | � | j� d S r )r �
_CREATE_TABLEr r r r r 9 s z'DatabaseConnection._create_record_tablec C s4 z | � | j� W d S tjy t�d� Y d S w )Nz Failed to enable sqlite WAL.)r � _ENABLE_WALr �Error�LOG�debugr r r r r < s
�z%DatabaseConnection._try_to_enable_walc C s | j jS r �r
� row_factoryr r r r r" D s z DatabaseConnection.row_factoryc C s || j _d S r r! )r r" r r r r" H s N)�__name__�
__module__� __qualname__r r r r r r r r �propertyr" �setterr r r r r s
r c sD e Zd Zdd� Zdd� Zdd� Zdd� Z� fd d
�Zd d � Z� Z S )
�PayloadSerializerc C s t |�S r )�dict�r �objr r r �_encode_mutable_mappingN � z)PayloadSerializer._encode_mutable_mappingc C s |� � S r )� isoformatr* r r r �_encode_datetimeQ r- z"PayloadSerializer._encode_datetimec C s* z|� d�}W |S ty d}Y |S w )Nzutf-8z<Byte sequence>)�decode�UnicodeDecodeErrorr* r r r �_try_decode_bytesT s ��z#PayloadSerializer._try_decode_bytesc sd t |t�r � �|�}|S t |t�r t� fdd�|�� D ��}|S t |ttf�r0� fdd�|D �}|S )Nc 3 s" � | ] \}}|� � |�fV qd S r ��_remove_non_unicode_stings)�.0�k�vr r r � <genexpr>_ s � z?PayloadSerializer._remove_non_unicode_stings.<locals>.<genexpr>c s g | ]}� � |��qS r r3 )r5 �or r r �
<listcomp>b s z@PayloadSerializer._remove_non_unicode_stings.<locals>.<listcomp>)�
isinstance�strr2 r) �items�list�tupler* r r r r4 [ s
� ��z,PayloadSerializer._remove_non_unicode_stingsc s@ z t t| ��|�W S ty | �|�}t t| ��|� Y S w r )�superr( �encoder1 r4 )r r+ � scrubbed_obj�� __class__r r rA e s
�zPayloadSerializer.encodec C sH t |tj�r | �|�S t |tj�r| �|�S t |t�r | �|�S t|�S r ) r; �datetimer/ r �MutableMappingr, r r2 �reprr* r r r �defaultw s
zPayloadSerializer.default)
r# r$ r% r, r/ r2 r4 rA rH �
__classcell__r r rC r r( M s
r( c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �DatabaseRecordWriterz
INSERT INTO records(
id, request_id, source, event_type, timestamp, payload)
VALUES (?,?,?,?,?,?) c C s || _ t�� | _d S r )r
� threading�Lock�_lock�r �
connectionr r r r � � z DatabaseRecordWriter.__init__c C r r r r r r r r � r zDatabaseRecordWriter.closec C sF | � |�}| j� | j�| j|� W d � d S 1 s w Y d S r )�_create_db_recordrM r
r �
_WRITE_RECORD)r �record� db_recordr r r � write_record� s
"�z!DatabaseRecordWriter.write_recordc C s@ |d }t j|d td�}|d |�d�|d ||d |f}|S )N�
event_type�payload)�cls�
command_id�
request_id�source� timestamp)�json�dumpsr( �get)r rS rV �json_serialized_payloadrT r r r rQ � s
��z&DatabaseRecordWriter._create_db_recordN)r# r$ r% rR r r rU rQ r r r r rJ � s rJ c @ sX e Zd ZdZde Zde Zde Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dS )�DatabaseRecordReaderzORDER BY timestampz�
SELECT * FROM records
WHERE id =
(SELECT id FROM records WHERE timestamp =
(SELECT max(timestamp) FROM records)) %s;z%SELECT * from records where id = ? %sz�SELECT a.id AS id_a, b.id AS id_b, a.timestamp as timestamp, a.payload AS args, b.payload AS rc FROM records a, records b where a.event_type == "CLI_ARGUMENTS" AND b.event_type = "CLI_RC" AND id_a == id_b %s DESCc C s || _ | j| j _d S r )r
� _row_factoryr" rN r r r r � rP z DatabaseRecordReader.__init__c C r r r r r r r r � r zDatabaseRecordReader.closec C sF i }t |j�D ]\}}|| }|d dkrt�|�}|||d < q|S )Nr rW )� enumerate� descriptionr] �loads)r �cursor�row�d�idx�col�valr r r rb � s
z!DatabaseRecordReader._row_factoryc c �$ � | j �| j�}|D ]}|V q
d S r )r
r �_GET_LAST_ID_RECORDS�r rf rg r r r �iter_latest_records� �
��z(DatabaseRecordReader.iter_latest_recordsc c s( � | j �| j|g�}|D ]}|V q d S r )r
r �_GET_RECORDS_BY_ID)r � record_idrf rg r r r � iter_records� s
��z!DatabaseRecordReader.iter_recordsc c rl r )r
r �_GET_ALL_RECORDSrn r r r �iter_all_records� rp z%DatabaseRecordReader.iter_all_recordsN)
r# r$ r% � _ORDERINGrm rq rt r r rb ro rs ru r r r r ra � s � ��
ra c @ sL e Zd Zeg d��ZdZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dS )�
RecordBuilder)�API_CALL� HTTP_REQUEST�
HTTP_RESPONSE�PARSED_RESPONSErx c C s d | _ t�� | _d S r )� _identifierrK �local�_localsr r r r r � rP zRecordBuilder.__init__c C s t | jdd �}|S �NrZ )�getattrr~ )r rZ r r r � _get_current_thread_request_id� s z,RecordBuilder._get_current_thread_request_idc C s t | jdtt�� �� d S r )�setattrr~ r< �uuid�uuid4r r r r �_start_http_lifecycle� s z#RecordBuilder._start_http_lifecyclec C s, || j kr | �� || jv r| �� }|S d S r )�!_START_OF_REQUEST_LIFECYCLE_EVENTr� �_REQUEST_LIFECYCLE_EVENTSr� )r rV rZ r r r �_get_request_id� s
z RecordBuilder._get_request_idc C s | j d u r tt�� �| _ | j S r )r| r< r� r� r r r r �_get_identifier� s
z RecordBuilder._get_identifierc C s> | � � }||||tt�� d �d�}| �|�}|r ||d< |S )Ni� )rY rV rW r[ r\ rZ )r� �int�timer� )r rV rW r[ �uidrS rZ r r r � build_record� s �
zRecordBuilder.build_recordN) r# r$ r% �setr� r� r r� r� r� r� r� r r r r rw � s � rw c @ s e Zd Zdd� Zdd� ZdS )�DatabaseHistoryHandlerc C s || _ || _d S r )�_writer�_record_builder)r �writer�record_builderr r r r
s
z DatabaseHistoryHandler.__init__c C s | j �|||�}| j�|� d S r )r� r� r� rU )r rV rW r[ rS r r r �emit s zDatabaseHistoryHandler.emitN)r# r$ r% r r� r r r r r� s r� )r� r� r] rE rK �logging�
awscli.compatr �botocore.historyr r r � getLoggerr# r �objectr � JSONEncoderr( rJ ra rw r� r r r r �<module> s"
/:!5,