import requests
import json

 # 设置缓存时间
data = json.dumps({
    "coverage": {
        "metadata": {
            "entry": [
                {
                    "@key": "cacheAgeMax",
                    "$": "1800"
                },
                {
                    "@key": "cachingEnabled",
                    "$": "true"
                }
            ]
        }
    }
})
resp = requests.put(
    f"http://localhost:1231/geoserver/rest/workspaces/seaice/coveragestores/{name}/coverages/{name}.json",
    data, headers={'Content-Type': 'application/json'}, auth=('admin', 'geoserver'))
print(resp)

from geoserver.catalog import Catalog
import os
import glob
import requests
import json


def AutoPublicWMSLayer():
    """
    Python批量发布WMS服务
    """
    server_url = 'http://localhost:9090/geoserver'
    workspace = 'seaice'
    username = "admin"
    password = "geoserver"
    for tifFile in glob.glob(f'{tif_out_dir}/*.tif'):
        # 示例  ./data/date/Tif/ice_thickness_1020.tif
        name = os.path.splitext(os.path.basename(tifFile))[0]
        # 这里提前在geoserver设置了ice_thickness等样式，这里是根据图层名称取对于样式
        style = '_'.join(name.split('_')[:-1])
        print(name)
        cat = Catalog(f"{server_url}/rest/", username=username, password=password)
        cat.create_coveragestore_external_geotiff(name, tifFile, workspace=cat.get_workspace(workspace),
                                                  overwrite=True)
        # 设置默认样式
        response = requests.put(f'{server_url}/rest/layers/{workspace}:{name}',
                                headers={'Content-Type': 'application/xml'},
                                data=f"<layer><defaultStyle><name>{workspace}:{style}</name></defaultStyle><queryable>false"
                                     f"</queryable></layer>",
                                auth=(username, password))
        print(response)
        # 设置缓存时间
        data = json.dumps({
            "coverage": {
                "metadata": {
                    "entry": [
                        {
                            "@key": "cacheAgeMax",
                            "$": "1800"
                        },
                        {
                            "@key": "cachingEnabled",
                            "$": "true"
                        }
                    ]
                }
            }
        })
        resp = requests.put(
            f"{server_url}/rest/workspaces/{workspace}/coveragestores/{name}/coverages/{name}.json",
            data, headers={'Content-Type': 'application/json'}, auth=(username, password))
        print(resp)



def AutoDeleteWMSLayer():
    """
    Python批量删除WMS服务
    """
    server_url = 'http://localhost:9090/geoserver'
    workspace = 'seaice'
    username = "admin"
    password = "geoserver"
    for tifFile in glob.glob(f'{tif_out_dir}/*.tif'):
        name = os.path.splitext(os.path.basename(tifFile))[0]
        try:
            url = f"{server_url}/rest/workspaces/{workspace}/coveragestores/{name}.xml?recurse=true"
            response = requests.delete(url, auth=(username, password))
            if response.status_code != requests.codes.ok:
                raise Exception(response.status_code)
        except Exception as e:
            print(f"ERROR:{name}>>{e}")

if __name__ == '__main__':
    tif_out_dir = '../data/date/Tif'
    AutoPublicWMSLayer()
    # AutoDeleteWMSLayer()