swoop_api_payload_test.py

import json
import requests
import random
import string

SWOOP_API_HOST = input("Enter SWOOP API Host (localhost:8000): ")
STAC_ASSET_BUCKET_NAME = input("Enter STAC ASSETS BUCKET NAME: ")

PAYLOAD = {
	"inputs": {
		"payload": {
			"id": "test",
			"type": "FeatureCollection",
			"features": [
				{
					"id": "tx_m_2609719_se_14_060_20201217",
					"bbox": [
						-97.690252,
						26.622563,
						-97.622203,
						26.689923
					],
					"type": "Feature",
					"links": [
						{
							"rel": "collection",
							"type": "application/json",
							"href": "https://planetarycomputer.microsoft.com/api/stac/v1/collections/naip"
						},
						{
							"rel": "parent",
							"type": "application/json",
							"href": "https://planetarycomputer.microsoft.com/api/stac/v1/collections/naip"
						},
						{
							"rel": "root",
							"type": "application/json",
							"href": "https://planetarycomputer.microsoft.com/api/stac/v1/"
						},
						{
							"rel": "self",
							"type": "application/geo+json",
							"href": "https://planetarycomputer.microsoft.com/api/stac/v1/collections/naip/items/tx_m_2609719_se_14_060_20201217"
						},
						{
							"rel": "preview",
							"href": "https://planetarycomputer.microsoft.com/api/data/v1/item/map?collection=naip&item=tx_m_2609719_se_14_060_20201217",
							"title": "Map of item",
							"type": "text/html"
						}
					],
					"assets": {
						"image": {
							"href": "https://naipeuwest.blob.core.windows.net/naip/v002/tx/2020/tx_060cm_2020/26097/m_2609719_se_14_060_20201217.tif",
							"type": "image/tiff; application=geotiff; profile=cloud-optimized",
							"roles": [
								"data"
							],
							"title": "RGBIR COG tile",
							"eo:bands": [
								{
									"name": "Red",
									"common_name": "red"
								},
								{
									"name": "Green",
									"common_name": "green"
								},
								{
									"name": "Blue",
									"common_name": "blue"
								},
								{
									"name": "NIR",
									"common_name": "nir",
									"description": "near-infrared"
								}
							]
						},
						"thumbnail": {
							"href": "https://naipeuwest.blob.core.windows.net/naip/v002/tx/2020/tx_060cm_2020/26097/m_2609719_se_14_060_20201217.200.jpg",
							"type": "image/jpeg",
							"roles": [
								"thumbnail"
							],
							"title": "Thumbnail"
						}
					},
					"geometry": {
						"type": "Polygon",
						"coordinates": [
							[
								[
									-97.623004,
									26.622563
								],
								[
									-97.622203,
									26.689286
								],
								[
									-97.68949,
									26.689923
								],
								[
									-97.690252,
									26.623198
								],
								[
									-97.623004,
									26.622563
								]
							]
						]
					},
					"collection": "naip",
					"properties": {
						"gsd": 0.6,
						"datetime": "2020-12-17T00:00:00Z",
						"naip:year": "2020",
						"proj:bbox": [
							630384,
							2945370,
							637080,
							2952762
						],
						"proj:epsg": 26914,
						"naip:state": "tx",
						"proj:shape": [
							12320,
							11160
						],
						"proj:transform": [
							0.6,
							0,
							630384,
							0,
							-0.6,
							2952762,
							0,
							0,
							1
						]
					},
					"stac_extensions": [
						"https://stac-extensions.github.io/eo/v1.0.0/schema.json",
						"https://stac-extensions.github.io/projection/v1.0.0/schema.json"
					],
					"stac_version": "1.0.0"
				}
			],
			"process": [{
                "description": "string",
                "workflow": "mirror",
				"upload_options": {
					"path_template": "s3://%s/data/${collection}/${id}/" % STAC_ASSET_BUCKET_NAME,
					"collections": {
						"naip": "*"
					},
					"public_assets": [],
					"s3_urls": False
				},
				"tasks": {
					"copy-assets": {
						"assets": [
							"thumbnail"
						],
						"drop_assets": [
							"image"
						]
					},
					"publish": {
						"public": False,
						"stac_validate": True
					}
				}
			}]
		}
	},
	"response": "document"
}

TEST_ID = "".join(random.choices(string.ascii_letters, k=10))
print(f'TEST ID: {TEST_ID}')
PAYLOAD["inputs"]["payload"]["id"] = TEST_ID
PAYLOAD["inputs"]["payload"]["features"][0]["id"] = TEST_ID
content=json.dumps(PAYLOAD)
print('***** INPUTS SUMMARY *****')
print(f'SWOOP API Host: {SWOOP_API_HOST}')
print(f'STAC ASSETS BUCKET NAME: {STAC_ASSET_BUCKET_NAME}')
print(f'POST URL: http://{SWOOP_API_HOST}/processes/mirror/execution')
print(f'POST Input payload:\n{content}')
print('**************************')
x = requests.post(f'http://{SWOOP_API_HOST}/processes/mirror/execution', json = PAYLOAD)
print('**** RESPONSE SUMMARY ****')
print(f'Response from POST to http://{SWOOP_API_HOST}/processes/mirror/execution:\n', x.text)
print('**************************')