| 1 | #! @PYTHON@ |
| 2 | |
| 3 | from cdb import cdbmake |
| 4 | from sre import 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 | cdb = cdbmake(argv[1], argv[1] + '.new') |
| 27 | for f in files(argv[2:]): |
| 28 | for line in f: |
| 29 | if len(line) and line[-1] == '\n': line = line[:-1] |
| 30 | k, v = line.split(':', 1) |
| 31 | cdb.add(k, v) |
| 32 | cdb.finish() |