U
K��^> � @ sN d dl Z d dlZd dlZd dlZG dd� dej�ZG dd� d�Zd dd�ZdS )
� Nc @ s$ e Zd Ze�� Ze�� Ze�� ZdS )�StatusN)�__name__�
__module__� __qualname__�enum�auto�SUCCESS�FAILZWARN� 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 s: || _ || _|| _|| _|| _|d kr*|}|| _i | _d S �N)�app�name� description�parent�level�
childlevel�data)�selfr r r r r r r
r
r �__init__1 s zContext.__init__c C s | ||j dd d�S )N� �INFO)Zproject)�clsr r
r
r �new< s z Context.newr c C s& |d kr| j }t| �| j||| ||�S r
)r �typer )r r r r r r
r
r �child@ s z
Context.childc C s2 | }g }|d k r$|� |j� |j}qd�t|��S )N�/)�appendr r �join�reversed)r �c�namesr
r
r � full_nameE s zContext.full_namec C s |d kr| j }| j�| |� d S r
)r r Zreport_start_event)r r r
r
r �enterM s z
Context.enterc C s" |d kr| j }| j�| ||� d S r
)r r Zreport_finish_event)r r �resultr
r
r �exitR s z Context.exitc C s | � � | S r
)r$ )r r
r
r � __enter__W s zContext.__enter__c C sD |d k r*t j}t|tj�r d}q4t|�}n
t j}d }| �||� d S )NZ cancelled)r r �
isinstance�asyncioZCancelledError�strr r& )r �exc�value�tbr% r r
r
r �__exit__[ s
zContext.__exit__c C s || j |< d S r
)r )r �keyr, r
r
r �setg s z Context.setc C s, | }|d k r(||j kr |j | S |j}q|S r
)r r )r r/ �defaultr! r
r
r �getj s
z Context.get)N)r NN)N)N)r r r �__doc__r � classmethodr r r# r$ r r r& r' r. r0 r2 r
r
r
r r s
r r c s � ��fdd�}|S )Nc sd �d kr�j ����fdd�� t���� �fdd��}t���� �fdd��}t���r\|S |S d S )Nc sF |� d�}|d kr| j}|jf �jf |��jf |�d�� ��|d< |S )N�context)r r )r2 r5 r �format)r �kwr5 ��
context_kwr r r
r �convargsy s
��
z0with_context.<locals>.decorate.<locals>.convargsc
s6 � | |�}|d � �| f|�W 5 Q R � S Q R X d S �Nr5 r
�r r7 �r: �methr
r �decorated_sync� s
z6with_context.<locals>.decorate.<locals>.decorated_syncc
� s< � | |�}|d � �| f|�I d H W 5 Q R � S Q R X d S r; r
r< r= r
r �decorated_async� s
z7with_context.<locals>.decorate.<locals>.decorated_async)r � functools�wraps�inspectZiscoroutinefunction)r> r? r@ r8 r= r �decoratet s
z with_context.<locals>.decorater
)r r r9 rD r
r8 r � with_contexts s rE )Nr )r) r rA rC �Enumr r rE r
r
r
r �<module> s X |