docker-osm/pbf_downloader.py

61 wiersze
1.6 KiB
Python

#!/usr/bin/python
import sys
from json import loads
from subprocess import call
URL = 'http://download.geofabrik.de/'
if len(sys.argv) < 2:
print('Not enough argument. "list" or a name (continent or country)')
exit()
# The JSON file comes from https://gist.github.com/Gustry/4e14bf096cdec09a3e57
json_data = open('countries.json').read()
data = loads(json_data)
if sys.argv[1] == 'list':
for continent, countries in list(data.items()):
print(continent)
for country in countries:
print(' ' + country)
exit()
else:
area = sys.argv[1]
url = None
for continent, countries in list(data.items()):
if area == continent:
url = URL + area
else:
if area in countries:
url = URL + continent + '/' + area
if url:
poly_file = url + '.poly'
pbf_file = url + '-latest.osm.pbf'
diff = url + '-updates/'
state = diff + 'state.txt'
print('Polygon file : ' + poly_file)
print('PBF file : ' + pbf_file)
print('Diff URL : ' + diff)
print('state : ' + state)
print('Downloading PBF')
commands = ['wget', '-c', '-O', 'settings/country.pbf', pbf_file]
call(commands)
print('Downloading polygon')
commands = ['wget', '-c', '-O', 'settings/country.poly', poly_file]
call(commands)
print('Downloading state')
commands = ['wget', '-c', '-O', 'settings/country.state.txt', state]
call(commands)
print('Setting custom URL diff')
with open('settings/custom_url_diff.txt', 'w') as f:
f.write(diff)
else:
print('This area is unknown in geofabrik or in our script. Check with the list argument.')