view textshaper/main.py @ 44:8addd6e12b29

corename
author Jeff Hammel <k0scist@gmail.com>
date Wed, 11 Mar 2015 10:15:20 -0700
parents 21b6a9569f21
children
line wrap: on
line source

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
package to shape text blocks
"""

import argparse
import os
import subprocess
import sys
import time
from .commands import Commands
from .indent import deindent, indent
from .onelineit import onelineit
from .tablify import make_csv
from .whitespace import underscore
from which import which

HR = '--'

def info(content):
    """gathers info about the content and returns a dict"""

    lines = content.splitlines()
    return {'lines': len(lines),
            'chars': len(content),
            'columns': max([len(line) for line in lines])}


def display(content, keys=('lines', 'chars', 'columns'), hr=HR):
    """displays the content"""
    print (content)
    if keys:
        _info = info(content)
        print (hr)
    print ('; '.join(['{}: {}'.format(key, _info[key])
                      for key in keys]))

def add_commands():
    # TODO: do this dynamically
    commands = Commands()
    commands.add(indent, 'i')
    commands.add(deindent, 'd')
    commands.add(onelineit, 'o')
    commands.add(underscore, 'u')
    commands.add(make_csv, 'c')
    return commands


def add_options(parser):
    """add options to the parser instance"""

    parser.add_argument('-f', '--file',  dest='input',
                        type=argparse.FileType('r'), default=sys.stdin,
                        help="file to read from [DEFAULT: stdin]")
    parser.add_argument('-n', '--no-strip', dest='strip',
                        action='store_false', default=True,
                        help="do not strip whitespace before processing")

    if which('xclip'): # TODO: support e.g. xsel or python native
        parser.add_argument('-c', '-o', '--clip', '--copy', dest='copy_to_clipboard',
                            action='store_true', default=False,
                            help="copy to clipboard")
        parser.add_argument('-i', dest='input_from_clipboard',
                            action='store_true', default=False,
                            help="copy from clipboard")


def main(args=sys.argv[1:]):
    """CLI"""

    # TODO: read ~/.textshaper

    # parse command line options
    parser = argparse.ArgumentParser(description=__doc__)
    add_options(parser)
    options = parser.parse_args(args)

    # read input
    if getattr(options, 'input_from_clipboard', False):
        content = subprocess.check_output(['xclip', '-o'])
    else:
        content = options.input.read()

    # get formatting commands
    commands = add_commands()

    # pre-process output
    if options.strip:
        content = content.strip()

    # main display loop
    # TODO: read input + commands
    try:
        while True:
            # print formatted content
            display(content)
            print (commands.display())
            choice = raw_input('? ')
            new_content = commands(choice, content)
            if new_content is None:
                print ("Choice '{}' not recognized".format(choice))
                continue
            content = new_content

            if options.copy_to_clipboard:
                # copy content to X clipboard
                process = subprocess.Popen(['xclip', '-i'], stdin=subprocess.PIPE)
                _, _ = process.communicate(content)
    except KeyboardInterrupt:
        sys.exit(0)

if __name__ == '__main__':
  main()