U ƒìõSöã@sˆdZdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z e  d¡Z Gdd„deƒZdd „Zed kr„eƒdS) zÞ Integration of debconf on the client side Provides the DebconfProxy class which allows to run the debconf frontend as normal user by connecting to the root running debconf through the socket of the passthrough frontend. )Ú DebconfProxyéN)ÚGLibzAptClient.DebconfProxyc@sJeZdZdZddd„Zdd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dS)rz§The DebconfProxy class allows to run the debconf frontend as normal user by connecting to the root debconf through the socket of the passthrough frontend. ÚgnomeNcCs€||_d|_|dkr4tjdd|_tj |jd¡|_t d|j¡t   t j t j ¡|_ |j   |j¡||_ d|_d|_g|_dS)a Initialize a new DebconfProxy instance. Keyword arguments: frontend -- the to be used debconf frontend (defaults to gnome) socket_path -- the path to the socket of the passthrough frontend. Will be created if not specified Nz aptdaemon-)Úprefixzdebconf.socketzdebconf socket: %s)Ú socket_pathÚtemp_dirÚtempfileZmkdtempÚosÚpathÚjoinÚlogÚdebugÚsocketZAF_UNIXZ SOCK_STREAMZbindÚfrontendÚ _listener_idÚ _active_connZ _watch_ids)Úselfrr©rú3/usr/lib/python3/dist-packages/aptdaemon/debconf.pyÚ__init__2szDebconfProxy.__init__cCs>t tj¡}d|d<d|d<|j|d<tjtjkr:d|d<|S)zcReturns a dictonary of the environment variables required by the debconf frontend. ZconfigdbZDEBCONF_DB_REPLACEzPipe{infd:none outfd:none}ZDEBCONF_DB_OVERRIDEZDEBIAN_FRONTENDÚ.Z DEBCONF_DEBUG)Úcopyr Úenvironrr ÚlevelÚloggingÚDEBUG)rÚenvrrrÚ_get_debconf_envGs   zDebconfProxy._get_debconf_envcCs4t d¡|j d¡t |jtjtj|j¡|_ dS)zStart listening on the socket.zdebconf.start()éN) rr rZlistenrÚ io_add_watchZPRIORITY_DEFAULT_IDLEÚIO_INÚ_accept_connectionr©rrrrÚstartSs  þzDebconfProxy.startcCslt d¡|j ¡|jdk r0t |j¡d|_|jrhzt  |j ¡t  |j¡Wnt k rfYnXdS)zStop listening on the socket.zdebconf.stop()N) rr rÚcloserrZ source_removerr ÚremoverÚrmdirÚOSErrorr"rrrÚstop[s     zDebconfProxy.stopcCs”|jrt d¡dS| ¡\}}||_tjtjBtjBtjB}t j dgt j t j |  ¡d|_ t |tj||j|j j¡t |j jtj||j|¡dS)NzDelaying connectionTzdebconf-communicate)ÚstdinÚstdoutr)rr r Zacceptrr ZIO_ERRZIO_HUPZIO_NVALÚ subprocessÚPopenÚPIPErÚhelperrZPRIORITY_HIGH_IDLEÚ _copy_connr)r*Ú _copy_stdout)rrÚ conditionÚconnZaddrÚmaskrrrr!is,  ý ÿÿzDebconfProxy._accept_connectionc Cs”t d¡z,| ¡}|r4t d|¡| |¡WdSWn2tjtfk rh}zt |¡W5d}~XYnXt d¡|jj   ¡|j   ¡d|_ dS)zaCallback to copy data from the stdout of debconf-communicate to the passthrough frontend.r0zFrom debconf: %sTNzStop reading from stdoutF) rr Úreadliner ÚsendrÚerrorÚIOErrorr.r*r$r)rÚsourcer1r2Z debconf_datar6rrrr0zs       zDebconfProxy._copy_stdoutc CsŽt d¡z6| d¡}|r>t d|¡| |¡| ¡WdSWn2tjtfk rr}zt |¡W5d}~XYnXt d¡|j j   ¡dS)z\Callback to copy data from the passthrough frontend to stdin of debconf-communicate.r/izFrom socket: %sTNzStop reading from connF) rr Zrecvr ÚwriteÚflushrr6r7r.r)r$)rr8r1r)Z socket_datar6rrrr/s       zDebconfProxy._copy_conn)rN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr#r(r!r0r/rrrrr+s  rcCsNtjtjdd}tj |¡r(t |¡td|ƒ}| ¡t   ¡}|  ¡dS)a4Run the DebconfProxy from the command line for testing purposes. You have to execute the following commands before in a separate terminal: $ echo "fset debconf/frontend seen false" | debconf-communicate $ export DEBCONF_PIPE=/tmp/debconf.socket $ dpkg-reconfigure debconf -f passthrough )rz/tmp/debconf.socketrN) rZ basicConfigrr r Úexistsr%rr#rZMainLoopÚrun)rÚproxyZlooprrrÚ_test s   rBÚ__main__)r>Ú__all__rrr Úos.pathrr+rÚsysZ gi.repositoryrZ getLoggerr ÚobjectrrBr;rrrrÚs  u