Merge pull request #243 from emmanuelmathot/feature/eo-bands-assets-only

Get eo bands defined in assets only

Author: lossyrob

pystac/extensions/eo.py

@@ -55,7 +55,8 @@ def get_bands(self, asset=None):
         """Gets an Item or an Asset bands.
 
         If an Asset is supplied and the bands property exists on the Asset,
-        returns the Asset's value. Otherwise returns the Item's value
+        returns the Asset's value. Otherwise returns the Item's value or
+        all the asset's eo bands
 
         Returns:
             List[Band]
@@ -65,6 +66,13 @@ def get_bands(self, asset=None):
         else:
             bands = self.item.properties.get('eo:bands')
 
+        # get assets with eo:bands even if not in item
+        if asset is None and bands is None:
+            bands = []
+            for (key, value) in self.item.get_assets().items():
+                if 'eo:bands' in value.properties:
+                    bands.extend(value.properties.get('eo:bands'))
+
         if bands is not None:
             bands = [Band(b) for b in bands]
 

tests/extensions/test_eo.py

@@ -61,6 +61,10 @@ def test_asset_bands(self):
         asset_bands = eo_item.ext.eo.get_bands(index_asset)
         self.assertIs(None, asset_bands)
 
+        # No asset specified
+        asset_bands = eo_item.ext.eo.get_bands()
+        self.assertIsNot(None, asset_bands)
+
         # Set
         b2_asset = eo_item.assets['B2']
         self.assertEqual(eo_item.ext.eo.get_bands(b2_asset)[0].name, "B2")