o Pf-V@s2dZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZd d lmZmZmZmZeed d dZGdddZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!Gdd d eZ"dS)!z: Element ------- A generic class for creating Elements. N)hexlify) OrderedDict)escape)urandom)Path)urlopen) Environment PackageLoaderTemplate) _camelify _parse_sizenone_maxnone_minbranca templates)loaderc@seZdZdZedZd#ddZddZdefd d Z d d Z d dZ ddZ d#ddZ d#ddZd#ddZd$ddZd%ddZddZdd Zd&d!d"ZdS)'ElementaJBasic Element object that does nothing. Other Elements may inherit from this one. Parameters ---------- template : str, default None A jinaj2-compatible template string for rendering the element. If None, template will be: .. code-block:: jinja {% for name, element in this._children.items() %} {{element.render(**kwargs)}} {% endfor %} so that all the element's children are rendered. template_name : str, default None If no template is provided, you can also provide a filename. z^{% for name, element in this._children.items() %} {{element.render(**kwargs)}}{% endfor %}NcCsdd|_ttd|_t|_d|_||_||_ |dur$t ||_ dS|dur0t ||_ dSdS)Nr)_namerrdecode_idr _children_parent _template_str_template_namer _templateENV get_template)selftemplate template_namer"Z/home/deployuser/azure_apps/autowriter/venv/lib/python3.10/site-packages/branca/element.py__init__6szElement.__init__cCs|j}|dd|S)zModify object state when pickling the object. jinja2 Templates cannot be pickled, so remove the instance attribute if it exists. It will be added back when unpickling (see __setstate__). rN)__dict__copypoprstater"r"r# __getstate__Cs  zElement.__getstate__r)cCsL|ddurt|d|d<n|ddurt|d|d<|j|dS)z3Re-add _template instance attribute when unpicklingrNrr)r rrr%updater(r"r"r# __setstate__Ms  zElement.__setstate__cCst|jd|jS)zReturns a string representation of the object. This string has to be unique and to be a python and javascript-compatible variable name. _)r rrrr"r"r#get_nameVszElement.get_namecCsddgddggS)zComputes the bounds of the object itself (not including it's children) in the form [[lat_min, lon_min], [lat_max, lon_max]] Nr"r.r"r"r#_get_self_bounds^szElement._get_self_boundscCs|}|jD]:}|}t|dd|ddt|dd|ddgt|dd|ddt|dd|ddgg}q |S)z~Computes the bounds of the object and all it's children in the form [[lat_min, lon_min], [lat_max, lon_max]]. rr )r0rvalues get_boundsrr)rboundschild child_boundsr"r"r#r2ds zElement.get_boundscCs tjdtdd|j|||dS) Add a child.zDMethod `add_children` is deprecated. Please use `add_child` instead.) stacklevelnameindex)warningswarn FutureWarning add_child)rr4r:r;r"r"r# add_childrenxs zElement.add_childrencsddur||dur||j<nfdd|jD}|t||ft||_||_|S)r6Ncsg|] }|dkr|qS)rr").0itemr:r"r# sz%Element.add_child..)r/ritemsinsertintrr)rr4r:r;rEr"rCr#r?s  zElement.add_childcCs|j|||d|S)zAdd element to a parent.r9)r?)rparentr:r;r"r"r#add_toszElement.add_toTc sT|rt}nt}|}|j|d<|j|d<dkr(|fdd|jD|d<|S),Returns a dict representation of the object.r:idrcs$g|]\}}||jddfqS)r depth)to_dict)rAr:r4rMr"r#rDsz#Element.to_dict..children)rdictrrrrE)rrNorderedkwargsdict_funoutr"rMr#rOs   zElement.to_dictcKstj|j|ddfi|S)z,Returns a JSON representation of the object.T)rNrR)jsondumpsrO)rrNrSr"r"r#to_jsonszElement.to_jsoncCs|jdur|S|jS)&Returns the root of the elements tree.N)rget_rootr.r"r"r#rZs  zElement.get_rootcKs|jj||dS)/Renders the HTML representation of the element.thisrS)rrender)rrSr"r"r#r^szElement.rendercKs\t|tttfrt|d}n|}|}|jdi|}||d|r,| dSdS)aSaves an Element into a file. Parameters ---------- outfile : str or file object The file (or filename) where you want to output the html. close_file : bool, default True Whether the file has to be closed after write. wbutf8Nr") isinstancestrbytesropenrZr^writeencodeclose)routfile close_filerSfidroothtmlr"r"r#saves  z Element.save)NN)rJTrJ)T)__name__ __module__ __qualname____doc__r rr$r*rQr,r/r0r2r@r?rIrOrXrZr^rmr"r"r"r#rs&       rcs*eZdZdZddZdfdd ZZS)Linkz3An abstract class for embedding a link in the HTML.cCs |jdur t|j|_|jS)z2Opens the link and returns the response's content.N)coderurlreadr.r"r"r#get_codes z Link.get_coderJc s$tjdddi|}|j|d<|S)rKrNrJruNr")superrOrurrNrSrU __class__r"r#rOs z Link.to_dictrn)rorprqrrrwrO __classcell__r"r"rzr#rssrsc*eZdZdZedZdfdd ZZS)JavascriptLinkzCreate a JavascriptLink object based on a url. Parameters ---------- url : str The url to be linked download : bool, default False Whether the target document shall be loaded right now. z{% if kwargs.get("embedded",False) %}{% else %}{% endif %}Fc0td|_||_d|_|r|dSdS)Nr~rxr$rrurtrwrrudownloadrzr"r#r$  zJavascriptLink.__init__Frorprqrrr rr$r|r"r"rzr#r~  r~cr})CssLinkzCreate a CssLink object based on a url. Parameters ---------- url : str The url to be linked download : bool, default False Whether the target document shall be loaded right now. z{% if kwargs.get("embedded",False) %}{% else %}{% endif %}Fcr)Nrrrrzr"r#r$ rzCssLink.__init__rrr"r"rzr#rrrcsdeZdZdZedZ     dfdd Zdfd d Zd d Zd dZ ddZ dddZ Z S)Figurea&Create a Figure object, to plot things into it. Parameters ---------- width : str, default "100%" The width of the Figure. It may be a percentage or pixel value (like "300px"). height : str, default None The height of the Figure. It may be a percentage or a pixel value (like "300px"). ratio : str, default "60%" A percentage defining the aspect ratio of the Figure. It will be ignored if height is not None. title : str, default None Figure title. figsize : tuple of two int, default None If you're a matplotlib addict, you can overwrite width and height. Values will be converted into pixels in using 60 dpi. For example figsize=(10, 5) will result in width="600px", height="300px". z {% if this.title %}{{this.title}}{% endif %} {{this.header.render(**kwargs)}} {{this.html.render(**kwargs)}} 100%N60%cstd|_t|_t|_t|_||j_||j_||j_||_||_ ||_ ||_ |durFt d|dd|_t d|dd|_ |jj tddddS) Nr<rpxr zE meta_httprC)rxr$rrheaderrlscriptrwidthheightratiotitlerbr?)rrrrrfigsizerzr"r#r$;s*  zFigure.__init__rJc sttjdd|i|}|jjdd|di||d<|jjdd|di||d<|jjdd|di||d<|S)rKrNr rrlrNr")rxrOrrlrryrzr"r#rO]s zFigure.to_dictcC|SrYr"r.r"r"r#rZezFigure.get_rootcKs4|jD] \}}|jdi|q|jj||dS)r[r\Nr")rrEr^r)rrSr:r4r"r"r#r^isz Figure.rendercKsNt|jdi|}|jdurdj||j|jd}|Sdj||j|jd}|S)z*Displays the Figure in a Jupyter notebook.Na
Make this Notebook Trusted to load map: File -> Trust Notebook
rlrrzrlrrr")rr^rformatrrrrSrliframer"r"r# _repr_html_os   zFigure._repr_html_皙?c Csd|}d|}|d||}|d||}|||}|||}|dd|}|dd|}tdd|dd|dd|dd|dd} || | S)aCreates a div child subplot in a matplotlib.figure.add_subplot style. Parameters ---------- x : int The number of rows in the grid. y : int The number of columns in the grid. n : int The cell number in the grid, counted from 1 to x*y. Example: >>> fig.add_subplot(3, 2, 5) # Create a div in the 5th cell of a 3rows x 2columns grid(bottom-left corner). g?r g@absolutegY@%)positionrrlefttop)Divr?) rxynmarginrrrrdivr"r"r# add_subplots"       zFigure.add_subplot)rNrNNrn)r) rorprqrrr rr$rOrZr^rrr|r"r"rzr#rs "rcs*eZdZdZedZdfdd ZZS)HtmlaCreate an HTML div object for embedding data. Parameters ---------- data : str The HTML data to be embedded. script : bool If True, data will be embedded without escaping (suitable for embedding html-ready code) width : int or str, default '100%' The width of the output div element. Ex: 120 , '80%' height : int or str, default '100%' The height of the output div element. Ex: 120 , '80%' z
{% if this.script %}{{this.data}}{% else %}{{this.data|e}}{% endif %}
Frcs4td|_||_||_t||_t||_dS)Nr)rxr$rrdatar rr)rrrrrrzr"r#r$s  z Html.__init__)Frrrr"r"rzr#rs rcsLeZdZdZedZ     dfdd Zdd Zd d Zd d Z Z S)raCreate a Div to be embedded in a Figure. Parameters ---------- width: int or str, default '100%' The width of the div in pixels (int) or percentage (str). height: int or str, default '100%' The height of the div in pixels (int) or percentage (str). left: int or str, default '0%' The left-position of the div in pixels (int) or percentage (str). top: int or str, default '0%' The top-position of the div in pixels (int) or percentage (str). position: str, default 'relative' The position policy of the div. Usual values are 'relative', 'absolute', 'fixed', 'static'. a{% macro header(this, kwargs) %}{% endmacro %}{% macro html(this, kwargs) %}
{{this.html.render(**kwargs)}}
{% endmacro %}r0%relativecsxtt|d|_t||_t||_t||_t||_||_ t |_ t d|_ t |_ ||j _||j _||j _dS)NrzY{% for name, element in this._children.items() %}{{element.render(**kwargs)}}{% endfor %})rxrr$rr rrrrrrrrlrr)rrrrrrrzr"r#r$s     z Div.__init__cCrrr"r.r"r"r#rZ rz Div.get_rootcKs8|j}t|ts Jd|jD] \}}|jdi|q|jjD] \}}|jj||dq$|jjD] \}}|jj||dq7|j j j dd}|dur`|jjt ||||d|j j j dd}|dur||jjt ||||d|j j j dd}|dur|jjt ||||ddSdS)r[8You cannot render this Element if it is not in a Figure.rCrNrlrr")rrarrrEr^rr?rrmoduler%getrr/rl)rrSfigurer:elementrrlrr"r"r#r^s( "z Div.rendercKsH|jdur|t|jjdi|}d|_|S|jjdi|}|S)z'Displays the Div in a Jupyter notebook.Nr")rrIrr)rrSrUr"r"r#r-s  zDiv._repr_html_)rrrrr) rorprqrrr rr$rZr^rr|r"r"rzr#rsrcs.eZdZdZd fdd ZfddZZS) IFrameaMCreate a Figure object, to plot things into it. Parameters ---------- html : str, default None Eventual HTML code that you want to put in the frame. width : str, default "100%" The width of the Figure. It may be a percentage or pixel value (like "300px"). height : str, default None The height of the Figure. It may be a percentage or a pixel value (like "300px"). ratio : str, default "60%" A percentage defining the aspect ratio of the Figure. It will be ignored if height is not None. figsize : tuple of two int, default None If you're a matplotlib addict, you can overwrite width and height. Values will be converted into pixels in using 60 dpi. For example figsize=(10, 5) will result in width="600px", height="300px". Nrrcstd|_||_||_||_|dur+td|dd|_td|dd|_t|ts5t|tr>| t |dS|durI| |dSdS)Nrrrrr ) rxr$rrrrrbrarcr?r)rrlrrrrrzr"r#r$Os zIFrame.__init__c sftjdi|}dt|dd}|jdur'dj||j|j d}|Sdj||j|jd}|S) r[z$data:text/html;charset=utf-8;base64,r`Nz
rz`rr") rxr^base64 b64encoderfrrrrrrrzr"r#r^_s"    z IFrame.render)NrNrN)rorprqrrr$r^r|r"r"rzr#r8srcs0eZdZdZedZfddZddZZS) MacroElementaThis is a parent class for Elements defined by a macro template. To compute your own element, all you have to do is: * To inherit from this class * Overwrite the '_name' attribute * Overwrite the '_template' attribute with something of the form:: {% macro header(this, kwargs) %} ... {% endmacro %} {% macro html(this, kwargs) %} ... {% endmacro %} {% macro script(this, kwargs) %} ... {% endmacro %} cstd|_dS)Nr)rxr$rr.rzr"r#r$s  zMacroElement.__init__cKs|}t|ts Jd|jjjdd}|dur)|jjt |||| d|jjjdd}|durE|j jt |||| d|jjjdd}|dura|j jt |||| d|j D] \}}|jdi|qfdS)r[rrNrCrlrr")rZrarrrr%rrr?rr/rlrrrEr^)rrSrrrlrr:rr"r"r#r^s  zMacroElement.render) rorprqrrr rr$r^r|r"r"rzr#rys  r)#rrrrVr<binasciir collectionsrrlrosrpathlibrurllib.requestrjinja2rr r utilitiesr r rrrrrsr~rrrrrrr"r"r"r#s0      1"jA