
    aBelT                        d Z ddlmZ ddlZddlZddlZddlmZm	Z	m
Z
 ddlmZ ej                  dD  ci c]  } | d c}       Z G d d	e      Z G d
 de      Zd ZdZyc c} w )z"
Utilities to support XYZservices
    )annotationsN)OptionalCallableUnion)quotez., -_/ c                  X    e Zd ZdZd Zd Zd	dZd
dZ	 	 	 	 d	 	 	 	 	 	 	 	 	 ddZddZ	y)Buncha  A dict with attribute-access

    :class:`Bunch` is used to store :class:`TileProvider` objects.

    Examples
    --------
    >>> black_and_white = TileProvider(
    ...     name="My black and white tiles",
    ...     url="https://myserver.com/bw/{z}/{x}/{y}",
    ...     attribution="(C) xyzservices",
    ... )
    >>> colorful = TileProvider(
    ...     name="My colorful tiles",
    ...     url="https://myserver.com/color/{z}/{x}/{y}",
    ...     attribution="(C) xyzservices",
    ... )
    >>> MyTiles = Bunch(BlackAndWhite=black_and_white, Colorful=colorful)
    >>> MyTiles
    {'BlackAndWhite': {'name': 'My black and white tiles', 'url': 'https://myserver.com/bw/{z}/{x}/{y}', 'attribution': '(C) xyzservices'}, 'Colorful': {'name': 'My colorful tiles', 'url': 'https://myserver.com/color/{z}/{x}/{y}', 'attribution': '(C) xyzservices'}}
    >>> MyTiles.BlackAndWhite.url
    'https://myserver.com/bw/{z}/{x}/{y}'
    c                V    	 | j                  |      S # t        $ r t        |      w xY wN)__getitem__KeyErrorAttributeError)selfkeys     /lib/python3.12/site-packages/xyzservices/lib.py__getattr__zBunch.__getattr__*   s2    	&##C(( 	& %%	&s    (c                "    | j                         S r   keysr   s    r   __dir__zBunch.__dir__0   s    yy{    c                4   d}| j                         D ]_  }t        | |   t              rd}nd}t        t	        j
                               }|d| d| d| d| d| |   j                  d	
       dz  }a |rdn	dt         d}d| dt        |        d| d}|S )Nr   zxyzservices.TileProviderzxyzservices.BunchzO
            <li class="xyz-child">
                <input type="checkbox" id="z5" class="xyz-checkbox"/>
                <label for="z">z <span>zM</span></label>
                <div class="xyz-inside">
                    T)insidez6
                </div>
            </li>
            <style></style>
        <div>
        z
            <div class="xyz-wrap">
                <div class="xyz-header">
                    <div class="xyz-obj">xyzservices.Bunch</div>
                    <div class="xyz-name">z items</div>
                </div>
                <div class="xyz-details">
                    <ul class="xyz-collapsible">
                        z\
                    </ul>
                </div>
            </div>
        </div>
        )	r   
isinstanceTileProviderstruuiduuid4_repr_html_	CSS_STYLElen)r   r   childrenr   objuidstylehtmls           r   r$   zBunch._repr_html_3   s    99; 	C$s)\20)djjl#C ,,/5 1 EC5u 5#Y**$*78 9 H	  GI;h"?		 + ,/t9+ 6 "
 #	" r   c                *    i fd |        S )a  Return the nested :class:`Bunch` collapsed into the one level dictionary.

        Dictionary keys are :class:`TileProvider` names (e.g. ``OpenStreetMap.Mapnik``)
        and its values are :class:`TileProvider` objects.

        Returns
        -------
        flattened : dict
            dictionary of :class:`TileProvider` objects

        Examples
        --------
        >>> import xyzservices.providers as xyz
        >>> len(xyz)
        36

        >>> flat = xyz.flatten()
        >>> len(xyz)
        207

        c                    t        | t              r| | j                  <   y | j                         D ]
  } |        y r   )r   r    namevalues)providerprov_get_providersflats     r   r2   z%Bunch.flatten.<locals>._get_providerss   s9    (L1&.X]]#$OO- )D"4()r    )r   r2   r3   s    @@r   flattenzBunch.flattenZ   s    . 	) 	tr   Nc                2    d fd | ||||      S )a		  Return a subset of the :class:`Bunch` matching the filter conditions

        Each :class:`TileProvider` within a :class:`Bunch` is checked against one or
        more specified conditions and kept if they are satisfied or removed if at least
        one condition is not met.

        Parameters
        ----------
        keyword : str (optional)
            Condition returns ``True`` if ``keyword`` string is present in any string
            value in a :class:`TileProvider` object.
            The comparison is not case sensitive.
        name : str (optional)
            Condition returns ``True`` if ``name`` string is present in
            the name attribute of :class:`TileProvider` object.
            The comparison is not case sensitive.
        requires_token : bool (optional)
            Condition returns ``True`` if :meth:`TileProvider.requires_token` returns
            ``True`` (i.e. if the object requires specification of API token).
        function : callable (optional)
            Custom function taking :class:`TileProvider` as an argument and returns
            bool. If ``function`` is given, other parameters are ignored.

        Returns
        -------
        filtered : Bunch

        Examples
        --------
        >>> import xyzservices.providers as xyz

        You can filter all free providers (not requiring API token):

        >>> free_providers = xyz.filter(requires_token=False)

        Or all providers with ``open`` in the name:

        >>> open_providers = xyz.filter(name="open")

        You can use keyword search to find all providers based on OpenStreetMap data:

        >>> osm_providers = xyz.filter(keyword="openstreetmap")

        You can combine multiple conditions to find providers based on OpenStreetMap
        data that require API token:

        >>> osm_locked = xyz.filter(keyword="openstreetmap", requires_token=True)

        You can also pass custom function that takes :class:`TileProvider` and returns
        boolean value. You can then find all providers with ``max_zoom`` smaller than
        18:

        >>> def zoom18(provider):
        ...    if hasattr(provider, "max_zoom") and provider.max_zoom < 18:
        ...        return True
        ...    return False
        >>> small_zoom = xyz.filter(function=zoom18)
        c                   g }|]d}| j                         D ]7  }t        |t              s|j                         |j                         v s5d} n |j	                  |       |?d}|j                         | j
                  j                         v rd}|j	                  |       |'d}| j                         |u rd}|j	                  |       t        |      S )NFT)r/   r   r!   lowerappendr.   requires_tokenall)	r0   keywordr.   r:   condkeyword_matchv
name_matchtoken_matchs	            r   	_validatezBunch.filter.<locals>._validate   s    D" %!* "A!!S)"==?aggi7,0M!	"
 M*"
::<8==#6#6#88!%JJ')#**,>"&KK(t9r   c                    t               }| j                         D ]N  \  }}t        |t              r$| 
||||      s%|||<   + ||      s4|||<   : 	|||||      }|sJ|||<   P |S )N)r<   r.   r:   r<   r.   r:   function)r
   itemsr   r    )bunchr<   r.   r:   rE   newr   valuefiltered_filter_bunchrB   s            r   rK   z#Bunch.filter.<locals>._filter_bunch   s    'C#kkm ,
Ue\2'$!$+!%+9	 (-CH#E?',CH  - '!'5!) H  #+C1,4 Jr   rD   r4   )r   r<   r.   r:   rE   rK   rB   s        @@r   filterzBunch.filter~   s,    D	6	< )
 	
r   c                (   | j                         j                         D ci c])  \  }}|j                  t              j	                         |+ }}}|j                  t              j	                         }||v r||   S t        d| d      c c}}w )a  Return :class:`TileProvider` based on the name query

        Returns a matching :class:`TileProvider` from the :class:`Bunch` if the ``name``
        contains the same letters in the same order as the provider's name irrespective
        of the letter case, spaces, dashes and other characters.
        See examples for details.

        Parameters
        ----------
        name : str
            Name of the tile provider. Formatting does not matter.

        Returns
        -------
        match: TileProvider

        Examples
        --------
        >>> import xyzservices.providers as xyz

        All these queries return the same ``CartoDB.Positron`` TileProvider:

        >>> xyz.query_name("CartoDB Positron")
        >>> xyz.query_name("cartodbpositron")
        >>> xyz.query_name("cartodb-positron")
        >>> xyz.query_name("carto db/positron")
        >>> xyz.query_name("CARTO_DB_POSITRON")
        >>> xyz.query_name("CartoDB.Positron")

        z*No matching provider found for the query 'z'.)r5   rF   	translateQUERY_NAME_TRANSLATIONr8   
ValueError)r   r.   kr?   xyz_flat_lower
name_cleans         r   
query_namezBunch.query_name  s    B ,,.
1 KK./557:
 
 ^^$:;AAC
'!*--EdV2NOO
s   .BF)returndict)NNNN)
r<   Optional[str]r.   rX   r:   Optional[bool]rE   zCallable[[TileProvider], bool]rV   r
   r.   r!   rV   r    )
__name__
__module____qualname____doc__r   r   r$   r5   rL   rT   r4   r   r   r
   r
      sm    4&%N"L "&")-37A
A
 A
 '	A

 1A
 
A
F'Pr   r
   c                       e Zd ZdZ fdZd
dZd
dZ	 	 	 	 	 d	 	 	 	 	 	 	 	 	 	 	 ddZddZe	d        Z
ddZedd	       Z xZS )r    a$  
    A dict with attribute-access and that
    can be called to update keys


    Examples
    --------

    You can create custom :class:`TileProvider` by passing your attributes to the object
    as it would have been a ``dict()``. It is required to always specify ``name``,
    ``url``, and ``attribution``.

    >>> public_provider = TileProvider(
    ...     name="My public tiles",
    ...     url="https://myserver.com/tiles/{z}/{x}/{y}.png",
    ...     attribution="(C) xyzservices",
    ... )

    Alternatively, you can create it from a dictionary of attributes. When specifying a
    placeholder for the access token, please use the ``"<insert your access token
    here>"`` string to ensure that :meth:`~xyzservices.TileProvider.requires_token`
    method works properly.

    >>> private_provider = TileProvider(
    ...    {
    ...        "url": "https://myserver.com/tiles/{z}/{x}/{y}.png?apikey={accessToken}",
    ...        "attribution": "(C) xyzservices",
    ...        "accessToken": "<insert your access token here>",
    ...        "name": "my_private_provider",
    ...    }
    ... )

    It is customary to include ``html_attribution`` attribute containing HTML string as
    ``'&copy; <a href="https://www.openstreetmap.org/copyright">OpenStreetMap</a>
    contributors'`` alongisde a plain-text ``attribution``.

    You can then fetch all information as attributes:

    >>> public_provider.url
    'https://myserver.com/tiles/{z}/{x}/{y}.png'

    >>> public_provider.attribution
    '(C) xyzservices'

    To ensure you will be able to use the tiles, you can check if the
    :class:`TileProvider` requires a token or API key.

    >>> public_provider.requires_token()
    False
    >>> private_provider.requires_token()
    True

    You can also generate URL in the required format with or without placeholders:

    >>> public_provider.build_url()
    'https://myserver.com/tiles/{z}/{x}/{y}.png'
    >>> private_provider.build_url(x=12, y=21, z=11, accessToken="my_token")
    'https://myserver.com/tiles/11/12/21.png?access_token=my_token'

    c                    t        |   |i | g }dD ]&  }|| j                         vs|j                  |       ( t	        |      dkD  r ddj                  |       d}t        |      y )N)r.   urlattributionr   zyThe attributes `name`, `url`, and `attribution` are required to initialise a `TileProvider`. Please provide values for: `z`, ``)super__init__r   r9   r&   joinr   )r   argskwargsmissingelmsg	__class__s         r   re   zTileProvider.__init__i  s    $)&)0 	#B$r"	# w<! KK(),  !%% r   c                >    t        |       }|j                  |       |S r   )r    updater   rh   rH   s      r   __call__zTileProvider.__call__x  s    4 

6
r   c                    t        |       }|S r   )r    ro   s      r   copyzTileProvider.copy}  s    4 
r   c           	     n   | j                         }|d}|d}|d}|j                  |       |j                         rt        d      |j	                  d      }|r|}	|j	                  dd       n|j	                  dd      }	|r|j	                  d	d
      }
|
d   }nd} |j
                  d|||||	d|S )a  
        Build the URL of tiles from the :class:`TileProvider` object

        Can return URL with placeholders or the final tile URL.

        Parameters
        ----------

        x, y, z : int (optional)
            tile number
        scale_factor : str (optional)
            Scale factor (where supported). For example, you can get double resolution
            (512 x 512) instead of standard one (256 x 256) with ``"@2x"``. If you want
            to keep a placeholder, pass `"{r}"`.
        fill_subdomain : bool (optional, default True)
            Fill subdomain placeholder with the first available subdomain. If False, the
            URL will contain ``{s}`` placeholder for subdomain.

        **kwargs
            Other potential attributes updating the :class:`TileProvider`.

        Returns
        -------

        url : str
            Formatted URL

        Examples
        --------
        >>> import xyzservices.providers as xyz

        >>> xyz.CartoDB.DarkMatter.build_url()
        'https://a.basemaps.cartocdn.com/dark_all/{z}/{x}/{y}.png'

        >>> xyz.CartoDB.DarkMatter.build_url(x=9, y=11, z=5)
        'https://a.basemaps.cartocdn.com/dark_all/5/9/11.png'

        >>> xyz.CartoDB.DarkMatter.build_url(x=9, y=11, z=5, scale_factor="@2x")
        'https://a.basemaps.cartocdn.com/dark_all/5/9/11@2x.png'

        >>> xyz.MapBox.build_url(accessToken="my_token")
        'https://api.mapbox.com/styles/v1/mapbox/streets-v11/tiles/{z}/{x}/{y}?access_token=my_token'

        Nz{x}z{y}z{z}zToken is required for this provider, but not provided. You can either update TileProvider or pass respective keywords to build_url().ra   rr   
subdomainsabcr   z{s})xyzsrt   r4   )rr   rn   r:   rP   popformat)r   rw   rx   ry   scale_factorfill_subdomainrh   r0   ra   rt   ru   rz   s               r   	build_urlzTileProvider.build_url  s    j 99;9A9A9A""$"  ll5!ALLd#S"%A!lE:J1AAszz>Aa1>X>>r   c                    | j                         D ]+  \  }}t        |t              sd|v s|| j                  v s+ y y)al  
        Returns ``True`` if the TileProvider requires access token to fetch tiles.

        The token attribute name vary and some :class:`TileProvider` objects may require
        more than one token (e.g. ``HERE``). The information is deduced from the
        presence of `'<insert your...'` string in one or more of attributes. When
        specifying a placeholder for the access token, please use the ``"<insert your
        access token here>"`` string to ensure that
        :meth:`~xyzservices.TileProvider.requires_token` method works properly.

        Returns
        -------
        bool

        Examples
        --------
        >>> import xyzservices.providers as xyz
        >>> xyz.MapBox.requires_token()
        True

        >>> xyz.CartoDB.Positron
        False

        We can specify this API key by calling the object or overriding the attribute.
        Overriding the attribute will alter existing object:

        >>> xyz.OpenWeatherMap.Clouds["apiKey"] = "my-private-api-key"

        Calling the object will return a copy:

        >>> xyz.OpenWeatherMap.Clouds(apiKey="my-private-api-key")


        z<insert yourTF)rF   r   r!   ra   )r   r   vals      r   r:   zTileProvider.requires_token  sB    H 

 	 HC#s##(=$((?	  r   c                :    d| j                         v r| d   S | d   S )Nhtml_attributionrb   r   r   s    r   r   zTileProvider.html_attribution  s'    ,*++M""r   c                    d}| j                         D ]  \  }}|dk7  s|d| d| dz  } |rdn	dt         d}d| d	| j                   d
| d}|S )Nr   r.   z
<dt><span>z</span></dt><dd>z</dd>r   r   r   z
            <div class="xyz-wrap">
                <div class="xyz-header">
                    <div class="xyz-obj">xyzservices.TileProvider</div>
                    <div class="xyz-name">z</div>
                </div>
                <div class="xyz-details">
                    <dl class="xyz-attrs">
                        z\
                    </dl>
                </div>
            </div>
        </div>
        )rF   r%   r.   )r   r   provider_infor   r   r*   r+   s          r   r$   zTileProvider._repr_html_  s    

 	NHCf}:cU2B3%u!MM	N GI;h"?		 + ,099+ 6 ' (	" r   c           	        d}t        j                  t        j                  j	                  | dt        |       d            }|D ]  }|d   |k(  s n t        d| d      |d   }t        j                  t        j                  j	                  | d|             } | |d   |d	   |j                  d
      |j                  d      |j                  d            S )a  
        Creates a :class:`TileProvider` object based on the definition from
        the `Quick Map Services <https://qms.nextgis.com/>`__ open catalog.

        Parameters
        ----------
        name : str
            Service name

        Returns
        -------
        :class:`TileProvider`

        Examples
        --------
        >>> from xyzservices.lib import TileProvider
        >>> provider = TileProvider.from_qms("OpenTopoMap")
        z*https://qms.nextgis.com/api/v1/geoservicesz	/?search=z	&type=tmsr.   z	Service 'z' not found.id/ra   z_minz_maxcopyright_text)r.   ra   min_zoommax_zoomrb   )jsonloadurllibrequesturlopenr   rP   get)clsr.   QMS_API_URLservicesservice
service_idservice_detailss          r   from_qmszTileProvider.from_qms"  s    ( C99NN""k])E$K=	#RS
   	=Gv$&	= yl;<<T]
))NN""k]!J<#@A
  (&$((1$((1'++,<=
 	
r   )rV   r    )NNNNT)rw   Optional[Union[int, str]]rx   r   ry   r   r}   rX   r~   rY   rV   r!   )rV   boolrU   rZ   )r[   r\   r]   r^   re   rp   rr   r   r:   propertyr   r$   classmethodr   __classcell__)rl   s   @r   r    r    +  s    ;z&
 (,'+'+&*)-U?$U? %U? %	U?
 $U? 'U? 
U?n(T # #
4 *
 *
r   r    c                   t        j                  |       }t               }|j                         D ]Q  }||   }d|j                         v rt	        |      ||<   )t        |D ci c]  }|t	        ||          c}      ||<   S |S c c}w )Nra   )r   loadsr
   r   r    )fdata	providersprovider_namer0   is         r   
_load_jsonr   P  s    ::a=DI 	&HMMO#'3H'=Im$ (-7?@!L!--@(Im$	  As   "B
u8  
/* CSS stylesheet for displaying xyzservices objects in Jupyter.*/
.xyz-wrap {
    --xyz-border-color: var(--jp-border-color2, #ddd);
    --xyz-font-color2: var(--jp-content-font-color2, rgba(128, 128, 128, 1));
    --xyz-background-color-white: var(--jp-layout-color1, white);
    --xyz-background-color: var(--jp-layout-color2, rgba(128, 128, 128, 0.1));
}

html[theme=dark] .xyz-wrap,
body.vscode-dark .xyz-wrap,
body.vscode-high-contrast .xyz-wrap {
    --xyz-border-color: #222;
    --xyz-font-color2: rgba(255, 255, 255, 0.54);
    --xyz-background-color-white: rgba(255, 255, 255, 1);
    --xyz-background-color: rgba(255, 255, 255, 0.05);

}

.xyz-header {
    padding-top: 6px;
    padding-bottom: 6px;
    margin-bottom: 4px;
    border-bottom: solid 1px var(--xyz-border-color);
}

.xyz-header>div {
    display: inline;
    margin-top: 0;
    margin-bottom: 0;
}

.xyz-obj,
.xyz-name {
    margin-left: 2px;
    margin-right: 10px;
}

.xyz-obj {
    color: var(--xyz-font-color2);
}

.xyz-attrs {
    grid-column: 1 / -1;
}

dl.xyz-attrs {
    padding: 0 5px 0 5px;
    margin: 0;
    display: grid;
    grid-template-columns: 135px auto;
    background-color: var(--xyz-background-color);
}

.xyz-attrs dt,
dd {
    padding: 0;
    margin: 0;
    float: left;
    padding-right: 10px;
    width: auto;
}

.xyz-attrs dt {
    font-weight: normal;
    grid-column: 1;
}

.xyz-attrs dd {
    grid-column: 2;
    white-space: pre-wrap;
    word-break: break-all;
}

.xyz-details ul>li>label>span {
    color: var(--xyz-font-color2);
    padding-left: 10px;
}

.xyz-inside {
    display: none;
}

.xyz-checkbox:checked~.xyz-inside {
    display: contents;
}

.xyz-collapsible li>input {
    display: none;
}

.xyz-collapsible>li>label {
    cursor: pointer;
}

.xyz-collapsible>li>label:hover {
    color: var(--xyz-font-color2);
}

ul.xyz-collapsible {
    list-style: none!important;
    padding-left: 20px!important;
}

.xyz-checkbox+label:before {
    content: '►';
    font-size: 11px;
}

.xyz-checkbox:checked+label:before {
    content: '▼';
}

.xyz-wrap {
    margin-bottom: 10px;
}
)r^   
__future__r   r   r"   urllib.requestr   typingr   r   r   urllib.parser   r!   	maketransrO   rW   r
   r    r   r%   )rw   s   0r   <module>r      ss    #    , , x'@!2'@A YPD YPxb
5 b
J	(t	q (As   
A!