class DataStore(ResourceInfo):
resource_type = "dataStore"
save_method = "PUT"
def __init__(self, catalog, workspace, name):
super(DataStore, self).__init__()
assert isinstance(workspace, ws.Workspace)
assert isinstance(name, str)
self.catalog = catalog
self.workspace = workspace
self.name = name
@property
def href(self):
return url(self.catalog.service_url, [
"workspaces", self.workspace.name, "datastores", self.name + ".xml"
])
enabled = xml_property("enabled", lambda x: x.text == "true")
name = xml_property("name")
type = xml_property("type")
connection_parameters = xml_property("connectionParameters",
key_value_pairs)
writers = dict(enabled=write_bool("enabled"),
name=write_string("name"),
type=write_string("type"),
connectionParameters=write_dict("connectionParameters"))
@property
def resource_url(self):
return url(self.catalog.service_url, [
"workspaces", self.workspace.name, "datastores", self.name,
"featuretypes.xml"
])
def get_resources(self, name=None, available=False):
res_url = self.resource_url
if available:
res_url += "?list=available"
xml = self.catalog.get_xml(res_url)
def ft_from_node(node):
return featuretype_from_index(self.catalog, self.workspace, self,
node)
# if name passed, return only one FeatureType, otherwise return all FeatureTypes in store:
if name is not None:
for node in xml.findall("featureType"):
if node.findtext("name") == name:
return ft_from_node(node)
return None
if available:
return [str(node.text) for node in xml.findall("featureTypeName")]
else:
return [ft_from_node(node) for node in xml.findall("featureType")]
class CoverageStore(ResourceInfo):
resource_type = 'coverageStore'
save_method = "PUT"
def __init__(self, catalog, workspace, name):
super(CoverageStore, self).__init__()
assert isinstance(workspace, ws.Workspace)
assert isinstance(name, str)
self.catalog = catalog
self.workspace = workspace
self.name = name
@property
def href(self):
return url(self.catalog.service_url, [
"workspaces", self.workspace.name, "coveragestores",
self.name + ".xml"
])
enabled = xml_property("enabled", lambda x: x.text == "true")
name = xml_property("name")
url = xml_property("url")
type = xml_property("type")
writers = dict(enabled=write_bool("enabled"),
name=write_string("name"),
url=write_string("url"),
type=write_string("type"),
workspace=write_string("workspace"))
def get_resources(self, name=None):
res_url = url(self.catalog.service_url, [
"workspaces", self.workspace.name, "coveragestores", self.name,
"coverages.xml"
])
xml = self.catalog.get_xml(res_url)
def cov_from_node(node):
return coverage_from_index(self.catalog, self.workspace, self,
node)
# if name passed, return only one Coverage, otherwise return all Coverages in store:
if name is not None:
for node in xml.findall("coverage"):
if node.findtext("name") == name:
return cov_from_node(node)
return None
return [cov_from_node(node) for node in xml.findall("coverage")]
class Workspace(ResourceInfo):
resource_type = "workspace"
def __init__(self, catalog, name):
super(Workspace, self).__init__()
self.catalog = catalog
self.name = name
@property
def href(self):
return url(self.catalog.service_url, ["workspaces", self.name + ".xml"])
@property
def coveragestore_url(self):
return url(self.catalog.service_url, ["workspaces", self.name, "coveragestores.xml"])
@property
def datastore_url(self):
return url(self.catalog.service_url, ["workspaces", self.name, "datastores.xml"])
@property
def wmsstore_url(self):
return "%s/workspaces/%s/wmsstores.xml" % (self.catalog.service_url, self.name)
enabled = xml_property("enabled", lambda x: x.lower() == 'true')
writers = dict(
enabled = write_bool("enabled")
)
def __repr__(self):
return "%s @ %s" % (self.name, self.href)
class WmsStore(ResourceInfo):
resource_type = "wmsStore"
save_method = "PUT"
def __init__(self, catalog, workspace, name, user, password):
super(WmsStore, self).__init__()
assert isinstance(workspace, ws.Workspace)
assert isinstance(name, str)
self.catalog = catalog
self.workspace = workspace
self.name = name
self.metadata = {}
self.metadata['user'] = user
self.metadata['password'] = password
@property
def href(self):
return "%s/workspaces/%s/wmsstores/%s.xml" % (
self.catalog.service_url, self.workspace.name, self.name)
enabled = xml_property("enabled", lambda x: x.text == "true")
name = xml_property("name")
nativeName = xml_property("nativeName")
capabilitiesURL = xml_property("capabilitiesURL")
type = xml_property("type")
metadata = xml_property("metadata", key_value_pairs)
writers = dict(enabled=write_bool("enabled"),
name=write_string("name"),
capabilitiesURL=write_string("capabilitiesURL"),
type=write_string("type"),
metadata=write_dict("metadata"))
def get_resources(self, name=None, available=False):
res_url = "%s/workspaces/%s/wmsstores/%s/wmslayers.xml" % (
self.catalog.service_url, self.workspace.name, self.name)
layer_name_attr = "wmsLayer"
if available:
res_url += "?list=available"
layer_name_attr += 'Name'
xml = self.catalog.get_xml(res_url)
def wl_from_node(node):
return wmslayer_from_index(self.catalog, self.workspace, self,
node)
# if name passed, return only one layer, otherwise return all layers in store:
if name is not None:
for node in xml.findall(layer_name_attr):
if node.findtext("name") == name:
return wl_from_node(node)
return None
if available:
return [str(node.text) for node in xml.findall(layer_name_attr)]
else:
return [
wl_from_node(node) for node in xml.findall(layer_name_attr)
]
class LayerGroup(ResourceInfo):
""" Represents a layer group in geoserver
"""
resource_type = "layerGroup"
save_method = "PUT"
def __init__(self, catalog, name, workspace=None):
super(LayerGroup, self).__init__()
assert isinstance(name, str)
self.catalog = catalog
self.name = name
self.workspace = workspace
# the XML format changed in 2.3.x - the element listing all the layers
# and the entries themselves have changed
if self.catalog.gsversion() == "2.2.x":
parent, element, attributes = "layers", "layer", None
else:
parent, element, attributes = "publishables", "published", {
'type': 'layer'
}
self._layer_parent = parent
self._layer_element = element
self._layer_attributes = attributes
self.writers = dict(name=write_string("name"),
styles=_write_styles,
layers=lambda b, l: _write_layers(
b, l, parent, element, attributes),
bounds=write_bbox("bounds"),
workspace=write_string("workspace"))
@property
def href(self):
path_parts = ["layergroups", self.name + ".xml"]
if self.workspace is not None:
workspace_name = getattr(self.workspace, 'name', self.workspace)
path_parts = ["workspaces", workspace_name] + path_parts
return url(self.catalog.service_url, path_parts)
styles = xml_property("styles", _style_list)
bounds = xml_property("bounds", bbox)
def _layers_getter(self):
if "layers" in self.dirty:
return self.dirty["layers"]
else:
if self.dom is None:
self.fetch()
node = self.dom.find(self._layer_parent)
return _layer_list(
node, self._layer_element) if node is not None else None
def _layers_setter(self, value):
self.dirty["layers"] = value
def _layers_delete(self):
self.dirty["layers"] = None
layers = property(_layers_getter, _layers_setter, _layers_delete)
def __str__(self):
return "<LayerGroup %s>" % self.name
__repr__ = __str__
class WmsLayer(ResourceInfo):
resource_type = "wmsLayer"
save_method = "PUT"
def __init__(self, catalog, workspace, store, name):
super(WmsLayer, self).__init__()
self.catalog = catalog
self.workspace = workspace
self.store = store
self.name = name
@property
def href(self):
return "%s/workspaces/%s/wmsstores/%s/wmslayers/%s.xml" % (
self.catalog.service_url, self.workspace.name, self.store.name,
self.name)
title = xml_property("title")
description = xml_property("description")
abstract = xml_property("abstract")
keywords = xml_property("keywords", string_list)
# nativeCRS
projection = xml_property("srs")
native_bbox = xml_property("nativeBoundingBox", bbox)
latlon_bbox = xml_property("latLonBoundingBox", bbox)
projection_policy = xml_property("projectionPolicy")
enabled = xml_property("enabled", lambda x: x.text == "true")
advertised = xml_property("advertised",
lambda x: x.text == "true",
default=True)
metadata_links = xml_property("metadataLinks", metadata_link_list)
writers = dict(
title=write_string("title"),
description=write_string("description"),
abstract=write_string("abstract"),
keywords=write_string_list("keywords"),
# nativeCRS
srs=write_string("srs"),
nativeBoundingBox=write_bbox("nativeBoundingBox"),
latLonBoundingBox=write_bbox("latLonBoundingBox"),
projectionPolicy=write_string("projectionPolicy"),
enabled=write_bool("enabled"),
advertised=write_bool("advertised"),
metadataLinks=write_metadata_link_list("metadataLinks"))
class Coverage(_ResourceBase):
resource_type = "coverage"
url_part_stores = 'coveragestores'
url_part_types = 'coverages'
title = xml_property("title")
abstract = xml_property("abstract")
enabled = xml_property("enabled")
advertised = xml_property("advertised", default="true")
native_bbox = xml_property("nativeBoundingBox", bbox)
latlon_bbox = xml_property("latLonBoundingBox", bbox)
projection = xml_property("srs")
projection_policy = xml_property("projectionPolicy")
keywords = xml_property("keywords", string_list)
request_srs_list = xml_property("requestSRS", string_list)
response_srs_list = xml_property("responseSRS", string_list)
supported_formats = xml_property("supportedFormats", string_list)
metadata_links = xml_property("metadataLinks", metadata_link_list)
metadata = xml_property("metadata", metadata)
writers = dict(title=write_string("title"),
abstract=write_string("abstract"),
enabled=write_bool("enabled"),
advertised=write_bool("advertised"),
nativeBoundingBox=write_bbox("nativeBoundingBox"),
latLonBoundingBox=write_bbox("latLonBoundingBox"),
srs=write_string("srs"),
projection_policy=write_string("projectionPolicy"),
keywords=write_string_list("keywords"),
metadataLinks=write_metadata_link_list("metadataLinks"),
requestSRS=write_string_list("requestSRS"),
responseSRS=write_string_list("responseSRS"),
supportedFormats=write_string_list("supportedFormats"),
metadata=write_metadata("metadata"))
class FeatureType(_ResourceBase):
resource_type = "featureType"
url_part_stores = 'datastores'
url_part_types = 'featuretypes'
title = xml_property("title")
native_name = xml_property("nativeName")
abstract = xml_property("abstract")
enabled = xml_property("enabled")
advertised = xml_property("advertised", default="true")
native_bbox = xml_property("nativeBoundingBox", bbox)
latlon_bbox = xml_property("latLonBoundingBox", bbox)
projection = xml_property("srs")
projection_policy = xml_property("projectionPolicy")
keywords = xml_property("keywords", string_list)
attributes = xml_property("attributes", attribute_list)
metadata_links = xml_property("metadataLinks", metadata_link_list)
metadata = xml_property("metadata", metadata)
writers = dict(name=write_string("name"),
nativeName=write_string("nativeName"),
title=write_string("title"),
abstract=write_string("abstract"),
enabled=write_bool("enabled"),
advertised=write_bool("advertised"),
nativeBoundingBox=write_bbox("nativeBoundingBox"),
latLonBoundingBox=write_bbox("latLonBoundingBox"),
srs=write_string("srs"),
nativeCRS=write_string("nativeCRS"),
projectionPolicy=write_string("projectionPolicy"),
keywords=write_string_list("keywords"),
metadataLinks=write_metadata_link_list("metadataLinks"),
metadata=write_metadata("metadata"))
class Style(ResourceInfo):
supported_formats = ["sld10", "sld11", "zip"]
content_types = {
"sld10": "application/vnd.ogc.sld+xml",
"sld11": "application/vnd.ogc.se+xml",
"zip": "application/zip"
}
def __init__(self, catalog, name, workspace=None, style_format="sld10"):
super(Style, self).__init__()
assert isinstance(name, str)
assert style_format in Style.supported_formats
self.catalog = catalog
self.workspace = workspace
self.name = name
self.style_format = style_format
self._sld_dom = None
@property
def fqn(self):
return self.name if not self.workspace else '%s:%s' % (self.workspace,
self.name)
@property
def href(self):
return self._build_href('.xml')
@property
def body_href(self):
return self._build_href('.sld')
@property
def create_href(self):
return self._build_href('.xml', True)
@property
def content_type(self):
return Style.content_types[self.style_format]
def _build_href(self, extension, create=False):
path_parts = ["styles"]
query = {}
if not create:
path_parts.append(self.name + extension)
else:
query['name'] = self.name
if self.workspace is not None:
path_parts = [
"workspaces",
getattr(self.workspace, 'name', self.workspace)
] + path_parts
return url(self.catalog.service_url, path_parts, query)
filename = xml_property("filename")
def _get_sld_dom(self):
if self._sld_dom is None:
self._sld_dom = self.catalog.get_xml(self.body_href)
return self._sld_dom
@property
def sld_title(self):
user_style = self._get_sld_dom().find(
"{http://www.opengis.net/sld}NamedLayer/{http://www.opengis.net/sld}UserStyle"
)
if not user_style:
user_style = self._get_sld_dom().find(
"{http://www.opengis.net/sld}UserLayer/{http://www.opengis.net/sld}UserStyle"
)
title_node = None
if user_style:
try:
# it is not mandatory
title_node = user_style.find(
"{http://www.opengis.net/sld}Title")
except:
title_node = None
return title_node.text if title_node is not None else None
@property
def sld_name(self):
user_style = self._get_sld_dom().find(
"{http://www.opengis.net/sld}NamedLayer/{http://www.opengis.net/sld}UserStyle"
)
if not user_style:
user_style = self._get_sld_dom().find(
"{http://www.opengis.net/sld}UserLayer/{http://www.opengis.net/sld}UserStyle"
)
name_node = None
if user_style:
try:
# it is not mandatory
name_node = user_style.find("{http://www.opengis.net/sld}Name")
except:
name_node = None
return name_node.text if name_node is not None else None
@property
def sld_body(self):
content = self.catalog.http.request(self.body_href)[1]
return content
def update_body(self, body):
headers = {"Content-Type": self.content_type}
self.catalog.http.request(self.body_href, "PUT", body, headers)