134 lines
4.2 KiB
Python
134 lines
4.2 KiB
Python
import os
|
|
|
|
import requests
|
|
|
|
|
|
class GeoServer(object):
|
|
|
|
def __init__(self, url, name, password, workspace):
|
|
self.url = url
|
|
self.credential = (name, password)
|
|
self.workspace = workspace
|
|
self.geoserver_data_directory = '/var/lib/tomcat8/webapps/geoserver/data'
|
|
|
|
def create_and_publish_geotiff(self, file_path):
|
|
if not file_path:
|
|
return
|
|
name = os.path.splitext(os.path.basename(file_path))[0]
|
|
|
|
response = self.create_coveragestorage(name, file_path)
|
|
response.raise_for_status()
|
|
response = self.publish_geotiff(name)
|
|
response.raise_for_status()
|
|
return name
|
|
|
|
def create_and_publish_shp(self, file_path):
|
|
if not file_path:
|
|
return
|
|
name = os.path.splitext(os.path.basename(file_path))[0]
|
|
response = self.create_datastorage(name, file_path)
|
|
response.raise_for_status()
|
|
response = self.publish_shp(name, file_path)
|
|
response.raise_for_status()
|
|
return name
|
|
|
|
def publish_shp(self, name, file_path):
|
|
full_file_path = os.path.join(self.geoserver_data_directory, file_path)
|
|
print(full_file_path)
|
|
payload = 'file:%s' % (full_file_path,)
|
|
headers = {
|
|
'content-type': 'text/plain'
|
|
}
|
|
|
|
request_url = '%s/rest/workspaces/%s/datastores/%s/external.shp' % (self.url, self.workspace, name)
|
|
|
|
r = requests.put(
|
|
request_url,
|
|
data=payload,
|
|
headers=headers,
|
|
auth=self.credential
|
|
)
|
|
print(r.status_code)
|
|
print(r.text)
|
|
return r
|
|
|
|
def create_datastorage(self, name, file_path):
|
|
payload = {
|
|
'dataStore': {
|
|
'name': name,
|
|
'connectionParameters': {
|
|
'entry': [
|
|
{'@key': 'url', '$': 'file:%s' % (file_path,)}
|
|
]
|
|
}
|
|
}
|
|
}
|
|
headers = {
|
|
'content-type': 'application/json'
|
|
}
|
|
create_datastorage_url = '%s/rest/workspaces/%s/datastores' % (self.url, self.workspace)
|
|
r = requests.post(create_datastorage_url,
|
|
json=payload,
|
|
headers=headers,
|
|
auth=self.credential)
|
|
|
|
return r
|
|
|
|
def create_coveragestorage(self, name, file_path):
|
|
payload = {
|
|
'coverageStore': {
|
|
'name': name,
|
|
'workspace': self.workspace,
|
|
'type': 'GeoTIFF',
|
|
'enabled': 'true',
|
|
'url': 'file:%s' % (file_path,)
|
|
}
|
|
}
|
|
headers = {
|
|
'content-type': 'application/json'
|
|
}
|
|
|
|
create_coveragestore_url = '%s/rest/workspaces/%s/coveragestores' % (self.url, self.workspace)
|
|
r = requests.post(create_coveragestore_url,
|
|
json=payload,
|
|
headers=headers,
|
|
auth=self.credential)
|
|
return r
|
|
|
|
def publish_geotiff(self, name):
|
|
payload = {
|
|
'coverage': {
|
|
'title': name,
|
|
'name': name,
|
|
'nativeName': name,
|
|
'srs': 'EPSG:4326',
|
|
},
|
|
}
|
|
headers = {
|
|
'content-type': 'application/json'
|
|
}
|
|
publish_geotiff_url = '%s/rest/workspaces/%s/coveragestores/%s/coverages' % (self.url, self.workspace, name)
|
|
r = requests.post(publish_geotiff_url,
|
|
json=payload,
|
|
headers=headers,
|
|
auth=self.credential)
|
|
return r
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# the geotiff file should be named starts with bj_ , such as bj_sample.tiff,
|
|
|
|
gs = GeoServer('http://210.77.68.250:8080/geoserver', 'admin', 'geoserver', 'baoji')
|
|
|
|
for y in range(1979, 2018):
|
|
for m in range(1, 13):
|
|
file_name = "%s-%s-rain.tif" % (y, m)
|
|
print(file_name)
|
|
name = gs.create_and_publish_geotiff('baoji/rain/%s' % (file_name))
|
|
print('coverage storage name :' + name)
|
|
print('layer name :' + name)
|
|
|
|
# name = gs.create_and_publish_shp('baoji/sample/baoji_grid.shp')
|
|
# print('data storage name:' + name)
|
|
# print('layer name:' + name)
|