[Raster.__init__] create auto names if band names not in HDR file

johntruckenbrodt · Jan 25, 2019 · b981d76 · b981d76
1 parent b2f94cc
commit b981d76
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spatialist/raster.py b/spatialist/raster.py
@@ -59,7 +59,11 @@ def __init__(self, filename):
         self.__data = [None] * self.bands
 
         if self.format == 'ENVI':
-            self.bandnames = HDRobject(self.filename + '.hdr').band_names
+            with HDRobject(self.filename + '.hdr') as hdr:
+                if hasattr(hdr, 'band_names'):
+                    self.bandnames = hdr.band_names
+                else:
+                    self.bandnames = ['band{}'.format(x) for x in range(1, self.bands + 1)]
         else:
             self.bandnames = ['band{}'.format(x) for x in range(1, self.bands + 1)]