Context Navigation

Ticket #25072: 25072.diff

File 25072.diff, 1.3 KB (added by Claude Paroz, 9 years ago)
Allow GDALRaster garbage collection

diff --git a/django/contrib/gis/gdal/raster/band.py b/django/contrib/gis/gdal/raster/band.py
index f1eb50e..19f39e7 100644

-              a
+import weakref
 from ctypes import byref, c_int
 from django.contrib.gis.gdal.base import GDALBase
-…
+               class GDALBand(GDALBase):
     Wraps a GDAL raster band, needs to be obtained from a GDALRaster object.
     """
     def __init__(self, source, index):
         self.source = source
+        self.source = weakref.proxy(source)
         self._ptr = capi.get_ds_raster_band(source._ptr, index)
     @property

diff --git a/tests/gis_tests/gdal_tests/test_raster.py b/tests/gis_tests/gdal_tests/test_raster.py
index aabec06..057329e 100644

                class GDALBandTests(unittest.TestCase):
     def setUp(self):
         self.rs_path = os.path.join(os.path.dirname(upath(__file__)),
                                '../data/rasters/raster.tif')
         rs = GDALRaster(self.rs_path)
         self.band = rs.bands[0]
+        self.rs = GDALRaster(self.rs_path)
+        self.band = self.rs.bands[0]
     def test_band_data(self):
         self.assertEqual(self.band.width, 163)