Z:S&@s&dZddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddlm Z m Z ddlmZmZGdddejZGd d d ejZdd d Zedkr"GdddeZeddd ZejndS)a yoton.clientserver.py Yoton comes with a small framework to setup a request-reply pattern using a client-server model (over a non-persistent connection), similar to telnet. This allows one process to easily ask small pieces of information from another process. To create a server, create a class that inherits from yoton.RequestServer and implement its handle_request() method. A client process can simply use the yoton.do_request function. Example: ``yoton.do_request('www.google.com:80', 'GET http/1.1\r\n')`` The client server model is implemented using one function and one class: yoton.do_request and yoton.RequestServer. Details ------- The server implements a request/reply pattern by listening at a socket. Similar to telnet, each request is handled using a connection and the socket is closed after the response is send. The request server can setup to run in the main thread, or can be started using its own thread. In the latter case, one can easily create multiple servers in a single process, that listen on different ports. N)select) basestringbytesstrlong) split_address getErrorMsg)send_allrecv_allc@speZdZdZddddZddZdd Zd d Zd d ZddZ ddZ dS) RequestServera^ RequestServer(address, async=False, verbose=0) Setup a simple server that handles requests similar to a telnet server, or asyncore. Starting the server using run() will run the server in the calling thread. Starting the server using start() will run the server in a separate thread. To create a server, subclass this class and re-implement the handle_request method. It accepts a request and should return a reply. This server assumes utf-8 encoded messages. Parameters ---------- address : str Should be of the shape hostname:port. async : bool If True, handles each incoming connection in a separate thread. This might be advantageous if a the handle_request() method takes a long time to execute. verbose : bool If True, print a message each time a connection is accepted. Notes on hostname ----------------- The hostname can be: * The IP address, or the string hostname of this computer. * 'localhost': the connections is only visible from this computer. Also some low level networking layers are bypassed, which results in a faster connection. The other context should also connect to 'localhost'. * 'publichost': the connection is visible by other computers on the same network. FrcCstjj|||_||_t|\}}}tjtjtj}|j tj tj d|j ||f||_ d|_|jddS)NFT) threadingThread__init___async_verbosersocketAF_INET SOCK_STREAM setsockopt SOL_SOCKET SO_REUSEADDRbind _bsd_socket_stop_me setDaemon)selfaddressZasyncverboseprotocolhostportsr#M/Applications/pyzo2014a/lib/python3.4/site-packages/iep/yoton/clientserver.pyrRs    zRequestServer.__init__cCsd|_tjj|dS)z@ start() Start the server in a separate thread. FN)rr rstart)rr#r#r$r%qs zRequestServer.startcCs d|_dS)z) stop() Stop the server. TN)r)rr#r#r$stopyszRequestServer.stopc Cs)|j}|j\}}|jr;td||fn|jd|jjdx|jsy|jj\}}Wn&tj k rYq[t k rYq[X|jrtdt |n|j rt ||}|jq[|j|q[Wy|jjWntjk r$YnXdS)z0 run() The server's main loop. zYoton: hosting at %s, port %ir g?zhandling request from: N)r getsocknamerprintlisten settimeoutracceptrtimeoutInterruptedErrorrr SocketHandlerr%_handle_connectioncloseerror)rr"hostnamer!addrZrhr#r#r$runs.        zRequestServer.runc CsAy|j|Wn)tk r<tdttYnXdS)zG _handle_connection(s) Handle an incoming connection. zError handling request:N)_really_handle_connection Exceptionr(r)rr"r#r#r$r/s   z RequestServer._handle_connectionc Cst|d}|dkrdS|j|}t|tsLtdnt||dy|jWntjk rYnXdS)zT _really_handle_connection(s) Really handle an incoming connection. TNz(handle_request() should return a string.) r handle_request isinstancer ValueErrorr r0rr1)rr"requestreplyr#r#r$r5s z'RequestServer._really_handle_connectioncCs<|dkrtjdn|dkr4d|_nd|S)a) handle_request(request) Return a reply, given the request. Overload this method to create a server. De standard implementation echos the request, waits one second when receiving 'wait' and stop the server when receiving 'stop'. waitg?r&Tz Requested: )timesleepr)rr:r#r#r$r7s   zRequestServer.handle_requestN) __name__ __module__ __qualname____doc__rr%r&r4r/r5r7r#r#r#r$r .s "   - r c@s.eZdZdZddZddZdS)r.zL SocketHandler(server, s) Simple thread that handles a connection. cCs&tjj|||_||_dS)N)r rr_serverr)rZserverr"r#r#r$rs zSocketHandler.__init__cCs|jj|jdS)N)rCr/r)rr#r#r$r4szSocketHandler.runN)r?r@rArBrr4r#r#r#r$r.s  r.r cCst|\}}}t|ts3tdn|dkrHd}ntjtjtj}y|j||fWn!tjk rt dYnXt ||dt ||}y|j Wntjk rYnX|S)aK do_request(address, request, timeout=-1) Do a request at the server at the specified address. The server can be a yoton.RequestServer, or any other server listening on a socket and following a REQ/REP pattern, such as html or telnet. For example: ``html = do_request('www.google.com:80', 'GET http/1.1\r\n')`` Parameters ---------- address : str Should be of the shape hostname:port. request : string The request to make. timeout : float If larger than 0, will wait that many seconds for the respons, and return None if timed out. Notes on hostname ----------------- The hostname can be: * The IP address, or the string hostname of this computer. * 'localhost': the connections is only visible from this computer. Also some low level networking layers are bypassed, which results in a faster connection. The other context should also connect to 'localhost'. * 'publichost': the connection is visible by other computers on the same network. zrequest should be a string.Nr z)No server is listening at the given port.T) rr8rr9rrrconnectr1 RuntimeErrorr r r0)rr:r,rr r!r"r;r#r#r$ do_requests"   rG__main__c@seZdZddZdS)LalacCs!tdt|dtjS)NzREQ:zThe current time is %i)r(reprr=)rZreqr#r#r$r77szLala.handle_requestN)r?r@rAr7r#r#r#r$rI6s rIzlocalhost:testrD)rBossysr=rr r yoton.miscrrrrrr yoton.corer r rr r.rGr?rIr"r%r#r#r#r$"s$  "A