From bd743adb4f43d66e2caa36a7ab1d6f44465a3a6d Mon Sep 17 00:00:00 2001
From: david-i-berry <dberry@wmo.int>
Date: Fri, 7 Jul 2023 09:24:38 +0200
Subject: [PATCH] pygeoapi plugin added

---
 requirements.txt              |   2 +
 synop2bufr/pygeoapi_plugin.py | 149 ++++++++++++++++++++++++++++++++++
 2 files changed, 151 insertions(+)
 create mode 100644 synop2bufr/pygeoapi_plugin.py

diff --git a/requirements.txt b/requirements.txt
index b743142..b2992b9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,4 @@
+attrs==22.2.0
 click==8.1.3
+numpy==1.21.6
 csv2bufr
diff --git a/synop2bufr/pygeoapi_plugin.py b/synop2bufr/pygeoapi_plugin.py
new file mode 100644
index 0000000..6b44db4
--- /dev/null
+++ b/synop2bufr/pygeoapi_plugin.py
@@ -0,0 +1,149 @@
+###############################################################################
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+###############################################################################
+import logging
+from pygeoapi.process.base import BaseProcessor
+
+from synop2bufr import transform
+
+LOGGER = logging.getLogger(__name__)
+
+PROCESS_METADATA = {
+    "version": "0.1.0",
+    "id": "x-wmo:synop2bufr",
+    "title": {"en": "synop2bufr"},
+    "description": {"en": "Process to convert FM 12-SYNOP bulletin to BUFR"},
+    "keywords": ["SYNOP", "BUFR", "FM 12"],
+    "links": [{
+        "type": "text/html",
+        "rel": "about",
+        "title": "homepage",
+        "href": "https://github.com/wmo-im/synop2bufr",
+        "hreflang": "en-US",
+    }],
+    "inputs": {
+        "data": {
+            "title": "FM 12-SYNOP bulletin string",
+            "description": "Input FM 12-SYNOP bulletin to convert to BUFR.",
+            "minOccurs": 1,
+            "maxOccurs": 1,
+            "metadata": None,
+            "keywords": [],
+        },
+        "metadata":{
+            "title": "Station metadata",
+            "description": "CSV formatted data containing list of stations required by synop2bufr.",  # noqa
+            "schema": {"type": "string"},
+            "minOccurs": 1,
+            "maxOccurs": 1,
+            "metadata": None,
+            "keywords": [],
+        },
+        "year": {
+            "title": "Year",
+            "description": "Year (UTC) corresponding to FM 12-SYNOP bulletin",
+            "schema": {"type": "integer"},
+            "minOccurs": 1,
+            "maxOccurs": 1,
+            "metadata": None,
+            "keywords": []
+        },
+        "month":{
+            "title": "Month",
+            "description": "Month (UTC) corresponding to FM 12-SYNOP bulletin",
+            "schema": {"type": "integer"},
+            "minOccurs": 1,
+            "maxOccurs": 1,
+            "metadata": None,
+            "keywords": []
+        }
+    },
+    "outputs": {
+        "messages": {
+            "title": "BUFR encoded data in base64",  # noqa
+            "schema": {"type": "array"},
+            "description": "One result per line in input data"  # noqa
+        }
+    },
+    "example": {
+        "inputs": {
+            "data":r"AAXX 21121 15015 02999 02501 10103 21090 39765 42952 57020 60001 333 4/000 55310 0//// 22591 3//// 60007 91003 91104=",  # noqa
+            "metadata": r"station_name,wigos_station_identifier,traditional_station_identifier,facility_type,latitude,longitude,elevation,territory_name,wmo_region\nOCNA SUGATAG,0-20000-0-15015,15015,Land (fixed),47.77706163,23.94046026,503,Romania,6",
+            "year": 2022,
+            "month": 02
+        },
+        "output": {"messages": []}
+    },
+}
+
+
+class processor(BaseProcessor):
+    def __init__(self, processor_def):
+        """
+        Initialize object
+        :param processor_def: provider definition
+        :returns: pygeoapi.process.csv2bufr.csv2bufr
+        """
+
+        super().__init__(processor_def, PROCESS_METADATA)
+
+    def execute(self, data):
+        """
+        This method is invoked by pygeoapi when this class is set as a
+        `process` type resource in pygeoapi config file.
+
+        :param data: It is the value of `inputs` passed in payload. e.g.
+        {
+            "inputs": {
+                "data": "csv data to encode",
+                "mappings": "csv2bufr mapping json file"
+            }
+        }
+
+        :return: media_type, json
+        """
+
+        mimetype = "application/json"
+        try:
+            fm12 = data['data']
+            metadata = data['metadata']
+            year = data['year']
+            month = data['month']
+            bufr_generator = transform(data = fm12,
+                                       metadata = metadata,
+                                       year =  year,
+                                       month = month)
+            # transform returns a generator, we need to iterate over
+            # and add to single output object
+            bufr = []
+            for result in bufr_generator:
+                bufr.append(result)
+
+            output = {"messages": bufr}
+
+            LOGGER.error(output)
+        except Exception as e:
+            LOGGER.exception(e)
+            output = {"messages": None, "errors": e}
+
+        return mimetype, output
+
+    def __repr__(self):
+        return "<synop2bufr> {}".format(self.name)