Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
lib
/
python3
/
dist-packages
/
awscli
/
customizations
/
history
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: 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,
Save
Cancel