-
Notifications
You must be signed in to change notification settings - Fork 105
/
pbf_downloader.py
60 lines (49 loc) · 1.64 KB
/
pbf_downloader.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/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.')