annotate textshaper/main.py @ 47:6c7ca72777af

add a test all convenience
author Jeff Hammel <k0scist@gmail.com>
date Sat, 16 May 2015 19:02:03 -0700
parents 21b6a9569f21
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
1 #!/usr/bin/env python
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
2 # -*- coding: utf-8 -*-
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
3
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
4 """
4
26bb5e99f76c textshaper/main.py
Jeff Hammel <jhammel@mozilla.com>
parents: 3
diff changeset
5 package to shape text blocks
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
6 """
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
7
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
8 import argparse
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
9 import os
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
10 import subprocess
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
11 import sys
3
Jeff Hammel <jhammel@mozilla.com>
parents: 0
diff changeset
12 import time
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
13 from .commands import Commands
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
14 from .indent import deindent, indent
43
21b6a9569f21 onelineit
Jeff Hammel <k0scist@gmail.com>
parents: 37
diff changeset
15 from .onelineit import onelineit
37
8cf9a26b9aaa STUB: textshaper/main.py textshaper/tablify.py
Jeff Hammel <k0scist@gmail.com>
parents: 36
diff changeset
16 from .tablify import make_csv
36
55e0579e2143 STUB: textshaper/main.py textshaper/whitespace.py
Jeff Hammel <k0scist@gmail.com>
parents: 34
diff changeset
17 from .whitespace import underscore
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
18 from which import which
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
19
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
20 HR = '--'
19
70dde00a4df0 selectable, albeit single, input
Jeff Hammel <k0scist@gmail.com>
parents: 18
diff changeset
21
12
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
22 def info(content):
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
23 """gathers info about the content and returns a dict"""
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
24
12
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
25 lines = content.splitlines()
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
26 return {'lines': len(lines),
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
27 'chars': len(content),
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
28 'columns': max([len(line) for line in lines])}
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
29
12
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
30
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
31 def display(content, keys=('lines', 'chars', 'columns'), hr=HR):
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
32 """displays the content"""
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
33 print (content)
12
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
34 if keys:
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
35 _info = info(content)
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
36 print (hr)
5b70bbff04b1 STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 9
diff changeset
37 print ('; '.join(['{}: {}'.format(key, _info[key])
13
c3df7dccb02b STUB: textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 12
diff changeset
38 for key in keys]))
4
26bb5e99f76c textshaper/main.py
Jeff Hammel <jhammel@mozilla.com>
parents: 3
diff changeset
39
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
40 def add_commands():
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
41 # TODO: do this dynamically
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
42 commands = Commands()
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
43 commands.add(indent, 'i')
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
44 commands.add(deindent, 'd')
43
21b6a9569f21 onelineit
Jeff Hammel <k0scist@gmail.com>
parents: 37
diff changeset
45 commands.add(onelineit, 'o')
36
55e0579e2143 STUB: textshaper/main.py textshaper/whitespace.py
Jeff Hammel <k0scist@gmail.com>
parents: 34
diff changeset
46 commands.add(underscore, 'u')
37
8cf9a26b9aaa STUB: textshaper/main.py textshaper/tablify.py
Jeff Hammel <k0scist@gmail.com>
parents: 36
diff changeset
47 commands.add(make_csv, 'c')
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
48 return commands
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
49
36
55e0579e2143 STUB: textshaper/main.py textshaper/whitespace.py
Jeff Hammel <k0scist@gmail.com>
parents: 34
diff changeset
50
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
51 def add_options(parser):
19
70dde00a4df0 selectable, albeit single, input
Jeff Hammel <k0scist@gmail.com>
parents: 18
diff changeset
52 """add options to the parser instance"""
70dde00a4df0 selectable, albeit single, input
Jeff Hammel <k0scist@gmail.com>
parents: 18
diff changeset
53
22
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
54 parser.add_argument('-f', '--file', dest='input',
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
55 type=argparse.FileType('r'), default=sys.stdin,
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
56 help="file to read from [DEFAULT: stdin]")
32
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
57 parser.add_argument('-n', '--no-strip', dest='strip',
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
58 action='store_false', default=True,
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
59 help="do not strip whitespace before processing")
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
60
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
61 if which('xclip'): # TODO: support e.g. xsel or python native
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
62 parser.add_argument('-c', '-o', '--clip', '--copy', dest='copy_to_clipboard',
22
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
63 action='store_true', default=False,
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
64 help="copy to clipboard")
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
65 parser.add_argument('-i', dest='input_from_clipboard',
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
66 action='store_true', default=False,
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
67 help="copy from clipboard")
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
68
9
71fb16088d54 add file for indentation; wth did my work go??? :(
Jeff Hammel <k0scist@gmail.com>
parents: 5
diff changeset
69
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
70 def main(args=sys.argv[1:]):
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
71 """CLI"""
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
72
32
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
73 # TODO: read ~/.textshaper
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
74
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
75 # parse command line options
22
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
76 parser = argparse.ArgumentParser(description=__doc__)
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
77 add_options(parser)
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
78 options = parser.parse_args(args)
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
79
22
586631375670 STUB: README.txt textshaper/main.py
Jeff Hammel <k0scist@gmail.com>
parents: 19
diff changeset
80 # read input
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
81 if getattr(options, 'input_from_clipboard', False):
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
82 content = subprocess.check_output(['xclip', '-o'])
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
83 else:
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
84 content = options.input.read()
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
85
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
86 # get formatting commands
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
87 commands = add_commands()
3
Jeff Hammel <jhammel@mozilla.com>
parents: 0
diff changeset
88
32
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
89 # pre-process output
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
90 if options.strip:
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
91 content = content.strip()
929a5e97af3e prestrip text unless specified
Jeff Hammel <k0scist@gmail.com>
parents: 22
diff changeset
92
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
93 # main display loop
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
94 # TODO: read input + commands
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
95 try:
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
96 while True:
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
97 # print formatted content
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
98 display(content)
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
99 print (commands.display())
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
100 choice = raw_input('? ')
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
101 new_content = commands(choice, content)
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
102 if new_content is None:
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
103 print ("Choice '{}' not recognized".format(choice))
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
104 continue
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
105 content = new_content
3
Jeff Hammel <jhammel@mozilla.com>
parents: 0
diff changeset
106
34
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
107 if options.copy_to_clipboard:
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
108 # copy content to X clipboard
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
109 process = subprocess.Popen(['xclip', '-i'], stdin=subprocess.PIPE)
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
110 _, _ = process.communicate(content)
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
111 except KeyboardInterrupt:
88a69d587326 round1 of commands
Jeff Hammel <k0scist@gmail.com>
parents: 32
diff changeset
112 sys.exit(0)
18
df52326aa08d start filling out options
Jeff Hammel <k0scist@gmail.com>
parents: 13
diff changeset
113
0
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
114 if __name__ == '__main__':
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
115 main()
53cfd6e1ad0f initial commit,
Jeff Hammel <jhammel@mozilla.com>
parents:
diff changeset
116