tag_features.py

#!/usr/bin/env python

"""
This script takes a file containing a collection of one or more feature 
definitions, that is pointed to by the -f flag and a tag name pointed to by 
the -t flag.  The tag is added to each feature (if that feature does not 
already have that tag.  Results are written back to the same feature file.
"""

import os.path
import json
import argparse
from collections import defaultdict
from utils.feature_write_utils import write_all_features


parser = argparse.ArgumentParser(description=__doc__, 
                                 formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument("-f", "--feature_file", dest="feature_file", 
                    help="Features collection file to be tagged", metavar="FILE",
                    required=True)
parser.add_argument("-t", "--tag", dest="tag", help="Tag to add to all features", 
                    metavar="TAG", required=True)
parser.add_argument("-r", "--remove", dest="remove", action='store_true',
                    help="Use this flag to signal removing a tag instead of adding")

args = parser.parse_args()


out_file_name = args.feature_file
all_features = defaultdict(list)

if os.path.exists(out_file_name):
  with open(out_file_name) as f:
    appended_file = json.load(f)
    for feature in appended_file['features']:
      try:
        feature_tags = feature['properties']['tags']
      except:
        feature_tags = ''
      feature_tag_list = feature_tags.split(';')
      if args.remove:
        if args.tag in feature_tag_list:
          feature_tag_list.remove(args.tag)
          if(len(feature_tag_list) == 0):
            feature['properties']['tags'] = ''
          else:
            feature_tags = feature_tag_list[0]
            for tag in feature_tag_list[1:len(feature_tag_list)]:
              feature_tags = '%s;%s'%(feature_tags,tag)
            feature['properties']['tags'] = feature_tags
      else:
        if args.tag not in feature_tag_list:
          if(feature_tags == ''):
            feature['properties']['tags'] = args.tag
          else:
            feature['properties']['tags'] = '%s;%s'%(feature_tags,args.tag)
  
      all_features['features'].append(feature)
    del appended_file

out_file = open(out_file_name, 'w')

out_file.write('{"type": "FeatureCollection",\n')
out_file.write(' "groupName": "enterNameHere",\n')
out_file.write(' "features":\n')
out_file.write('\t[\n')
write_all_features(all_features, out_file, '\t\t')
out_file.write('\n')
out_file.write('\t]\n')
out_file.write('}\n')

# vim: foldmethod=marker ai ts=4 sts=4 et sw=4 ft=python