From 18326d547a70db6e23d7fdb7b421ecb300d6d58e Mon Sep 17 00:00:00 2001
From: Corey White <ctwhite48@gmail.com>
Date: Mon, 18 Sep 2023 22:11:37 -0400
Subject: [PATCH] added get items function

---
 src/temporal/t.stac.import/t.stac.import.py   | 62 ++++++++++++++-----
 .../testsuite/test_t_stac_import.py           | 11 ++++
 2 files changed, 58 insertions(+), 15 deletions(-)

diff --git a/src/temporal/t.stac.import/t.stac.import.py b/src/temporal/t.stac.import/t.stac.import.py
index e0368cb623..446bc94604 100644
--- a/src/temporal/t.stac.import/t.stac.import.py
+++ b/src/temporal/t.stac.import/t.stac.import.py
@@ -252,12 +252,16 @@ def validate_collections_option(client, collections=[]):
         boolean: Returns true if the collection is avaliable.
     """
     avaliable_collections = client.get_collections()
-    if collections in avaliable_collections:
+    avaliable_collections_ids = [c.id for c in list(avaliable_collections)]
+
+    gs.warning(_(f"Avaliable Collections: {avaliable_collections_ids}"))
+
+    if collections in avaliable_collections_ids:
         return True
-    gs.warning(_("The specified collections do not exisit."))
 
-    for collection in avaliable_collections:
-        gs.warning(_(f"{collection} collection found"))
+    for collection in collections:
+        if collection not in avaliable_collections_ids:
+            gs.warning(_(f"{collection} collection not found"))
 
     return False
 
@@ -301,15 +305,40 @@ def get_collection_items(client, collection_name):
     #     gs.message(_(i.id))
 
 
+def import_items(
+    items,
+    region_params=None,
+    reprojection=None,
+    output=None,
+    strds_output=None,
+    method=None,
+    resolution=None,
+    memory=None,
+):
+    """Import items"""
+    for item in items:
+        gs.message(_(f"Item: {item.id}"))
+        gs.message(_(f"Spatial Extent: {item.geometry}"))
+        gs.message(_(f"Temporal Extent: {item.datetime}"))
+        gs.message(_(f"Assets: {item.assets}"))
+        gs.message(_(f"Links: {item.links}"))
+        gs.message(_(f"Properties: {item.properties}"))
+        gs.message(_(f"Collection ID: {item.collection_id}"))
+
+
 def main():
     """Main function"""
 
     client_url = options["url"]
     ids = options["ids"]  # optional
     collections = options["collections"]  # Maybe limit to one?
-    limit = options["limit"]  # optional
-    max_items = options["max_items"]  # optional
+    limit = int(options["limit"])  # optional
+    max_items = int(options["max_items"])  # optional
     bbox = options["bbox"]  # optional
+
+    # if bbox:
+    #     bbox = [float(i) for i in bbox.strip("[]").split(",")]
+
     intersects = options["intersects"]  # optional
     datetime = options["datetime"]  # optional
     query = options["query"]  # optional
@@ -331,19 +360,22 @@ def main():
         return None
 
     if validate_collections_option(client, collections):
-        items = search_stac_api(
+        items_search = search_stac_api(
             client=client,
-            ids=ids,
+            collections=collections,
             limit=limit,
             max_items=max_items,
-            bbox=bbox,
-            intersects=intersects,
-            datetime=datetime,
-            query=query,
-            filter=filter,
-            filter_lang=filter_lang,
+            # bbox=[-72.5, 40.5, -72.0, 41.0],
+            # intersects=intersects,
+            # datetime=datetime,
+            # query=query,
+            # filter=filter,
+            # filter_lang=filter_lang,
         )
-        print(items)
+        gs.message(_("Import Items..."))
+        import_items(list(items_search.items()))
+
+    # if reprojection:
 
 
 if __name__ == "__main__":
diff --git a/src/temporal/t.stac.import/testsuite/test_t_stac_import.py b/src/temporal/t.stac.import/testsuite/test_t_stac_import.py
index 02bd43f014..ce186c5e86 100644
--- a/src/temporal/t.stac.import/testsuite/test_t_stac_import.py
+++ b/src/temporal/t.stac.import/testsuite/test_t_stac_import.py
@@ -85,6 +85,17 @@ def test_validate_collection(self):
         isValid = in_stac_lib.validate_collections_option(client)
         self.assertFalse(isValid)
 
+    def test_import_asset(self):
+        """
+        Tests importing STAC asset
+        """
+        self.assertModule(
+            "t.stac.import",
+            url=EARTH_SEARCH_API_URL,
+            collections=["naip"],
+            bbox="-72.5,40.5,-72,41",
+        )
+
 
 if __name__ == "__main__":
     test()