Changeset 14437

Timestamp:

05/10/08 17:12:40 (2 months ago)

Author:

rouault

Message:

Add gdaltest.download_file that enables to download test data files if GDAL_DOWNLOAD_TEST_DATA environment variable is defined and cache them for later faster use

Files:

• trunk/autotest/pymod/gdaltest.py (modified) (3 diffs)

trunk/autotest/pymod/gdaltest.py

@@ -31 +31 @@
-import string
-import os
+import urllib2
-try:
-    from osgeo import gdal
@@ -50 +51 @@
-
-argv = gdal.GeneralCmdLineProcessor( sys.argv )
-
-
-###############################################################################
@@ -832 +832 @@
-
-    return 1
+
+
+###############################################################################
+# Download file at url 'url' and put it as 'filename' in 'tmp/cache/'
+#
+# If 'filename' already exits in 'tmp/cache/', it is not downloaded
+# If GDAL_DOWNLOAD_TEST_DATA is not defined, the function fails
+# If GDAL_DOWNLOAD_TEST_DATA is defined, 'url' is downloaded  as 'filename' in 'tmp/cache/'
+
+def download_file(url, filename):
+    try:
+        os.stat( 'tmp/cache/' + filename )
+        return True
+    except:
+        if os.environ.has_key('GDAL_DOWNLOAD_TEST_DATA'):
+            val = None
+            try:
+                handle = urllib2.urlopen(url)
+                print 'Downloading %s...' % (url)
+                val = handle.read()
+            except urllib2.HTTPError, e:
+                print 'HTTP service for %s is down (HTTP Error: %d)' % (url, e.code)
+                return False
+            except:
+                print 'HTTP service for %s is down.' %(url)
+                return False
+
+            try:
+                os.stat( 'tmp/cache' )
+            except:
+                os.mkdir('tmp/cache')
+
+            try:
+                open( 'tmp/cache/' + filename, 'wb').write(val)
+                return True
+            except:
+                print 'Cannot write %s' % (filename)
+                return False
+        else:
+            return False