�
g��fc@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZddlmZ
ddlm	Z	
ddl
mZ
ddlm
Z

ddlmZ
ddlmZ
ddlmZmZ
dd	lmZ
d
ef
d��YZd
S(i����N(
tconfigparser(
tRawMessageDialog(
tCommandlineRenderer(
tCannotConsumeErrorMessages(
tAnswerStore(
tWorkflowCommand(tDialogModelt
ErrorModel(
tget_api_modelst
BaseMessagingcBs
eZ
dZeddd
�Zd�Zed��
Zed��
Z	d�Z
d�Zed��
Zd�Z
d	�Zd
�Zd�Zd�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Zed�
Zd�Zd�Zd�Zd�ZRS(s�
    BaseMessaging is the Base class for all messaging implementations. It provides the basic interface that is
    supported within the framework. These are called the `produce` and `consume` methods.
    cCs�tj
�|_t�|_|jj�|_|pBtd
|j�
|_|jj�|_	|jj�|_
|jj�|_|
|_|r�|f
nd|_
|jj�|_|jjtt�|_dS(Ntmanager((tmultiprocessingtManagert_managerRt_dialog_renderertlistt_dataRt_answerst	_new_datat	_commandst_errorst_storedt_config_modelst_dialogstValuetbooltFalset_stop_after_phase(tselftstoredtconfig_modeltanswer_store((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyt__init__s







	


cCs|jj
|
|�
d
S(sK

        Loads answers from a given answer file

        :param answer_file: Path to file to load as answer file
        :type answer_file: str
        :param workflow: :py:class:`leapp.workflows.Workflow` instance to load the answers for.
        :type workflow: :py:class:`leapp.workflows.Workflow`
        :return: None
        N(Rtload_and_translate_for_workflow(Rtanswer_filetworkflow((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytload_answers&s
c

Cs
t|j
�
S(
sZ
        :return: List of commands that have been sent to the workflow execution.
        (RR(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytcommands2sc


Cs|jS(
so
        :return: If the messages are stored immediately, this function returns True, otherwise False.
        (
R(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyR9sc

Cs
t|j
�
S(
sy
        Gets all dialogs actually encountered during workflow run

        :return: List of encountered dialogs
        (RR(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytdialogs@sc

Cs
t|j
�
S(
s[
        Gets all produced errors.

        :return: List of newly produced errors
        (RR(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyterrorsHsc


Cs
|jj
�S(
s�
        Returns True if execution stop was requested after the current

        :return: True if the executed was requested to be stopped.
        (Rtget(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytstop_after_phasePsc

Cs
t|j
�
S(
sf
        Gets all newly produced messages.

        :return: List of newly processed messages
        (RR(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytmessagesYsc
Cs
t��
d
S(s�
        Loads all messages that are requested from the `consumes` attribute of :py:class:`leapp.actors.Actor`

        :param consumes: Tuple or list of :py:class:`leapp.models.Model` types to preload
        :return: None
        N(
tNotImplementedError(Rtconsumes((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyt
_perform_loadasc
Cs
t��
d
S(s3

        This method performs the actual message sending, which can be sent over the network or stored
        in a database.

        :param message: The message data to process
        :type message: dict
        :return: Pass through a message that might get updated through the sending process.
        N(
R+(Rtmessage((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyt_process_messagejs	cCs0t|
krt
��
n|j|
|j�

d
S(sG

        Loads all messages that are requested from the `consumes` attribute of :py:class:`leapp.actors.Actor`

        :param consumes: Tuple or list of :py:class:`leapp.models.Model` types to preload
        :return: None
        :raises leapp.exceptions.CannotConsumeErrorMessages: When trying to consume ErrorModel
        N(RRR-R(RR,((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytloadus

cCse|rtj
|�
}ntd
|
d|jd|d|dtjj��}|j|||j�
dS(s�

        Reports an execution error

        :param message: Message to print the error
        :type message: str
        :param severity: Severity of the error
        :type severity: leapp.models.error_severity.ErrorSeverity
        :param actor: Actor name that produced the message
        :type actor: leapp.actors.Actor
        :param details: A dictionary where additional context information can be passed along with the error
        :type details: dict
        :return: None
        R.tactortseveritytdetailsttimeN(tjsontdumpsRtnametdatetimetnowt_do_produceR(RR.R2R1R3tmodel((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytreport_error�s


!

cCsJtd
|
d|d|dddt
j
j��}|j|||j�
dS(NR.R3R1R2tfatalR4(RR8R9R:R(RR.ttracet	actornameR;((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytreport_stacktrace�s


c

Cs|jj
t�

d
S(sk
        If called, it will cause the workflow to stop the execution after the current phase ends.
        N(RtsettTrue(
R((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytrequest_stop_after_phase�scCsT|
j|j
�
}g|
jj�D](}|jd
d�d|kr"|^q"}|S(Nt
.ii
(tget_answersRtanswerfile_sectionstkeystsplit(Rtdialogtuserchoicest
stres((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyt_unanswered_questions�s

>
c	Cs�|jj
|
�

|j|
�
}|r�|
jr7|
jn3tjd
jt|
jj	��
�
j
d�
�
j�}|jt
d|jd|
jd|�|�
nP|
j|j�
}x;|j�D]-\}}|
j|�
}|r�||_q�q�WdS(Nt
,sutf-8R1RFtkey(RtappendRMROthashlibtsha1tjointsortedRFRGtencodet	hexdigesttproduceRR7RERtitemstcomponent_by_keytvalue(	RRIR1t
unansweredt
stable_keyRJt	componentRZtdialog_component((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytregister_dialog�s


0

	




cCs8t|
t
�r(|jj|
j��

ntd
�
�
dS(s�
        Called to send a command to the workflow execution

        :param command: A command to send to the workflow execution.
        :type command: Instance of :py:class:`leapp.messaging.commands.WorkflowCommand`
        :return: None
        s'Expected an instance of WorkflowCommandN(t
isinstanceRRRPRUt	TypeError(Rtcommand((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyRb�s
cCs|j|
||j
�S(
sX

        Called to send a message available for other actors.

        :param model: Model to send as message payload
        :type model: :py:class:`leapp.models.Model`
        :param actor: Actor that sends the message
        :type actor: :py:class:`leapp.actors.Actor`
        :return: the updated message dict
        :rtype: dict
        (R:R(RR;R1((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyRW�scCs|j|
||j
d
t�
S(sn

        Called to pre-fill sent messages and make them available for other actors.

        :param model: Model to send as message payload
        :type model: :py:class:`leapp.models.Model`
        :param actor: Actor that sends the message
        :type actor: :py:class:`leapp.actors.Actor`
        :return: the updated message dict
        :rtype: dict
        R(R:RR(RR;R1((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytfeed�scCsE
tj
jd
d�s+tj�tj
d
<ntj|
j�dt	�

}it
|
�
jd6t|t
�swt
|�
jn|d6|
jjd6tjj�j�dd6tj
jdd	�d
6tj
jdd�d
6tj
d
d6i|d6tj|jd�
�
j�d6d6}|r4
|jr4
|j|j��

n|j|�

|S(NtLEAPP_HOSTNAMEt	sort_keysttypeR1ttopict
ZtstamptLEAPP_CURRENT_PHASEsNON-WORKFLOW-EXECUTIONtphasetLEAPP_EXECUTION_IDsTESTING-CONTEXTtcontextthostnametdatasutf-8thashR.(tostenvironR(tNonetsockettgetfqdnR5R6tdumpRBRft__name__R`tstrR7RgR8tutcnowt	isoformatRQtsha256RURVRR/tcopyRP(RR;R1ttargetRRoR.((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyR:�s$





%








&


cCs|
j|j
�
S(
N(RER(RRI((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyRE�scCs|
j|j
|j�S(
N(trequest_answersRR(RRI((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyR~�scCs#td
|
�
j
|j|j�
dS(s�
        Display a message in user interterface currently in use (CLI, GUI).

        It uses one of the dialog renderers in :py:mod:`leapp.dialogs.renderer`.

        :param message: Message to show
        :type message: str
        R.N(R
R~RR(RR.((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytshow_message

s	cs�td
�|D�
�
}t
|j�
t
|j�
}d�tt|
�
d�|jD�
�|r�td�|D�
�
}g|D]}|d|kr{|^q{}n�f
d�|D�
S(s7

        Returns all consumable messages and filters them by `types`

        :param types: Variable number of :py:class:`leapp.models.Model` derived types to filter messages to be consumed
        :param actor: Actor that consumes the data
        :return: Iterable with messages matching the criteria
        c
ss!|]}
t|
d|
�V
qd
S(t	_resolvedN(
tgetattr(t.0t
t((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pys	<genexpr>
sc
Ssi|]}
|
|
j�qS((
Rw(R�R;((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pys
<dictcomp>
s	R,c
ss|]}
|
jV
qdS(
N(
Rw(R�t	requested((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pys	<genexpr>
sRfc
3s7|]-}
�|
djt
j|
d
d�
�
V
qdS(RfR.RoN(tcreateR5tloads(R�R.(
tlookup(s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pys	<genexpr>
s(ttupleRRRRRfRRA(RR1ttypesR*tfilteredR.((
R�s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pytconsume
s
&


,
N(Rwt
__module__t__doc__RBRsR R$tpropertyR%RR&R'R)R*R-R/R0R<R@RCRMR_RbRWRcR:RER~RR�(((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyR	s2
															
	
	
			(R8RQR5RRqRtt	six.movesRt
leapp.dialogsR
tleapp.dialogs.rendererRtleapp.exceptionsRtleapp.messaging.answerstoreRtleapp.messaging.commandsRtleapp.modelsRRtleapp.utilsRtobjectR	(((s</usr/lib/python2.7/site-packages/leapp/messaging/__init__.pyt<module>
s