diff --git a/.gitignore b/.gitignore index c55829b..816f4f2 100644 --- a/.gitignore +++ b/.gitignore @@ -4,7 +4,6 @@ dist env env3 -*.vrt *.egg-info *.pyc diff --git a/setup.py b/setup.py index 1a7eb43..3ae2840 100644 --- a/setup.py +++ b/setup.py @@ -13,8 +13,7 @@ requirements = [ 'mercantile', - 'PyYAML', - 'numpy' + 'PyYAML' ] scripts = [ diff --git a/wms_downloader/download.py b/wms_downloader/download.py index f15bb87..cce9a50 100755 --- a/wms_downloader/download.py +++ b/wms_downloader/download.py @@ -1,5 +1,6 @@ #!/usr/bin/env python import argparse +import glob import os import subprocess import yaml @@ -34,11 +35,19 @@ def main(): with open(args.config) as f: config = yaml.load(f.read()) + create_directory(config) + download_images(config) + create_vrt_file(config) + + +def create_directory(config): try: os.makedirs(config['directory']) except OSError: pass + +def download_images(config): west_range = list(arange(config['bbox']['west'], config['bbox']['east'], config['size'])) south_range = list(arange(config['bbox']['south'], config['bbox']['north'], config['size'])) @@ -71,8 +80,10 @@ def main(): args = ['gdal_translate', '-of', 'JPEG', config['tmpfile'], filename] subprocess.check_call(args) - file_pattern = '%s/*.gdal.tif' - args = ['gdalbuildvrt', '-a_srs', config['projection'], '-overwrite', config['vrtfile'], file_pattern] + +def create_vrt_file(config): + args = ['gdalbuildvrt', '-a_srs', config['projection'], '-overwrite', config['vrtfile']] + args += glob.glob('%s/*.gdal.tif' % config['directory']) subprocess.check_call(args)