mdw@git.distorted.org.uk Git - misc/blob - cdb-assign

   1 #! /usr/bin/python
   2
   3 from cdb import cdbmake
   4 from sre import compile as r_compile, sub as r_sub
   5 from sys import argv, stdin, stderr, exit
   6
   7 ego = r_sub(r'^.*[/\\]', '', argv[0])
   8 def die(msg, prefix = True):
   9   if prefix: msg ='%s: %s' % (ego, msg)
  10   print >>stderr, msg
  11   exit(1)
  12
  13 def files(args):
  14   if len(args) == 0:
  15     yield stdin
  16   else:
  17     for a in args:
  18       if a == '-':
  19         yield stdin
  20       else:
  21         yield open(a, 'r')
  22
  23 if len(argv) < 2:
  24   die('usage: %s CDB [INPUT ...]' % ego, False)
  25
  26 rx_comment = r_compile(r'^\s*(\#|$)')
  27 rx_split = r_compile(r'^\s*([-\w]+)\s*=\s*(.*\S|)\s*$')
  28
  29 cdb = cdbmake(argv[1], argv[1] + '.new')
  30 for f in files(argv[2:]):
  31   for line in f:
  32     if len(line) and line[-1] == '\n': line = line[:-1]
  33     if rx_comment.match(line):
  34       continue
  35     m = rx_split.match(line)
  36     if not m:
  37       die("bad assignment: `%s'" % line)
  38     k, v = m.groups([1, 2])
  39     cdb.add(k, v)
  40 cdb.finish()