/usr/share/doc/python3-pyosmium/examples/normalize_boolean.py is in python3-pyosmium 2.6.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 rusults back.
It changes all tag values 'yes/no' to '1/0'.
"""
import osmium as o
import sys
class BoolNormalizer(o.SimpleHandler):
def __init__(self, writer):
o.SimpleHandler.__init__(self)
self.writer = writer
def normalize(self, o):
# if there are no tags we are done
if len(o.tags) == 0:
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, o):
self.writer.add_node(self.normalize(o))
def way(self, o):
self.writer.add_way(self.normalize(o))
def relation(self, o):
self.writer.add_relation(self.normalize(o))
if __name__ == '__main__':
if len(sys.argv) != 3:
print("Usage: python normalize_boolean.py <infile> <outfile>")
sys.exit(-1)
writer = o.SimpleWriter(sys.argv[2])
BoolNormalizer(writer).apply_file(sys.argv[1])
writer.close()
|