Mercurial > hg > TextShaper
annotate textshaper/indent.py @ 23:7ffe000f6f42
removing stub
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Sun, 23 Feb 2014 10:14:06 -0800 |
parents | 0d0db0d79bfd |
children | 0930c6884f8a |
rev | line source |
---|---|
9
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
1 #!/usr/bin/env python |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
2 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
3 """ |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
4 indentation of text blocks |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
5 """ |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
6 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
7 import optparse |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
8 import os |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
9 import sys |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
10 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
11 def indent(text, indentation=4, space=' ', strict=False): |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
12 """ |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
13 indent a block of text |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
14 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
15 text -- lines of text to indent |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
16 indentation -- number of spaces to indent |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
17 space -- what to indent with |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
18 strict -- whether to enforce required whitespace for negative indentation |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
19 """ |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
20 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
21 if not indentation: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
22 # nothing to do |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
23 return text |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
24 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
25 if indentation > 0: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
26 retval = [space * indentation + line for line in text] |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
27 else: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
28 # negative indentation |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
29 indentation = -indentation |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
30 retval = [] |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
31 for line in text: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
32 prefix = line[:indentation] |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
33 for index, char in enumerate(prefix): |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
34 if not char == space: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
35 if strict: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
36 raise AssertionError("Found non-'%s' charcter at column %d for indentation -%d" % (space, index, indentation)) |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
37 break |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
38 else: |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
39 index = indentation |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
40 retval.append(line[index:]) |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
41 |
10 | 42 return retval |
43 | |
44 | |
9
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
45 def main(args=sys.argv[1:]): |
11 | 46 # TODO : refactor to be more general and stuff |
9
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
47 |
10 | 48 # parse command line |
49 usage = '%prog [options] [file] [file2] [...]' | |
50 description = """indent files or stdin if no files given""" | |
9
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
51 parser = optparse.OptionParser(usage=usage, description=__doc__) |
10 | 52 parser.add_option('-o', '--output', dest='output', |
53 help="output file or stdout if not given") | |
54 options, files = parser.parse_args(args) | |
9
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
55 |
10 | 56 # input from files or stdin |
57 if files: | |
58 missing = [not os.path.exists(filename) for filename in files] | |
59 if missing: | |
60 parser.error("File(s) not found: %s" % ', '.join(missing)) | |
61 def _files(): | |
62 for filename in files: | |
63 with open(f, 'r') as f: | |
64 yield f | |
65 else: | |
66 def _files(): | |
67 yield sys.stdin | |
68 | |
69 # process input | |
70 for f in _files(): | |
71 print f | |
9
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
72 |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
73 if __name__ == '__main__': |
71fb16088d54
add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents:
diff
changeset
|
74 main() |