| 1 | #! @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_shave = r_compile(r'\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 | line = rx_shave.sub(r'\1', line) |
| 36 | cdb.add(line, '') |
| 37 | cdb.finish() |