mirror of
https://github.com/kennethreitz/clint.git
synced 2026-06-05 23:00:18 +00:00
Compare commits
59 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| f651e71093 | |||
| e500773da3 | |||
| 4b9e35c631 | |||
| e85fffdbb6 | |||
| b58f8dccdc | |||
| 3d63ed01f9 | |||
| 81165754a4 | |||
| 6f106e2f12 | |||
| 8adda14a27 | |||
| cc072a18f3 | |||
| c226b6b736 | |||
| 3406ab5329 | |||
| cf0c0bda70 | |||
| a4c2be8865 | |||
| fa2bbf9a0d | |||
| 9736b4cd71 | |||
| 2e68207d0a | |||
| 164d68ce37 | |||
| aa5008b2e0 | |||
| 8b7161b15b | |||
| 763ca94e7b | |||
| 2d200e3709 | |||
| d0f2375a7b | |||
| 2369bc387d | |||
| f12e47fa7c | |||
| 5d733d123d | |||
| 9e1ea1ef91 | |||
| 6d151a0794 | |||
| 13afb11dea | |||
| faeacbbfad | |||
| 401344ff1e | |||
| 598deda4d9 | |||
| bf7b5d0930 | |||
| 6404b0e49c | |||
| b92f3204c9 | |||
| 61225d90e6 | |||
| 3688a2a67b | |||
| dda561387c | |||
| 06af338874 | |||
| 9b91cf16b5 | |||
| ec8fe4a0b7 | |||
| dc6a9c63ea | |||
| 82d31e555f | |||
| 2f6e9cbfaa | |||
| b576ccbaae | |||
| 8e021f95e9 | |||
| 83bf2288dd | |||
| f2fc21755b | |||
| 52488ace36 | |||
| 8b6eb543b9 | |||
| 2cd9bd5263 | |||
| 76bee9c4ba | |||
| 14c4cdf904 | |||
| f6a20fc413 | |||
| fc3eeca54f | |||
| ebdb04848a | |||
| f0206b21fc | |||
| 5366c7e5b2 | |||
| 35ec821e37 |
+3
-1
@@ -1,2 +1,4 @@
|
||||
.idea
|
||||
MANIFEST
|
||||
MANIFEST
|
||||
*.pyc
|
||||
.tox
|
||||
|
||||
@@ -12,4 +12,5 @@ Patches and Suggestions
|
||||
|
||||
- Jeff Forcier
|
||||
- Morgan Goose
|
||||
- Travis Swicegood
|
||||
- Travis Swicegood
|
||||
- Will Thames
|
||||
|
||||
+41
-4
@@ -1,10 +1,45 @@
|
||||
History
|
||||
-------
|
||||
|
||||
0.1.0 (2011-03-20)
|
||||
0.2.3
|
||||
+++++
|
||||
|
||||
* Only init colors if they are used (iPython compatability)
|
||||
* New progress module
|
||||
* Various bugfixes
|
||||
|
||||
|
||||
0.2.2
|
||||
+++++
|
||||
|
||||
* Auto Color Disabling
|
||||
* Progress Namespace Change
|
||||
* New Progress Bars
|
||||
* textui.puts newline fix
|
||||
|
||||
|
||||
0.2.1 (2011-03-24)
|
||||
++++++++++++++++++
|
||||
|
||||
* Initial Release!
|
||||
* Python 2.5 Support
|
||||
* List of available colors
|
||||
|
||||
|
||||
0.2.0 (2011-03-23)
|
||||
++++++++++++++++++
|
||||
|
||||
* Column Printing!!!
|
||||
* (Auto/Manual) Disabling of Colors
|
||||
* Smarter Colors
|
||||
* max_width, min_width
|
||||
* Strip cli colors
|
||||
* bug fixes
|
||||
|
||||
|
||||
0.1.2 (2011-03-21)
|
||||
++++++++++++++++++
|
||||
|
||||
* Bugfixes
|
||||
|
||||
|
||||
0.1.1 (2011-03-20)
|
||||
@@ -16,7 +51,9 @@ History
|
||||
* Lots of Examples
|
||||
|
||||
|
||||
0.1.2 (2011-03-21)
|
||||
|
||||
0.1.0 (2011-03-20)
|
||||
++++++++++++++++++
|
||||
|
||||
* Bugfixes
|
||||
* Initial Release!
|
||||
|
||||
|
||||
+1
-1
@@ -1 +1 @@
|
||||
include HISTORY.rst README.rst LICENSE AUTHORS NOTICE
|
||||
include HISTORY.rst README.rst LICENSE AUTHORS NOTICE
|
||||
@@ -1,4 +1,4 @@
|
||||
Clint includes some vendorized python libraries: appdirs, colorama, ordereddict.
|
||||
Clint includes some vendorized python libraries: appdirs, applib, colorama, ordereddict.
|
||||
|
||||
AppDirs License
|
||||
===============
|
||||
@@ -25,6 +25,33 @@ TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
||||
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
||||
|
||||
AppLib License (console width detection functions are from this library)
|
||||
==============
|
||||
|
||||
# This is the MIT license
|
||||
|
||||
# Copyright (c) 2010 ActiveState Software Inc. All rights reserved.
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a
|
||||
copy of this software and associated documentation files (the
|
||||
"Software"), to deal in the Software without restriction, including
|
||||
without limitation the rights to use, copy, modify, merge, publish,
|
||||
distribute, sublicense, and/or sell copies of the Software, and to
|
||||
permit persons to whom the Software is furnished to do so, subject to
|
||||
the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included
|
||||
in all copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||||
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||||
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
||||
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
||||
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
||||
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
||||
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
||||
|
||||
|
||||
Colorama License
|
||||
================
|
||||
|
||||
+30
-3
@@ -12,16 +12,43 @@ commandline applications.
|
||||
**T** ools
|
||||
.
|
||||
|
||||
Features:
|
||||
---------
|
||||
|
||||
Clint is awesome. Crazy awesome. It supports colors, but detects if the session is a TTY, so doesn't render the colors if you're piping stuff around. Automagically.
|
||||
|
||||
Awesome nest-able indentation context manager. Example: (``with indent(4): puts('indented text')``). It supports custom email-style quotes. Of course, it supports color too, if and when needed.
|
||||
|
||||
It has an awesome Column printer with optional auto-expanding columns. It detects how wide your current console is and adjusts accordingly. It wraps your words properly to fit the column size. With or without colors mixed in. All with a single function call.
|
||||
|
||||
The world's easiest to use implicit argument system w/ chaining methods for filtering. Seriously.
|
||||
|
||||
|
||||
Run the various executables in examples_ to get a good feel for what Clint offers.
|
||||
|
||||
.. _examples: https://github.com/kennethreitz/clint/tree/develop/examples
|
||||
|
||||
You'll never want to not use it.
|
||||
|
||||
|
||||
|
||||
Current Features:
|
||||
-----------------
|
||||
- Little Documentation (bear with me for now)
|
||||
- CLI Colors and Indents
|
||||
- Extremely Simple + Powerful Column Printer
|
||||
- Iterator-based Progress Bar
|
||||
- Implicit Argument Handling
|
||||
- Simple Support for Unix Pipes
|
||||
- Simple Support for Incoming Unix Pipes
|
||||
- Application Directory management
|
||||
|
||||
|
||||
Future Features:
|
||||
----------------
|
||||
- Documentation!
|
||||
- Simple choice system ``Are you sure? [Yn]``
|
||||
- Default query system ``Installation Path [/usr/local/bin/]``
|
||||
- Suggestions welcome.
|
||||
|
||||
|
||||
Example
|
||||
-------
|
||||
|
||||
|
||||
+2
-2
@@ -19,8 +19,8 @@ from .pipes import piped_in
|
||||
|
||||
|
||||
__title__ = 'clint'
|
||||
__version__ = '0.1.2'
|
||||
__build__ = 0x000102
|
||||
__version__ = '0.2.3'
|
||||
__build__ = 0x000203
|
||||
__author__ = 'Kenneth Reitz'
|
||||
__license__ = 'ISC'
|
||||
__copyright__ = 'Copyright 2011 Kenneth Reitz'
|
||||
|
||||
+33
-14
@@ -25,19 +25,19 @@ __all__ = ('Args', )
|
||||
|
||||
|
||||
def _expand_path(path):
|
||||
"""Expands directories and globs in given path."""
|
||||
"""Expands directories and globs in given path."""
|
||||
|
||||
paths = []
|
||||
paths = []
|
||||
|
||||
if os.path.isdir(path):
|
||||
if os.path.isdir(path):
|
||||
|
||||
for (dir, dirs, files) in os.walk(path):
|
||||
for file in files:
|
||||
paths.append(os.path.join(dir, file))
|
||||
else:
|
||||
paths.extend(glob(path))
|
||||
for (dir, dirs, files) in os.walk(path):
|
||||
for file in files:
|
||||
paths.append(os.path.join(dir, file))
|
||||
else:
|
||||
paths.extend(glob(path))
|
||||
|
||||
return paths
|
||||
return paths
|
||||
|
||||
|
||||
class Args(object):
|
||||
@@ -69,7 +69,7 @@ class Args(object):
|
||||
|
||||
|
||||
def __contains__(self, x):
|
||||
return bool(self.first(x))
|
||||
return self.first(x) is not None
|
||||
|
||||
|
||||
def get(self, x):
|
||||
@@ -90,7 +90,7 @@ class Args(object):
|
||||
|
||||
def _remove(x):
|
||||
found = self.first(x)
|
||||
if found:
|
||||
if found is not None:
|
||||
self._args.pop(found)
|
||||
|
||||
if is_collection(x):
|
||||
@@ -133,7 +133,7 @@ class Args(object):
|
||||
if is_collection(x):
|
||||
for item in x:
|
||||
found = _find(item)
|
||||
if found:
|
||||
if found is not None:
|
||||
return found
|
||||
return None
|
||||
else:
|
||||
@@ -181,6 +181,25 @@ class Args(object):
|
||||
else:
|
||||
return _find(x)
|
||||
|
||||
|
||||
def start_with(self, x):
|
||||
"""Returns all arguments beginning with given string (or list thereof)"""
|
||||
|
||||
_args = []
|
||||
|
||||
for arg in self.all:
|
||||
if is_collection(x):
|
||||
for _x in x:
|
||||
if arg.startswith(x):
|
||||
_args.append(arg)
|
||||
break
|
||||
else:
|
||||
if arg.startswith(x):
|
||||
_args.append(arg)
|
||||
|
||||
return Args(_args, no_argv=True)
|
||||
|
||||
|
||||
def contains_at(self, x, index):
|
||||
"""Tests if given [list of] string is at given index."""
|
||||
|
||||
@@ -305,7 +324,7 @@ class Args(object):
|
||||
def flags(self):
|
||||
"""Returns Arg object including only flagged arguments."""
|
||||
|
||||
return self.all_with('-')
|
||||
return self.start_with('-')
|
||||
|
||||
|
||||
@property
|
||||
@@ -343,7 +362,7 @@ class Args(object):
|
||||
if not os.path.exists(arg):
|
||||
_args.append(arg)
|
||||
|
||||
return _args
|
||||
return Args(_args, no_argv=True)
|
||||
|
||||
@property
|
||||
def copy(self):
|
||||
|
||||
@@ -9,6 +9,7 @@ This module contains the helper functions for dealing with unix pipes.
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import with_statement
|
||||
|
||||
import sys
|
||||
|
||||
|
||||
+9
-8
@@ -10,6 +10,7 @@ This module contains all the application resource features of clint.
|
||||
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import with_statement
|
||||
|
||||
import errno
|
||||
from os import remove, removedirs
|
||||
@@ -38,7 +39,7 @@ class AppDir(object):
|
||||
|
||||
def __repr__(self):
|
||||
return '<app-dir: %s>' % (self.path)
|
||||
|
||||
|
||||
|
||||
def __getattribute__(self, name):
|
||||
|
||||
@@ -52,11 +53,11 @@ class AppDir(object):
|
||||
"""Raises if operations are carried out on an unconfigured AppDir."""
|
||||
if not self.path:
|
||||
raise NotConfigured()
|
||||
|
||||
|
||||
|
||||
def _create(self):
|
||||
"""Creates current AppDir at AppDir.path."""
|
||||
|
||||
|
||||
self._raise_if_none()
|
||||
if not self._exists:
|
||||
mkdir_p(self.path)
|
||||
@@ -65,7 +66,7 @@ class AppDir(object):
|
||||
|
||||
def open(self, filename, mode='r'):
|
||||
"""Returns file object from given filename."""
|
||||
|
||||
|
||||
self._raise_if_none()
|
||||
fn = path_join(self.path, filename)
|
||||
|
||||
@@ -120,11 +121,11 @@ class AppDir(object):
|
||||
else:
|
||||
raise why
|
||||
|
||||
|
||||
|
||||
def read(self, filename, binary=False):
|
||||
"""Returns contents of given file with AppDir.
|
||||
If file doesn't exist, returns None."""
|
||||
|
||||
|
||||
self._raise_if_none()
|
||||
fn = path_join(self.path, filename)
|
||||
|
||||
@@ -160,11 +161,11 @@ log = AppDir()
|
||||
def init(vendor, name):
|
||||
|
||||
global user, site, cache, log
|
||||
|
||||
|
||||
ad = AppDirs(name, vendor)
|
||||
|
||||
user.path = ad.user_data_dir
|
||||
|
||||
|
||||
site.path = ad.site_data_dir
|
||||
cache.path = ad.user_cache_dir
|
||||
log.path = ad.user_log_dir
|
||||
|
||||
@@ -8,8 +8,8 @@ This module provides the text output helper system.
|
||||
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
|
||||
|
||||
from . import colored
|
||||
from .core import *
|
||||
from . import progress
|
||||
|
||||
from core import *
|
||||
|
||||
+49
-18
@@ -11,18 +11,23 @@ This module provides a simple and elegant wrapper for colorama.
|
||||
|
||||
from __future__ import absolute_import
|
||||
|
||||
import re
|
||||
import sys
|
||||
|
||||
from ..packages import colorama
|
||||
|
||||
|
||||
|
||||
colorama.init(autoreset=True)
|
||||
|
||||
|
||||
__all__ = (
|
||||
'red', 'green', 'yellow', 'blue',
|
||||
'black', 'magenta', 'cyan', 'white'
|
||||
'black', 'magenta', 'cyan', 'white',
|
||||
'clean', 'disable'
|
||||
)
|
||||
|
||||
COLORS = __all__[:-2]
|
||||
DISABLE_COLOR = False
|
||||
|
||||
if not sys.stdout.isatty():
|
||||
DISABLE_COLOR = True
|
||||
|
||||
|
||||
class ColoredString(object):
|
||||
"""Enhanced string for __len__ operations on Colored output."""
|
||||
@@ -30,33 +35,53 @@ class ColoredString(object):
|
||||
super(ColoredString, self).__init__()
|
||||
self.s = s
|
||||
self.color = color
|
||||
self.color_str = '%s%s%s' % (
|
||||
getattr(colorama.Fore, self.color), self.s, colorama.Fore.RESET)
|
||||
|
||||
|
||||
@property
|
||||
def color_str(self):
|
||||
if sys.stdout.isatty() and not DISABLE_COLOR:
|
||||
colorama.init(autoreset=True)
|
||||
return '%s%s%s' % (
|
||||
getattr(colorama.Fore, self.color), self.s, colorama.Fore.RESET)
|
||||
else:
|
||||
return self.s
|
||||
|
||||
|
||||
def __len__(self):
|
||||
return len(self.s)
|
||||
|
||||
|
||||
def __repr__(self):
|
||||
return "<%s-string: '%s'>" % (self.color, self.s)
|
||||
|
||||
|
||||
def __str__(self):
|
||||
return self.__unicode__().encode('utf8')
|
||||
|
||||
|
||||
def __unicode__(self):
|
||||
return self.color_str
|
||||
|
||||
|
||||
def __add__(self, other):
|
||||
return str(self.color_str) + str(other)
|
||||
|
||||
|
||||
def __radd__(self, other):
|
||||
return str(other) + str(self.color_str)
|
||||
|
||||
|
||||
def __mul__(self, other):
|
||||
return (self.color_str * other)
|
||||
|
||||
def split(self, x=' '):
|
||||
return self.color_str.split(x)
|
||||
|
||||
def split(self, x=' '):
|
||||
return map(self._new, self.s.split(x))
|
||||
|
||||
def _new(self, s):
|
||||
return ColoredString(self.color, s)
|
||||
|
||||
|
||||
def clean(s):
|
||||
strip = re.compile("([^-_a-zA-Z0-9!@#%&=,/'\";:~`\$\^\*\(\)\+\[\]\.\{\}\|\?\<\>\\]+|[^\s]+)")
|
||||
txt = strip.sub('', str(s))
|
||||
|
||||
strip = re.compile(r'\[\d+m')
|
||||
txt = strip.sub('', txt)
|
||||
|
||||
return txt
|
||||
|
||||
|
||||
def black(string):
|
||||
@@ -82,3 +107,9 @@ def cyan(string):
|
||||
|
||||
def white(string):
|
||||
return ColoredString('WHITE', string)
|
||||
|
||||
def disable():
|
||||
"""Disables colors."""
|
||||
global DISABLE_COLOR
|
||||
|
||||
DISABLE_COLOR = True
|
||||
|
||||
@@ -0,0 +1,141 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
clint.textui.columns
|
||||
~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
Core TextUI functionality for column formatting.
|
||||
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
|
||||
from .formatters import max_width, min_width
|
||||
from ..utils import tsplit
|
||||
|
||||
import sys
|
||||
|
||||
|
||||
NEWLINES = ('\n', '\r', '\r\n')
|
||||
|
||||
|
||||
|
||||
def _find_unix_console_width():
|
||||
import termios, fcntl, struct, sys
|
||||
|
||||
# fcntl.ioctl will fail if stdout is not a tty
|
||||
if not sys.stdout.isatty():
|
||||
return None
|
||||
|
||||
s = struct.pack("HHHH", 0, 0, 0, 0)
|
||||
fd_stdout = sys.stdout.fileno()
|
||||
size = fcntl.ioctl(fd_stdout, termios.TIOCGWINSZ, s)
|
||||
height, width = struct.unpack("HHHH", size)[:2]
|
||||
return width
|
||||
|
||||
|
||||
def _find_windows_console_width():
|
||||
# http://code.activestate.com/recipes/440694/
|
||||
from ctypes import windll, create_string_buffer
|
||||
STDIN, STDOUT, STDERR = -10, -11, -12
|
||||
|
||||
h = windll.kernel32.GetStdHandle(STDERR)
|
||||
csbi = create_string_buffer(22)
|
||||
res = windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
|
||||
|
||||
if res:
|
||||
import struct
|
||||
(bufx, bufy, curx, cury, wattr,
|
||||
left, top, right, bottom,
|
||||
maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
|
||||
sizex = right - left + 1
|
||||
sizey = bottom - top + 1
|
||||
return sizex
|
||||
|
||||
|
||||
def console_width(kwargs):
|
||||
""""Determine console_width."""
|
||||
|
||||
if sys.platform.startswith('win'):
|
||||
console_width = _find_windows_console_width()
|
||||
else:
|
||||
console_width = _find_unix_console_width()
|
||||
|
||||
_width = kwargs.get('width', None)
|
||||
if _width:
|
||||
console_width = _width
|
||||
else:
|
||||
if not console_width:
|
||||
console_width = 80
|
||||
|
||||
return console_width
|
||||
|
||||
|
||||
|
||||
def columns(*cols, **kwargs):
|
||||
|
||||
columns = list(cols)
|
||||
|
||||
cwidth = console_width(kwargs)
|
||||
|
||||
_big_col = None
|
||||
_total_cols = 0
|
||||
|
||||
|
||||
for i, (string, width) in enumerate(cols):
|
||||
|
||||
if width is not None:
|
||||
_total_cols += (width + 1)
|
||||
cols[i][0] = max_width(string, width).split('\n')
|
||||
else:
|
||||
_big_col = i
|
||||
|
||||
if _big_col:
|
||||
cols[_big_col][1] = (cwidth - _total_cols) - len(cols)
|
||||
cols[_big_col][0] = max_width(cols[_big_col][0], cols[_big_col][1]).split('\n')
|
||||
|
||||
height = len(max([c[0] for c in cols], key=len))
|
||||
|
||||
for i, (strings, width) in enumerate(cols):
|
||||
|
||||
for _ in range(height - len(strings)):
|
||||
cols[i][0].append('')
|
||||
|
||||
for j, string in enumerate(strings):
|
||||
cols[i][0][j] = min_width(string, width)
|
||||
|
||||
stack = [c[0] for c in cols]
|
||||
_out = []
|
||||
|
||||
for i in range(height):
|
||||
_row = ''
|
||||
|
||||
for col in stack:
|
||||
_row += col[i]
|
||||
_row += ' '
|
||||
|
||||
_out.append(_row)
|
||||
# try:
|
||||
# pass
|
||||
# except:
|
||||
# pass
|
||||
|
||||
|
||||
|
||||
|
||||
return '\n'.join(_out)
|
||||
|
||||
|
||||
# string = max_width(string, width)
|
||||
# string = min_width(string, width)
|
||||
# pass
|
||||
# columns.append()
|
||||
|
||||
|
||||
|
||||
###########################
|
||||
|
||||
a = 'this is text that goes into a small column\n cool?'
|
||||
b = 'this is other text\nothertext\nothertext'
|
||||
|
||||
#columns((a, 10), (b, 20), (b, None))
|
||||
+16
-11
@@ -13,16 +13,20 @@ from __future__ import absolute_import
|
||||
|
||||
import sys
|
||||
|
||||
from .progress import progressbar
|
||||
from .formatters import max_width, min_width
|
||||
from .cols import columns
|
||||
from ..utils import tsplit
|
||||
|
||||
|
||||
__all__ = ('puts', 'puts_err', 'indent', 'progressbar')
|
||||
__all__ = ('puts', 'puts_err', 'indent', 'columns', 'max_width', 'min_width')
|
||||
|
||||
|
||||
STDOUT = sys.stdout.write
|
||||
STDERR = sys.stderr.write
|
||||
|
||||
NEWLINES = ('\n', '\r', '\r\n')
|
||||
|
||||
|
||||
|
||||
class Writer(object):
|
||||
"""WriterUtilized by context managers."""
|
||||
@@ -58,13 +62,14 @@ class Writer(object):
|
||||
|
||||
|
||||
def __call__(self, s, newline=True, stream=STDOUT):
|
||||
|
||||
|
||||
if newline:
|
||||
s = tsplit(s, ('\n', '\r', '\r\n'))
|
||||
s = tsplit(s, NEWLINES)
|
||||
s = map(str, s)
|
||||
indent = ''.join(self.shared['indent_strings'])
|
||||
|
||||
s = ('\n' + indent).join(s)
|
||||
|
||||
|
||||
s = (str('\n' + indent)).join(s)
|
||||
|
||||
_str = ''.join((
|
||||
''.join(self.shared['indent_strings']),
|
||||
str(s),
|
||||
@@ -73,14 +78,14 @@ class Writer(object):
|
||||
stream(_str)
|
||||
|
||||
|
||||
def puts(s, newline=True):
|
||||
def puts(s='', newline=True, stream=STDOUT):
|
||||
"""Prints given string to stdout via Writer interface."""
|
||||
Writer()(s, stream=STDOUT)
|
||||
Writer()(s, newline, stream=stream)
|
||||
|
||||
|
||||
def puts_err(s, newline=True):
|
||||
def puts_err(s='', newline=True, stream=STDERR):
|
||||
"""Prints given string to stderr via Writer interface."""
|
||||
Writer()(s, stream=STDERR)
|
||||
Writer()(s, newline, stream=stream)
|
||||
|
||||
|
||||
def indent(indent=4, quote=''):
|
||||
|
||||
@@ -0,0 +1,91 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
clint.textui.formatters
|
||||
~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
Core TextUI functionality for text formatting.
|
||||
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
|
||||
from .colored import ColoredString, clean
|
||||
from ..utils import tsplit, schunk
|
||||
|
||||
|
||||
NEWLINES = ('\n', '\r', '\r\n')
|
||||
|
||||
|
||||
def min_width(string, cols, padding=' '):
|
||||
"""Returns given string with right padding."""
|
||||
|
||||
is_color = isinstance(string, ColoredString)
|
||||
|
||||
stack = tsplit(str(string), NEWLINES)
|
||||
|
||||
for i, substring in enumerate(stack):
|
||||
_sub = clean(substring).ljust((cols + 0), padding)
|
||||
if is_color:
|
||||
_sub = (_sub.replace(clean(substring), substring))
|
||||
stack[i] = _sub
|
||||
|
||||
return '\n'.join(stack)
|
||||
|
||||
|
||||
def max_width(string, cols, separator='\n'):
|
||||
"""Returns a freshly formatted """
|
||||
|
||||
is_color = isinstance(string, ColoredString)
|
||||
|
||||
if is_color:
|
||||
offset = 10
|
||||
string_copy = string._new('')
|
||||
else:
|
||||
offset = 0
|
||||
|
||||
stack = tsplit(string, NEWLINES)
|
||||
|
||||
for i, substring in enumerate(stack):
|
||||
stack[i] = substring.split()
|
||||
|
||||
_stack = []
|
||||
|
||||
for row in stack:
|
||||
_row = ['',]
|
||||
_row_i = 0
|
||||
|
||||
for word in row:
|
||||
if (len(_row[_row_i]) + len(word)) < (cols + offset):
|
||||
_row[_row_i] += word
|
||||
_row[_row_i] += ' '
|
||||
|
||||
elif len(word) > (cols - offset):
|
||||
|
||||
# ensure empty row
|
||||
if len(_row[_row_i]):
|
||||
_row.append('')
|
||||
_row_i += 1
|
||||
|
||||
chunks = schunk(word, (cols + offset))
|
||||
for i, chunk in enumerate(chunks):
|
||||
if not (i + 1) == len(chunks):
|
||||
_row[_row_i] += chunk
|
||||
_row.append('')
|
||||
_row_i += 1
|
||||
else:
|
||||
_row[_row_i] += chunk
|
||||
_row[_row_i] += ' '
|
||||
else:
|
||||
_row.append('')
|
||||
_row_i += 1
|
||||
_row[_row_i] += word
|
||||
_row[_row_i] += ' '
|
||||
|
||||
_row = map(str, _row)
|
||||
_stack.append(separator.join(_row))
|
||||
|
||||
_s = '\n'.join(_stack)
|
||||
if is_color:
|
||||
_s = string_copy._new(_s)
|
||||
return _s
|
||||
+46
-10
@@ -12,21 +12,57 @@ from __future__ import absolute_import
|
||||
|
||||
import sys
|
||||
|
||||
STREAM = sys.stderr
|
||||
|
||||
def progressbar(it, prefix='', size=32, hide=False):
|
||||
BAR_TEMPLATE = '%s[%s%s] %i/%i\r'
|
||||
BAR_EMPTY_CHAR = '-'
|
||||
BAR_FILLED_CHAR = '='
|
||||
|
||||
DOTS_CHAR = '.'
|
||||
|
||||
|
||||
def bar(it, label='', width=32, hide=False):
|
||||
"""Progress iterator. Wrap your iterables with it."""
|
||||
count = len(it)
|
||||
if count:
|
||||
def _show(_i):
|
||||
x = int(size*_i/count)
|
||||
if not hide:
|
||||
sys.stdout.write("%s[%s>%s] %i/%i\r" % (prefix, "="*x, "-"*(size-x), _i, count))
|
||||
sys.stdout.flush()
|
||||
|
||||
def _show(_i):
|
||||
x = int(width*_i/count)
|
||||
if not hide:
|
||||
STREAM.write(BAR_TEMPLATE % (
|
||||
label, BAR_FILLED_CHAR*x, BAR_EMPTY_CHAR*(width-x), _i, count))
|
||||
STREAM.flush()
|
||||
|
||||
count = len(it)
|
||||
|
||||
if count:
|
||||
_show(0)
|
||||
|
||||
for i, item in enumerate(it):
|
||||
|
||||
yield item
|
||||
_show(i+1)
|
||||
|
||||
if not hide:
|
||||
sys.stdout.write("\n")
|
||||
sys.stdout.flush()
|
||||
STREAM.write('\n')
|
||||
STREAM.flush()
|
||||
|
||||
|
||||
def dots(it, label='', hide=False):
|
||||
"""Progress iterator. Prints a dot for each item being iterated"""
|
||||
|
||||
count = 0
|
||||
|
||||
if not hide:
|
||||
STREAM.write(label)
|
||||
|
||||
for item in it:
|
||||
if not hide:
|
||||
STREAM.write(DOTS_CHAR)
|
||||
sys.stderr.flush()
|
||||
|
||||
count += 1
|
||||
|
||||
yield item
|
||||
|
||||
STREAM.write('\n')
|
||||
STREAM.flush()
|
||||
|
||||
|
||||
+23
-1
@@ -9,6 +9,7 @@ Various Python helpers used within clint.
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import with_statement
|
||||
|
||||
import sys
|
||||
import errno
|
||||
@@ -28,7 +29,7 @@ def mkdir_p(path):
|
||||
"""Emulates `mkdir -p` behavior."""
|
||||
try:
|
||||
makedirs(path)
|
||||
except OSError as exc: # Python >2.5
|
||||
except OSError, exc: # Python >2.5
|
||||
if exc.errno == errno.EEXIST:
|
||||
pass
|
||||
else:
|
||||
@@ -49,3 +50,24 @@ def tsplit(string, delimiters):
|
||||
|
||||
return stack
|
||||
|
||||
def schunk(string, size):
|
||||
"""Splits string into n sized chunks."""
|
||||
|
||||
stack = []
|
||||
|
||||
substack = []
|
||||
current_count = 0
|
||||
|
||||
for char in string:
|
||||
if not current_count < size:
|
||||
stack.append(''.join(substack))
|
||||
substack = []
|
||||
current_count = 0
|
||||
|
||||
substack.append(char)
|
||||
current_count += 1
|
||||
|
||||
if len(substack):
|
||||
stack.append(''.join(substack))
|
||||
|
||||
return stack
|
||||
|
||||
@@ -12,5 +12,5 @@ text = 'THIS TEXT IS COLORED %s!'
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
for color in colored.__all__:
|
||||
for color in colored.COLORS:
|
||||
print getattr(colored, color)(text % color.upper())
|
||||
@@ -1,6 +1,9 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import with_statement
|
||||
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import with_statement
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import with_statement
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
+3
-1
@@ -1,6 +1,8 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import with_statement
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
@@ -21,4 +23,4 @@ if __name__ == '__main__':
|
||||
with indent(5, quote=colored.red(' |')):
|
||||
puts(in_data)
|
||||
else:
|
||||
puts(colored.red('Warning: ') + 'No data was piped in.')
|
||||
puts(colored.red('Warning: ') + 'No data was piped in.')
|
||||
|
||||
@@ -8,12 +8,13 @@ sys.path.insert(0, os.path.abspath('..'))
|
||||
|
||||
from time import sleep
|
||||
from random import random
|
||||
from clint.textui import progressbar
|
||||
from clint.textui import progress
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
for i in progressbar(range(100)):
|
||||
for i in progress.bar(range(100)):
|
||||
sleep(random() * 0.2)
|
||||
|
||||
|
||||
for i in progress.dots(range(100)):
|
||||
sleep(random() * 0.2)
|
||||
|
||||
Executable
+23
@@ -0,0 +1,23 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
sys.path.insert(0, os.path.abspath('..'))
|
||||
|
||||
from clint.textui import puts, colored
|
||||
from clint.textui import columns
|
||||
|
||||
lorem = 'Lorem ipsum dolor sit amet, consehdfhdfhdfhdfhdfhctetur adi pisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.'
|
||||
|
||||
if __name__ == '__main__':
|
||||
# puts(min_width('test\nit', 20) + ' me')
|
||||
# puts(max_width(lorem, 20) + ' me')
|
||||
|
||||
# print max_width(lorem, 45)
|
||||
|
||||
col = 60
|
||||
|
||||
puts(columns([(colored.red('Column 1')), col], [(colored.green('Column Two')), None], [(colored.magenta('Column III')), col]))
|
||||
puts(columns(['hi there my name is kenneth and this is a columns', col], [lorem, None], ['kenneths', col]))
|
||||
@@ -6,6 +6,9 @@ import sys
|
||||
|
||||
from distutils.core import setup
|
||||
|
||||
import clint
|
||||
|
||||
|
||||
|
||||
def publish():
|
||||
"""Publish to PyPi"""
|
||||
@@ -19,7 +22,7 @@ required = []
|
||||
|
||||
setup(
|
||||
name='clint',
|
||||
version='0.1.2',
|
||||
version=clint.__version__,
|
||||
description='Python Command-line Application Tools',
|
||||
long_description=open('README.rst').read() + '\n\n' +
|
||||
open('HISTORY.rst').read(),
|
||||
@@ -29,7 +32,6 @@ setup(
|
||||
packages= [
|
||||
'clint',
|
||||
'clint.textui',
|
||||
|
||||
'clint.packages', 'clint.packages.colorama'
|
||||
],
|
||||
install_requires=required,
|
||||
@@ -40,7 +42,7 @@ setup(
|
||||
'Natural Language :: English',
|
||||
'License :: OSI Approved :: ISC License (ISCL)',
|
||||
'Programming Language :: Python',
|
||||
# 'Programming Language :: Python :: 2.5',
|
||||
'Programming Language :: Python :: 2.5',
|
||||
'Programming Language :: Python :: 2.6',
|
||||
'Programming Language :: Python :: 2.7',
|
||||
# 'Programming Language :: Python :: 3.0',
|
||||
|
||||
+1
-3
@@ -6,7 +6,7 @@
|
||||
import unittest
|
||||
|
||||
|
||||
class TablibTestCase(unittest.TestCas):
|
||||
class TablibTestCase(unittest.TestCase):
|
||||
"""Tablib test cases."""
|
||||
|
||||
def setUp(self):
|
||||
@@ -16,7 +16,5 @@ class TablibTestCase(unittest.TestCas):
|
||||
def tearDown(self):
|
||||
pass
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user