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
61
62
63
64
65
66
|
"""
This example shows how to filter and modify tags and write the results back.
It changes all tag values 'yes/no' to '1/0'.
"""
import osmium
import sys
class BoolNormalizer(osmium.SimpleHandler):
def __init__(self, writer):
super(BoolNormalizer, self).__init__()
self.writer = writer
def normalize(self, o):
# if there are no tags we are done
if not o.tags:
return o
# new tags should be kept in a list so that the order is preserved
newtags = []
# pyosmium is much faster writing an original osmium object than
# a osmium.mutable.*. Therefore, keep track if the tags list was
# actually changed.
modified = False
for t in o.tags:
if t.v == 'yes':
# custom tags should be added as a key/value tuple
newtags.append((t.k, '1'))
modified = True
elif t.v == 'no':
newtags.append((t.k, '0'))
modified = True
else:
# if the tag is not modified, simply readd it to the list
newtags.append(t)
if modified:
# We have changed tags. Create a new object as a copy of the
# original one with the tag list replaced.
return o.replace(tags=newtags)
else:
# Nothing changed, so simply return the original object
# and discard the tag list we just created.
return o
def node(self, n):
self.writer.add_node(self.normalize(n))
def way(self, w):
self.writer.add_way(self.normalize(w))
def relation(self, r):
self.writer.add_relation(self.normalize(r))
if __name__ == '__main__':
if len(sys.argv) != 3:
print("Usage: python normalize_boolean.py <infile> <outfile>")
sys.exit(-1)
writer = osmium.SimpleWriter(sys.argv[2])
BoolNormalizer(writer).apply_file(sys.argv[1])
writer.close()
|
