Commit | Line | Data |
---|---|---|
bba986ae MW |
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() |