| Current File : //snap/core22/2115/usr/share/subiquity/subiquitycore/__pycache__/context.cpython-310.pyc |
o
K��^� � @ sR d dl Z d dlZd dlZd dlZG dd� dej�Zd aG dd� d�Zd dd�ZdS )
� Nc @ s$ e Zd Ze�� Ze�� Ze�� ZdS )�StatusN)�__name__�
__module__� __qualname__�enum�auto�SUCCESS�FAIL�WARN� r r �-/usr/share/subiquity/subiquitycore/context.pyr s r c @ st e Zd ZdZddd�Zedd� �Zddd �Zd
d � Zdd d
�Z de
j fdd�Z dd� Z
dd� Zdd� Zddd�ZdS )�Contexta� Class to report when things start and finish.
The expected way to use this is something like:
with somecontext.child("operation"):
await long_running_operation()
but you can also call .enter() and .exit() if use as a context
manager isn't possible.
start and finish events are reported via the report_start_event and
report_finish_event methods on app.
You can override the message shown on exit by passing it to the .exit
method or by assigning to description:
with somecontext.child("operation") as context:
result = await long_running_operation()
context.description = "result was {}".format(result)
Nc C sH t | _t d7 a || _|| _|| _|| _|| _|d u r |}|| _i | _d S )N� ) �
context_id�id�app�name� description�parent�level�
childlevel�data)�selfr r r r r r r r r �__init__4 s
zContext.__init__c C s | ||j dd d�S )N� �INFO)�project)�clsr r r r �newB s z Context.newr c C s& |d u r| j }t| �| j||| ||�S �N)r �typer )r r r r r r r r �childF s z
Context.childc C s8 | }g }|d ur|� |j� |j}|d usd�t|��S )N�/)�appendr r �join�reversed)r �c�namesr r r � full_nameK s �zContext.full_namec C s |d u r| j }| j�| |� d S r )r r �report_start_event)r r r r r �enterS s z
Context.enterc C s" |d u r| j }| j�| ||� d S r )r r �report_finish_event)r r �resultr r r �exitX s z Context.exitc C s | � � | S r )r* )r r r r � __enter__] s zContext.__enter__c C sD |d urt j}t|tj�rd}n
t|�}nt j}d }| �||� d S )N� cancelled)r r �
isinstance�asyncio�CancelledError�strr r- )r �exc�value�tbr, r r r r �__exit__a s
zContext.__exit__c C s || j |< d S r )r )r �keyr5 r r r �setm s z Context.setc C s2 | }|d ur||j v r|j | S |j}|d us|S r )r r )r r8 �defaultr& r r r �getp s
�z Context.getr )r NN)r r r �__doc__r � classmethodr r! r( r* r r r- r. r7 r9 r; r r r r r
s
r
r c s � ��fdd�}|S )Nc s` �d u r�j ����fdd�� t���� �fdd��}t���� �fdd��}t���r.|S |S )Nc sR |� d�}|d u r | j}|jd�jdi |���jdd| i|��d�� ��|d< |S )N�contextr )r r r )r; r> r! �format)r �kwr> ��
context_kwr r r r �convargs s
�
�z0with_context.<locals>.decorate.<locals>.convargsc sF � | |�}|d � �| fi |��W d � S 1 s w Y d S �Nr> r �r r@ �rC �methr r �decorated_sync� s
$�z6with_context.<locals>.decorate.<locals>.decorated_syncc � sN �� | |�}|d � �| fi |��I d H W d � S 1 s w Y d S rD r rE rF r r �decorated_async� s
�
$�z7with_context.<locals>.decorate.<locals>.decorated_async)r � functools�wraps�inspect�iscoroutinefunction)rG rH rI rA rF r �decoratez s
z with_context.<locals>.decorater )r r rB rN r rA r � with_contexty s rO )Nr ) r1 r rJ rL �Enumr r r
rO r r r r �<module> s [