o ñu]e/ã@sädZddlZddlZddlZddlZddlZddlZddlZddlZddl Zddl Zddl Zddl m Z mZmZmZmZddlmZzddlmZWneyYddlmZYnwdZdd„Zejd d „ƒZGd d „d ejƒZdS) z7 Handles making requests and formatting the responses. éN)ÚSTANDARD_OUTPUTÚ BOLD_OUTPUTÚ ERROR_OUTPUTÚ uses_settingsÚmsg)Úformat)ÚStringIOédc Cs|sz| d¡WStdƒ‚tjj |¡r|Stjj |¡r1z| |¡jWStd|ƒ‚d|vs.handle_event_wrapper) r)ÚcodeÚInteractiveConsoleÚ__init__rZcontrolÚ get_eventsÚ _controllerÚ_run_python_commandsÚis_multiline_contextZ _handle_event)r.rr/r r-r!r2qsü zControlInterpreter.__init__cs4t|jƒ}tttjˆƒƒ‰ˆr‡fdd„|Dƒ}|S)Ncsg|] }|jˆvr|‘qSr )Útype©Ú.0Úe©Ú event_typesr r!Ú –sz1ControlInterpreter.get_events..)rr)ÚmapÚstrÚupper)r.r<r(r r;r!r3‘s zControlInterpreter.get_eventscCstjj |j|¡S)z‡ Performs the '/help' operation, giving usage information for the given argument or a general summary if there wasn't one. )rZ interpreterÚhelprr4)r.rr r r!Údo_helpšszControlInterpreter.do_helpcCsJ| ¡ ¡}d|vr|jdd…=tdgt¢RŽSd dd„|j|ŽDƒ¡S)a Performs the '/events' operation, dumping the events that we've received belonging to the given types. If no types are specified then this provides all buffered events. If the user runs '/events clear' then this clears the list of events we've received. ZCLEARNzcleared event backlogÚ cSs g|] }tt|ƒgt¢RŽ‘qSr )rr?rr8r r r!r=²s z0ControlInterpreter.do_events..)r@Úsplitr)rrÚjoinr3)r.rr<r r r!Ú do_events¢s  zControlInterpreter.do_eventsc CsÔzt||jƒ}Wnty$}ztt|ƒgt¢RŽWYd}~Sd}~ww|j |d¡}|j |d¡}d}|j |d¡}|sHtd|gt¢RŽSt j j j dd}|  |¡} | |¡} | D]} | }q\| D]} | }qcg} z |  t |j¡d¡WnYz|  |j d|j¡¡WnY| r˜dd | ¡nd } |r¢t|jƒ}n |rªt|jƒ}nd }d |j|ftd gt¢RŽd |j|j| fg}|rÖ| tdgt¢RŽt|jƒ¡| tdgt¢RŽd |j¡¡| tdgt¢RŽ|¡|r(|jr(t jj |j¡}dD] }| |d¡}qdD] }| |d¡}q| tdgt¢RŽ|¡d|fd|fd|fd|fg}tdgt ¢RŽ}|D]$\}} | rc|d |t|gt¢RŽ|d g7}|dd„t| ƒ !¡Dƒ7}q@d |¡S)z³ Performs the '/info' operation, looking up a relay by fingerprint, IP address, or nickname and printing its descriptor and consensus entries in a pretty fashion. Nz+Unable to find consensus information for %sé)Ztimeoutrzip-to-country/%sz (%s)z, ÚZUnknownz%s (%s)z address: z%s:%s%sz tor version: zflags: z exit policy: )z at z AT ú@)z dot z DOT Ú.z contact: zServer Descriptor:zExtrainfo Descriptor:zMicrodescriptor:zRouter Status Entry:zP--------------------------------------------------------------------------------cSsg|] }t|gt¢RŽ‘qSr )rr)r9Úlr r r!r=sz.ControlInterpreter.do_info..rC)"r"r4rrr?rrZget_server_descriptorZget_microdescriptorrZ descriptorZremoteZDescriptorDownloaderZget_server_descriptorsZget_extrainfo_descriptorsÚappendÚsocketZ gethostbyaddrrr rEZ exit_policyZnicknamerrZ tor_versionÚflagsÚcontactrZ str_toolsZ _to_unicodeÚreplacerÚ splitlines)r.rr ÚexcZns_descZ server_descZextrainfo_descZ micro_descZ downloaderZserver_desc_queryZextrainfo_desc_queryrZaddress_extrainfoZaddress_extrainfo_labelZexit_policy_labelÚlinesrOÚaliasZdescriptor_sectionZdivZlabelr r r!Údo_info´sx €ÿ      þ "ü € zControlInterpreter.do_infocCs‚|s|jrdnd}td|gt¢RŽS| ¡dkrd|_n| ¡dkr'd|_n td|gt¢RŽS|jr7d }nd }t|gt¢RŽS) z] Performs the '/python' operation, toggling if we accept python commands or not. ÚenabledZdisabledzPython support is currently %s.ZenableTZdisableFzP'%s' is not recognized. Please run either '/python enable' or '/python disable'.zIPython support enabled, we'll now run non-interpreter commands as python.zBPython support disabled, we'll now pass along all commands to tor.)r5rrÚlowerr)r.rÚstatusrr r r!Ú do_pythons  zControlInterpreter.do_pythonFc Csú| ¡d}}d|vr| dd¡\}}d}| d¡rY| ¡}|dkr&t ¡‚|dkr0| |¡}n¿|dkr:| |¡}nµ|dkrD| |¡}n«|d krN|  |¡}n¡t d |gt ¢RŽ}n–|  ¡}|  d d¡d vrpt td ƒgt ¢RŽ}n|dkr~|j |¡t ¡‚|| di¡vo‹| ¡dk}|jr¸|s¸tƒ}t||ƒtj ||¡|_Wdƒn1s¬wY| ¡ ¡}n7zt |j |¡ ¡ ¡gt¢RŽ}Wn$tjyî} zt| tjƒrÚ‚t t| ƒgt ¢RŽ}WYd} ~ nd} ~ ww|rû|d7}|rût|ƒ|S)a Runs the given command. Requests starting with a '/' are special commands to the interpreter, and anything else is sent to the control port. :param stem.control.Controller controller: tor control connection :param str command: command to be processed :param bool print_response: prints the response to stdout if true :returns: **list** out output lines, each line being a list of (msg, format) tuples :raises: **stem.SocketClosed** if the control connection has been severed rHú r ú/z/quitz/eventsz/infoz/pythonz/helpz'%s' isn't a recognized commandú+)ZLOADCONFZPOSTDESCRIPTORz"msg.multiline_unimplemented_noticeZQUITz help.usager(NrC)ÚstriprDÚ startswithrWrZ SocketClosedrFrUrYrBrrr@rPrr4Úgetr5rr&r0r1Úpushr6ÚgetvalueZ raw_contentrZControllerErrorÚ isinstancer?Úprint) r.ZcommandZconfigZprint_responseÚcmdrÚoutputZis_tor_commandZconsole_outputrRr r r!Ú run_command-sV        ÿ$  €üzControlInterpreter.run_commandN)F) Ú__name__Ú __module__Ú __qualname__Ú__doc__r2r3rBrFrUrYrrfr r r r!r'ks  br')rjr0Ú contextlibrMr#rZ stem.controlZstem.descriptor.remoteZstem.interpreter.helpZstem.util.connectionZstem.util.str_toolsZstem.util.tor_toolsZstem.interpreterrrrrrZstem.util.termrÚ cStringIOrÚ ImportErrorÚior+r"Úcontextmanagerr&r1r'r r r r!Ús0  ÿ@