o Pf6 @sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZmZmZddlZddlmZm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&zddl'Z(Wn e)y{dZ(Ynweee*Z+ee+ee+fZ,ee-e*e.ee/fZ0ee0dfZ1ee.e-e*dfZ2eee*Z3e4eeeZ5e56de57dd ee*d ee*fd d Z8d e,d dfddZ9d e+d eee*fddZ:d e,d eeee*eeee*ffddZ;de d e fddZ<  dPde dee de-d e-fddZ=de-d e.fddZ>  dQde d ee*e*fde-d!ee?d ej@f d"d#ZAde d eee*d$ffd%d&ZB 'dRd e d(e.d eeee*fd)d*ZCd+e-d e-fd,d-ZDd.e-d/e-d e.fd0d1ZEd2e-d e-fd3d4ZFe de-d ee-fd5d6ZGd7ed efd8d9ZHd:ed;ed efde1d ee-e0ffd?d@ZJdAe-d e-fdBdCZKdAe-d e-fdDdEZLdFe-d e-fdGdHZMded e fdIdJZNGdKdLdLZOdMee-e?e*fd e-fdNdOZPdS)SN)contextmanager) AnyCallableDictIterableIteratorListOptionalSequenceTupleTypeUnion)urlparse uses_netloc uses_params uses_relative)ElementFigure)_locations_mirror _parse_sizenone_maxnone_min write_pngdatalocationreturnc Cst|tjstdurt|tjrt|}t|ds)td|dt |dt |dkr7t d|dz |d|d f}Wntt fyWtd |dt |dw|D]*}zt |Wntt fyxt d |dt |d wtt |rt d qZdd|DS)aValidate a single lat/lon coordinate pair and convert to a list Validate that location: * is a sized variable * with size 2 * allows indexing (i.e. has an ordering) * where both values are floats (or convertible to float) * and both values are not NaN N__len__zPLocation should be a sized variable, for example a list or a tuple, instead got z of type .z:Expected two (lat, lon) values for location, instead got: rzKLocation should support indexing, like a list or a tuple does, instead got z5Location should consist of two numerical values, but z is not convertible to float.z$Location values cannot contain NaNs.cSg|]}t|qS)float.0xr"r"\/home/deployuser/azure_apps/autowriter/venv/lib/python3.10/site-packages/folium/utilities.py fz%validate_location..) isinstancenpndarraypd DataFramesqueezetolisthasattr TypeErrortypelen ValueErrorKeyErrorr#mathisnan)rcoordscoordr"r"r'validate_location<sX    r; locationscCsVzt|Wntytd|dwz tt|WdSty*tdw)zHHelper function that does basic validation of line and multi-line types.zGLocations should be an iterable with coordinate pairs, but instead got rzLocations is empty.N)iterr2next StopIterationr5r<r"r"r'_validate_locations_basicsis   rAcCst|}t|dd|DS)z3Validate an iterable with lat/lon coordinate pairs.cSr!r"r;r% coord_pairr"r"r'r(|r)z&validate_locations..)if_pandas_df_convert_to_numpyrAr@r"r"r'validate_locationsxsrFc Csht|}t|zttttttt|Wnttfy,dd|DYSwdd|DS)zDValidate an iterable with possibly nested lists of coordinate pairs.cSr!r"rBrCr"r"r'r(r)z,validate_multi_locations..cSr!r")rF)r%lstr"r"r'r(r))rErAr#r>r=r2r?r@r"r"r'validate_multi_locationss$rHobjcCstdur t|tjr |jS|S)zReturn a Numpy array from a Pandas dataframe. Iterating over a DataFrame has weird side effects, such as the first row being the column names. Converting to Numpy is more safe. N)r-r*r.values)rIr"r"r'rEsrEupperimagecolormaporigincCst|tr?t|s?tj|ddd}t|d }|}Wdn1s)wYt | d}d|d|}n#d|j j vrZt |||d }t | d}d |}ntt|}|d d S) ap Infers the type of an image argument and transforms it into a URL. Parameters ---------- image: string, file or array-like object * If string, it will be written directly in the output file. * If file, it's content will be converted as embedded in the output file. * If array-like, it will be converted to PNG base64 string and embedded in the output. origin: ['upper' | 'lower'], optional, default 'upper' Place the [0, 0] index of the array in the upper left or lower left corner of the axes. colormap: callable, used only for `mono` image. Function of the form [x -> (r,g,b)] or [x -> (r,g,b,a)] for transforming a mono image into RGB. It must output iterables of length 3 or 4, with values between 0. and 1. You can use colormaps from `matplotlib.cm`. r Nrbzutf-8z data:image/z;base64,r,)rNrMzdata:image/png;base64,  )r*str_is_urlospathsplitextopenreadbase64 b64encodedecode __class____name__rjsonloadsdumpsreplace)rLrMrN fileformatfimg b64encodedurlr"r"r' image_to_urls     rhrgcCs&zt|jtvWStyYdSw)z+Check to see if `url` has a valid protocol.F)rscheme _VALID_URLS Exception)rgr"r"r'rTs  rT lat_bounds height_outc CsHdd}t|}|j\}}}t|dd} t|dd} |dur%|}|dkr7|ddd ddddf}| td |d d ||| | } || td |d d |||| || } t|||f} t|D] }t|D]}t | || |dd||f| dd||f<quqo|dkr| ddd ddddf} | S) aP Transforms an image computed in (longitude,latitude) coordinates into the a Mercator projection image. Parameters ---------- data: numpy array or equivalent list-like object. Must be NxM (mono), NxMx3 (RGB) or NxMx4 (RGBA) lat_bounds : length 2 tuple Minimal and maximal value of the latitude of the image. Bounds must be between -85.051128779806589 and 85.051128779806589 otherwise they will be clipped to that values. origin : ['upper' | 'lower'], optional, default 'upper' Place the [0,0] index of the array in the upper left or lower left corner of the axes. height_out : int, default None The expected height of the output. If None, the height of the input is used. See https://en.wikipedia.org/wiki/Web_Mercator for more details. cSs$tt|tjddtjS)Ngf@)r+arcsinhtanpi)r&r"r"r'mercators$z$mercator_transform..mercatorrgIECUr gIECU@NrKrOg?g?) r+ atleast_3dcopyshapemaxminlinspacezerosrangeinterp)rrlrNrmrqarrayheightwidthnblayerslat_minlat_maxlatslatslatsoutijr"r"r'mercator_transforms2!    0r.ccst|ttfr |}n@d|vrdd|dD}n2d|vr*|dr'|ddng}n!d|vrE|ddrEd|ddvrE|ddd}n|d|}|D]}t|ttfr^t|VdSt|EdHqMdS) zH Returns all the coordinate tuples from a geometry or feature. featurescSs g|] }|dr|ddqS)geometry coordinatesr")r%geomr"r"r'r(s  ziter_coords..rr geometriesrN)r*tuplelistgetr#int iter_coords)rIr9r:r"r"r'rs*   rFlonlatcCsddgddgg}t|D].}t|dd|dt|dd|dgt|dd|dt|dd|dgg}q |rAt|}|S)ze Computes the bounds of the object in the form [[lat_min, lon_min], [lat_max, lon_max]] Nrr )rrrr)r<rboundspointr"r"r' get_bounds.s  rkeycCsdddt|dDS)zConvert a python_style_variable_name to lowerCamelCase. Examples -------- >>> camelize("variable_name") 'variableName' >>> camelize("variableName") 'variableName' rcss(|]\}}|dkr|n|VqdS)rN) capitalize)r%rr&r"r"r' Rs&zcamelize.._)join enumeratesplit)rr"r"r'camelizeHs robj1obj2cCst|t|kS)zs Return True/False if the normalized rendered version of two folium map objects are the equal or not. ) normalize)rrr"r"r'compare_renderedUsrrenderedcCs(ddd|D}|dd}|S)zBReturn the input string without non-functional spaces or newlines.rcSsg|] }|r|qSr")strip)r%liner"r"r'r(`sznormalize..z, ,)r splitlinesrb)rrr"r"r'r^s rc csd}z1tjddd\}}t|t|tr|dn|t||VWtj |r3t |dSdStj |rAt |ww)z9Yields the path of a temporary HTML file containing data.rz.htmlfolium_)suffixprefixutf8N) tempfilemkstemprUwriter*rSencodecloserVisfileremove)rfilepathfidr"r"r'temp_html_filepathes    r item_originalcCsjt|}tj|_t|dr3t|jdkr3t }|j D]}t |}||_ ||| <q ||_|S)zBReturn a recursive deep-copy of item where each copy has a new ID. _childrenr)rsuuiduuid4hex_idr1r4r collections OrderedDictrJ deep_copy_parentget_name)ritem children_newsubitem_originalsubitemr"r"r'rss  relementclscCs4|j}|durtd|t||st||S|S)z:Return the first object in the parent tree of class `cls`.Nz2The top of the tree was reached without finding a )rr5r*get_obj_in_upper_tree)rrparentr"r"r'rs   rkwargscKsdd|DS)z=Return a dict with lower-camelcase keys and non-None values..cSs"i|] \}}|durt||qSN)r)r%rvaluer"r"r' s"z!parse_options..)items)rr"r"r' parse_optionssrtextcCstdd|S)z6Escape backticks so text can be used in a JS template.z(?.r)rr)rVr"r"r',javascript_identifier_path_to_array_notationsrcCs|}t|ts Jd|S)z=Return the root element of the tree and assert it's a Figure.z8You cannot render this Element if it is not in a Figure.)get_rootr*r)rIfigurer"r"r'get_and_assert_figure_rootsrc@s.eZdZdZdeedffddZddZdS)JsCodezWrapper around Javascript code.js_codecCs t|tr |j|_dS||_dSr)r*rr)selfrr"r"r'__init__s   zJsCode.__init__cCs|jSr)r)rr"r"r'__str__szJsCode.__str__N)r^ __module__ __qualname____doc__r rSrrr"r"r"r'rs rrcCsDt|ttfr |dS|dddkr |dddvr td|S)z,Parse a font size value, if number set as pxpxNrem)emrz2The font size must be expressed in rem, em, or px.)r*rr#r5)rr"r"r'parse_font_sizes   r)NrK)rKN)F)QrZrrsr_r7rUrrr contextlibrtypingrrrrrrr r r r r urllib.parserrrrnumpyr+branca.elementrrbranca.utilitiesrrrrrpandasr- ImportErrorr#TypeLine TypeMultiLinerSbooldictTypeJsonValueNoNone TypeJsonValueTypePathOptions TypeBoundssetrjdiscardaddr;rArFrHrErhrTrr,rrrrrrrrrrrrrrrrr"r"r"r's 4        -  *   B!