-
Notifications
You must be signed in to change notification settings - Fork 92
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #281 from cleder/develop
1.0 alpha 8
- Loading branch information
Showing
23 changed files
with
3,372 additions
and
2,746 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,274 @@ | ||
| """Container classes for KML elements.""" | ||
| import logging | ||
| import urllib.parse as urlparse | ||
| from typing import Any | ||
| from typing import Dict | ||
| from typing import Iterable | ||
| from typing import Iterator | ||
| from typing import List | ||
| from typing import Optional | ||
| from typing import Union | ||
|
|
||
| from fastkml import atom | ||
| from fastkml import gx | ||
| from fastkml.data import ExtendedData | ||
| from fastkml.data import Schema | ||
| from fastkml.enums import Verbosity | ||
| from fastkml.features import Placemark | ||
| from fastkml.features import Snippet | ||
| from fastkml.features import _Feature | ||
| from fastkml.geometry import LinearRing | ||
| from fastkml.geometry import LineString | ||
| from fastkml.geometry import MultiGeometry | ||
| from fastkml.geometry import Point | ||
| from fastkml.geometry import Polygon | ||
| from fastkml.styles import Style | ||
| from fastkml.styles import StyleMap | ||
| from fastkml.styles import StyleUrl | ||
| from fastkml.times import TimeSpan | ||
| from fastkml.times import TimeStamp | ||
| from fastkml.types import Element | ||
| from fastkml.views import Camera | ||
| from fastkml.views import LookAt | ||
| from fastkml.views import Region | ||
|
|
||
| logger = logging.getLogger(__name__) | ||
|
|
||
| KmlGeometry = Union[ | ||
| Point, | ||
| LineString, | ||
| LinearRing, | ||
| Polygon, | ||
| MultiGeometry, | ||
| gx.MultiTrack, | ||
| gx.Track, | ||
| ] | ||
|
|
||
|
|
||
| class _Container(_Feature): | ||
| """ | ||
| abstract element; do not create | ||
| A Container element holds one or more Features and allows the | ||
| creation of nested hierarchies. | ||
| subclasses are: | ||
| Document, | ||
| Folder. | ||
| """ | ||
|
|
||
| _features: Optional[List[_Feature]] | ||
|
|
||
| def __init__( | ||
| self, | ||
| ns: Optional[str] = None, | ||
| name_spaces: Optional[Dict[str, str]] = None, | ||
| id: Optional[str] = None, | ||
| target_id: Optional[str] = None, | ||
| name: Optional[str] = None, | ||
| visibility: Optional[bool] = None, | ||
| isopen: Optional[bool] = None, | ||
| atom_link: Optional[atom.Link] = None, | ||
| atom_author: Optional[atom.Author] = None, | ||
| address: Optional[str] = None, | ||
| phone_number: Optional[str] = None, | ||
| snippet: Optional[Snippet] = None, | ||
| description: Optional[str] = None, | ||
| view: Optional[Union[Camera, LookAt]] = None, | ||
| times: Optional[Union[TimeSpan, TimeStamp]] = None, | ||
| style_url: Optional[StyleUrl] = None, | ||
| styles: Optional[Iterable[Union[Style, StyleMap]]] = None, | ||
| region: Optional[Region] = None, | ||
| extended_data: Optional[ExtendedData] = None, | ||
| # Container specific | ||
| features: Optional[List[_Feature]] = None, | ||
| ) -> None: | ||
| super().__init__( | ||
| ns=ns, | ||
| name_spaces=name_spaces, | ||
| id=id, | ||
| target_id=target_id, | ||
| name=name, | ||
| visibility=visibility, | ||
| isopen=isopen, | ||
| atom_link=atom_link, | ||
| atom_author=atom_author, | ||
| address=address, | ||
| phone_number=phone_number, | ||
| snippet=snippet, | ||
| description=description, | ||
| view=view, | ||
| times=times, | ||
| style_url=style_url, | ||
| styles=styles, | ||
| region=region, | ||
| extended_data=extended_data, | ||
| ) | ||
| self._features = features or [] | ||
|
|
||
| def features(self) -> Iterator[_Feature]: | ||
| """Iterate over features.""" | ||
| assert self._features is not None | ||
| yield from self._features | ||
|
|
||
| def etree_element( | ||
| self, | ||
| precision: Optional[int] = None, | ||
| verbosity: Verbosity = Verbosity.normal, | ||
| ) -> Element: | ||
| element = super().etree_element(precision=precision, verbosity=verbosity) | ||
| for feature in self.features(): | ||
| element.append(feature.etree_element()) | ||
| return element | ||
|
|
||
| def append(self, kmlobj: _Feature) -> None: | ||
| """Append a feature.""" | ||
| if kmlobj is self: | ||
| msg = "Cannot append self" | ||
| raise ValueError(msg) | ||
| assert self._features is not None | ||
| self._features.append(kmlobj) | ||
|
|
||
| @classmethod | ||
| def _get_kwargs( | ||
| cls, | ||
| *, | ||
| ns: str, | ||
| name_spaces: Optional[Dict[str, str]] = None, | ||
| element: Element, | ||
| strict: bool, | ||
| ) -> Dict[str, Any]: | ||
| kwargs = super()._get_kwargs( | ||
| ns=ns, | ||
| name_spaces=name_spaces, | ||
| element=element, | ||
| strict=strict, | ||
| ) | ||
| kwargs["features"] = [] | ||
| folders = element.findall(f"{ns}Folder") | ||
| kwargs["features"] += [ | ||
| Folder.class_from_element(ns=ns, element=folder, strict=strict) | ||
| for folder in folders | ||
| ] | ||
| placemarks = element.findall(f"{ns}Placemark") | ||
| kwargs["features"] += [ | ||
| Placemark.class_from_element(ns=ns, element=placemark, strict=strict) | ||
| for placemark in placemarks | ||
| ] | ||
| documents = element.findall(f"{ns}Document") | ||
| kwargs["features"] += [ | ||
| Document.class_from_element(ns=ns, element=document, strict=strict) | ||
| for document in documents | ||
| ] | ||
| return kwargs | ||
|
|
||
|
|
||
| class Folder(_Container): | ||
| """ | ||
| A Folder is used to arrange other Features hierarchically | ||
| (Folders, Placemarks, #NetworkLinks, or #Overlays). | ||
| """ | ||
|
|
||
| __name__ = "Folder" | ||
|
|
||
|
|
||
| class Document(_Container): | ||
| """ | ||
| A Document is a container for features and styles. This element is | ||
| required if your KML file uses shared styles or schemata for typed | ||
| extended data. | ||
| """ | ||
|
|
||
| __name__ = "Document" | ||
| _schemata: Optional[List[Schema]] | ||
|
|
||
| def __init__( | ||
| self, | ||
| ns: Optional[str] = None, | ||
| name_spaces: Optional[Dict[str, str]] = None, | ||
| id: Optional[str] = None, | ||
| target_id: Optional[str] = None, | ||
| name: Optional[str] = None, | ||
| visibility: Optional[bool] = None, | ||
| isopen: Optional[bool] = None, | ||
| atom_link: Optional[atom.Link] = None, | ||
| atom_author: Optional[atom.Author] = None, | ||
| address: Optional[str] = None, | ||
| phone_number: Optional[str] = None, | ||
| snippet: Optional[Snippet] = None, | ||
| description: Optional[str] = None, | ||
| view: Optional[Union[Camera, LookAt]] = None, | ||
| times: Optional[Union[TimeSpan, TimeStamp]] = None, | ||
| style_url: Optional[StyleUrl] = None, | ||
| styles: Optional[Iterable[Union[Style, StyleMap]]] = None, | ||
| region: Optional[Region] = None, | ||
| extended_data: Optional[ExtendedData] = None, | ||
| features: Optional[List[_Feature]] = None, | ||
| schemata: Optional[Iterable[Schema]] = None, | ||
| ) -> None: | ||
| super().__init__( | ||
| ns=ns, | ||
| name_spaces=name_spaces, | ||
| id=id, | ||
| target_id=target_id, | ||
| name=name, | ||
| visibility=visibility, | ||
| isopen=isopen, | ||
| atom_link=atom_link, | ||
| atom_author=atom_author, | ||
| address=address, | ||
| phone_number=phone_number, | ||
| snippet=snippet, | ||
| description=description, | ||
| view=view, | ||
| times=times, | ||
| style_url=style_url, | ||
| styles=styles, | ||
| region=region, | ||
| extended_data=extended_data, | ||
| features=features, | ||
| ) | ||
| self._schemata = list(schemata) if schemata else [] | ||
|
|
||
| def schemata(self) -> Iterator[Schema]: | ||
| if self._schemata: | ||
| yield from self._schemata | ||
|
|
||
| def append_schema(self, schema: Schema) -> None: | ||
| assert self._schemata is not None | ||
| self._schemata.append(schema) | ||
|
|
||
| def etree_element( | ||
| self, | ||
| precision: Optional[int] = None, | ||
| verbosity: Verbosity = Verbosity.normal, | ||
| ) -> Element: | ||
| element = super().etree_element(precision=precision, verbosity=verbosity) | ||
| if self._schemata is not None: | ||
| for schema in self._schemata: | ||
| element.append(schema.etree_element()) | ||
| return element | ||
|
|
||
| def get_style_by_url(self, style_url: str) -> Optional[Union[Style, StyleMap]]: | ||
| id_ = urlparse.urlparse(style_url).fragment | ||
| return next((style for style in self.styles() if style.id == id_), None) | ||
|
|
||
| @classmethod | ||
| def _get_kwargs( | ||
| cls, | ||
| *, | ||
| ns: str, | ||
| name_spaces: Optional[Dict[str, str]] = None, | ||
| element: Element, | ||
| strict: bool, | ||
| ) -> Dict[str, Any]: | ||
| kwargs = super()._get_kwargs( | ||
| ns=ns, | ||
| name_spaces=name_spaces, | ||
| element=element, | ||
| strict=strict, | ||
| ) | ||
| schemata = element.findall(f"{ns}Schema") | ||
| kwargs["schemata"] = [ | ||
| Schema.class_from_element(ns=ns, element=schema, strict=strict) | ||
| for schema in schemata | ||
| ] | ||
| return kwargs |
Oops, something went wrong.