a  d@sdZddlZddlZddlZddlZddlmZddlmZ ddl Z ddl Z dZ GdddZ Gdd d Zdd d Zdd dZdddZdS)z A Simple server used to show altair graphics from a prompt or script. This is adapted from the mpld3 package; see https://github.com/mpld3/mpld3/blob/master/mpld3/_server.py N)server)BytesIOz Note: if you're in the Jupyter notebook, Chart.serve() is not the best way to view plots. Consider using Chart.display(). You must interrupt the kernel to cancel this command. c@seZdZddZddZdS) MockRequestcOstdS)NsGET /)IO)selfargskwargsr \/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/altair/utils/server.pymakefileszMockRequest.makefilecCsdSNr )rresponser r r sendallszMockRequest.sendallN)__name__ __module__ __qualname__r rr r r r rsrc@s$eZdZddZddZddZdS) MockServercCs|t|d|dS)Nr)r)rZip_portHandlerr r r __init__#szMockServer.__init__cCsdSr r rr r r serve_forever&szMockServer.serve_forevercCsdSr r rr r r server_close)szMockServer.server_closeN)rrrrrrr r r r r"srcs(dur iGfdddtj}|S)NcseZdZfddZdS)z#generate_handler..MyHandlercs|jdkr:|d|dd||jnR|jvr|j\}}|d|d|||j|n |ddS)zRespond to a GET request./z Content-typez text/htmliN)pathZ send_responseZ send_headerZ end_headerswfilewriteencodeZ send_error)r content_typecontentfileshtmlr r do_GET2s      z*generate_handler..MyHandler.do_GETN)rrrr#r r r r MyHandler1sr$)rZBaseHTTPRequestHandler)r"r!r$r r r generate_handler-sr%2c s~tfddt|Dtd|d|}|D]:ttjtj}||f}| |dkr6Sq6t ddS)z)Find an open port near the specified portc3s|]}|VqdSr r ).0iportr r Hz!find_open_port..rzno open ports foundN) itertoolschainrangerandomrandintsocketAF_INET SOCK_STREAM connect_exclose ValueError)ipr*nZportssresultr r)r find_open_portEs* r> 127.0.0.1"Tc st|t||}|dur0tf|} n|f|} |rfztWnty\Yn 0tttdt j |rfdd} t j | dz | WnttfytdYn0| dS)a{Start a server serving the given HTML, and (optionally) open a browser Parameters ---------- html : string HTML to serve ip : string (default = '127.0.0.1') ip address at which the HTML will be served. port : int (default = 8888) the port at which to serve the HTML n_retries : int (default = 50) the number of nearby ports to search if the specified port is in use. files : dictionary (optional) dictionary of extra content to serve jupyter_warning : bool (optional) if True (default), then print a warning if this is used within Jupyter open_browser : bool (optional) if True (default), then open a web browser to the given HTML http_server : class (optional) optionally specify an HTTPServer class to use for showing the figure. The default is Python's basic HTTPServer. Nz,Serving to http://{}:{}/ [Ctrl-C to exit]cstdS)Nz http://{}:{}) webbrowseropenformatr r:r*r r bszserve..b)targetz stopping Server...)r>r%rZ HTTPServerZ __IPYTHON__ NameErrorprintJUPYTER_WARNINGrCsysstdoutflush threadingThreadstartrKeyboardInterrupt SystemExitr) r"r:r*Z n_retriesr!Zjupyter_warningZ open_browserZ http_serverrZsrvrrEr rDr serveTs*     rR)N)r&)r?r@r&NTTN)__doc__rJrMrAr4httpriorrr/r2rIrrr%r>rRr r r r s*