mirror of
https://github.com/kennethreitz/clint.git
synced 2026-06-05 23:00:18 +00:00
Compare commits
43 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 4336c354de | |||
| 43b152cf0e | |||
| 0d0d482156 | |||
| 3a40b682f4 | |||
| 1d1bb71203 | |||
| b917e68837 | |||
| 66a6148088 | |||
| a95e803268 | |||
| 7e1c739755 | |||
| ec46d65bd2 | |||
| 73a9cd836b | |||
| c9bf3bac3a | |||
| 5f2df1218a | |||
| 526cf84ebb | |||
| 3709f090b4 | |||
| d6214ababb | |||
| ee089c2d39 | |||
| dbb063840a | |||
| 9a4705f090 | |||
| b16b2ce723 | |||
| 1ea4c5afdc | |||
| 39bb08253f | |||
| 8e6973253f | |||
| 84419f6bfb | |||
| 53ec4c2e09 | |||
| 7b6138eda6 | |||
| d19650b93c | |||
| 98edec43e9 | |||
| ef48a47b8d | |||
| 3be47add4a | |||
| bf99084ece | |||
| 705cda4443 | |||
| 79cf7a023b | |||
| 7900853dbc | |||
| 96827698df | |||
| 23689f8b0a | |||
| 1a9e15dbba | |||
| a07602fdeb | |||
| 0855525fca | |||
| 872c78a5b9 | |||
| 5d7485c144 | |||
| 25b8d6a10e | |||
| 003411ebcd |
+1
-1
@@ -2,4 +2,4 @@
|
||||
MANIFEST
|
||||
*.pyc
|
||||
.tox
|
||||
*~
|
||||
build/
|
||||
|
||||
@@ -14,3 +14,8 @@ Patches and Suggestions
|
||||
- Morgan Goose
|
||||
- Travis Swicegood
|
||||
- Will Thames
|
||||
- Greg Haskins
|
||||
- Miguel Araujo <maraujop>
|
||||
- takluyver
|
||||
- kracekumar
|
||||
- Alejandro Gómez <alejandrogomez>
|
||||
|
||||
@@ -11,4 +11,4 @@ All functionality should be available in pure Python. Optional C (via Cython)
|
||||
implementations may be written for performance reasons, but should never
|
||||
replace the Python implementation.
|
||||
|
||||
Lastly, don't take yourself too seriously :)
|
||||
Lastly, don't take yourself too seriously :)
|
||||
|
||||
+20
@@ -1,6 +1,26 @@
|
||||
History
|
||||
-------
|
||||
|
||||
0.3.1
|
||||
+++++
|
||||
|
||||
* Progress mill indicator.
|
||||
* Colored unicode support.
|
||||
* Fix ipython nuance.
|
||||
|
||||
|
||||
0.3.0
|
||||
+++++
|
||||
|
||||
* Python 3 support!
|
||||
|
||||
0.2.4
|
||||
+++++
|
||||
|
||||
* New eng module
|
||||
* Win32 Bugfix
|
||||
|
||||
|
||||
0.2.3
|
||||
+++++
|
||||
|
||||
|
||||
+3
-3
@@ -19,11 +19,11 @@ from .pipes import piped_in
|
||||
|
||||
|
||||
__title__ = 'clint'
|
||||
__version__ = '0.2.3'
|
||||
__build__ = 0x000203
|
||||
__version__ = '0.3.1'
|
||||
__build__ = 0x000301
|
||||
__author__ = 'Kenneth Reitz'
|
||||
__license__ = 'ISC'
|
||||
__copyright__ = 'Copyright 2011 Kenneth Reitz'
|
||||
__copyright__ = 'Copyright 2012 Kenneth Reitz'
|
||||
__docformat__ = 'restructuredtext'
|
||||
|
||||
|
||||
|
||||
+22
-38
@@ -13,33 +13,17 @@ from __future__ import absolute_import
|
||||
|
||||
import os
|
||||
from sys import argv
|
||||
from glob import glob
|
||||
|
||||
from .packages.ordereddict import OrderedDict
|
||||
from .utils import is_collection
|
||||
|
||||
try:
|
||||
from collections import OrderedDict
|
||||
except ImportError:
|
||||
from .packages.ordereddict import OrderedDict
|
||||
|
||||
from .utils import expand_path, is_collection
|
||||
|
||||
__all__ = ('Args', )
|
||||
|
||||
|
||||
|
||||
def _expand_path(path):
|
||||
"""Expands directories and globs in given path."""
|
||||
|
||||
paths = []
|
||||
|
||||
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))
|
||||
|
||||
return paths
|
||||
|
||||
|
||||
class Args(object):
|
||||
"""CLI Argument management."""
|
||||
|
||||
@@ -110,20 +94,20 @@ class Args(object):
|
||||
|
||||
def any_contain(self, x):
|
||||
"""Tests if given string is contained in any stored argument."""
|
||||
|
||||
|
||||
return bool(self.first_with(x))
|
||||
|
||||
|
||||
def contains(self, x):
|
||||
"""Tests if given object is in arguments list.
|
||||
"""Tests if given object is in arguments list.
|
||||
Accepts strings and lists of strings."""
|
||||
|
||||
|
||||
return self.__contains__(x)
|
||||
|
||||
|
||||
def first(self, x):
|
||||
"""Returns first found index of given value (or list of values)"""
|
||||
|
||||
|
||||
def _find( x):
|
||||
try:
|
||||
return self.all.index(str(x))
|
||||
@@ -212,7 +196,7 @@ class Args(object):
|
||||
return False
|
||||
else:
|
||||
return (x in self.all[index])
|
||||
|
||||
|
||||
except IndexError:
|
||||
return False
|
||||
|
||||
@@ -221,7 +205,7 @@ class Args(object):
|
||||
"""Returns true if argument exists at given index.
|
||||
Accepts: integer.
|
||||
"""
|
||||
|
||||
|
||||
try:
|
||||
self.all[x]
|
||||
return True
|
||||
@@ -231,15 +215,15 @@ class Args(object):
|
||||
|
||||
def value_after(self, x):
|
||||
"""Returns value of argument after given found argument (or list thereof)."""
|
||||
|
||||
|
||||
try:
|
||||
try:
|
||||
i = self.all.index(x)
|
||||
except ValueError:
|
||||
return None
|
||||
|
||||
|
||||
return self.all[i + 1]
|
||||
|
||||
|
||||
except IndexError:
|
||||
return None
|
||||
|
||||
@@ -266,21 +250,21 @@ class Args(object):
|
||||
|
||||
return collection
|
||||
|
||||
|
||||
|
||||
@property
|
||||
def last(self):
|
||||
"""Returns last argument."""
|
||||
|
||||
|
||||
try:
|
||||
return self.all[-1]
|
||||
except IndexError:
|
||||
return None
|
||||
|
||||
|
||||
|
||||
@property
|
||||
def all(self):
|
||||
"""Returns all arguments."""
|
||||
|
||||
|
||||
return self._args
|
||||
|
||||
|
||||
@@ -288,7 +272,7 @@ class Args(object):
|
||||
"""Returns all arguments containing given string (or list thereof)"""
|
||||
|
||||
_args = []
|
||||
|
||||
|
||||
for arg in self.all:
|
||||
if is_collection(x):
|
||||
for _x in x:
|
||||
@@ -327,7 +311,7 @@ class Args(object):
|
||||
return self.start_with('-')
|
||||
|
||||
|
||||
@property
|
||||
@property
|
||||
def not_flags(self):
|
||||
"""Returns Arg object excluding flagged arguments."""
|
||||
|
||||
@@ -341,7 +325,7 @@ class Args(object):
|
||||
_paths = []
|
||||
|
||||
for arg in self.all:
|
||||
for path in _expand_path(arg):
|
||||
for path in expand_path(arg):
|
||||
if os.path.exists(path):
|
||||
if absolute:
|
||||
_paths.append(os.path.abspath(path))
|
||||
@@ -358,7 +342,7 @@ class Args(object):
|
||||
_args = []
|
||||
|
||||
for arg in self.all:
|
||||
if not len(_expand_path(arg)):
|
||||
if not len(expand_path(arg)):
|
||||
if not os.path.exists(arg):
|
||||
_args.append(arg)
|
||||
|
||||
|
||||
@@ -0,0 +1,55 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
clint.eng
|
||||
~~~~~~~~~
|
||||
|
||||
This module provides English language string helpers.
|
||||
|
||||
"""
|
||||
from __future__ import print_function
|
||||
|
||||
MORON_MODE = False
|
||||
COMMA = ','
|
||||
CONJUNCTION = 'and'
|
||||
SPACE = ' '
|
||||
|
||||
try:
|
||||
unicode
|
||||
except NameError:
|
||||
unicode = str
|
||||
|
||||
|
||||
def join(l, conj=CONJUNCTION, im_a_moron=MORON_MODE, seperator=COMMA):
|
||||
"""Joins lists of words. Oxford comma and all."""
|
||||
|
||||
collector = []
|
||||
left = len(l)
|
||||
seperator = seperator + SPACE
|
||||
conj = conj + SPACE
|
||||
|
||||
for _l in l[:]:
|
||||
|
||||
left += -1
|
||||
|
||||
collector.append(_l)
|
||||
if left == 1:
|
||||
if len(l) == 2 or im_a_moron:
|
||||
collector.append(SPACE)
|
||||
else:
|
||||
collector.append(seperator)
|
||||
|
||||
collector.append(conj)
|
||||
|
||||
elif left is not 0:
|
||||
collector.append(seperator)
|
||||
|
||||
return unicode(str().join(collector))
|
||||
|
||||
if __name__ == '__main__':
|
||||
print(join(['blue', 'red', 'yellow'], conj='or', im_a_moron=True))
|
||||
print(join(['blue', 'red', 'yellow'], conj='or'))
|
||||
print(join(['blue', 'red'], conj='or'))
|
||||
print(join(['blue', 'red'], conj='and'))
|
||||
print(join(['blue'], conj='and'))
|
||||
print(join(['blue', 'red', 'yellow', 'green', 'ello'], conj='and'))
|
||||
@@ -1,6 +1,6 @@
|
||||
from .initialise import init
|
||||
from .initialise import init, deinit, reinit
|
||||
from .ansi import Fore, Back, Style
|
||||
from .ansitowin32 import AnsiToWin32
|
||||
|
||||
VERSION = '0.1.18'
|
||||
VERSION = '0.2.3'
|
||||
|
||||
|
||||
@@ -2,13 +2,11 @@
|
||||
This module generates ANSI character codes to printing colors to terminals.
|
||||
See: http://en.wikipedia.org/wiki/ANSI_escape_code
|
||||
'''
|
||||
from . import caps
|
||||
|
||||
CSI = '\033['
|
||||
|
||||
def code_to_chars(codes):
|
||||
codes = [str(code) for code in codes]
|
||||
return CSI + (";".join(codes)) + 'm'
|
||||
def code_to_chars(code):
|
||||
return CSI + str(code) + 'm'
|
||||
|
||||
class AnsiCodes(object):
|
||||
def __init__(self, codes):
|
||||
@@ -16,44 +14,34 @@ class AnsiCodes(object):
|
||||
if not name.startswith('_'):
|
||||
value = getattr(codes, name)
|
||||
setattr(self, name, code_to_chars(value))
|
||||
self._method = codes._caps_method
|
||||
|
||||
def __call__(self, code):
|
||||
if isinstance(code, basestring):
|
||||
return getattr(self, code)
|
||||
else: # extended (pass to capability)
|
||||
return code_to_chars(getattr(caps.capability, self._method)(code))
|
||||
|
||||
class AnsiFore:
|
||||
_caps_method = 'fg'
|
||||
BLACK = (30,)
|
||||
RED = (31,)
|
||||
GREEN = (32,)
|
||||
YELLOW = (33,)
|
||||
BLUE = (34,)
|
||||
MAGENTA = (35,)
|
||||
CYAN = (36,)
|
||||
WHITE = (37,)
|
||||
RESET = (39,)
|
||||
BLACK = 30
|
||||
RED = 31
|
||||
GREEN = 32
|
||||
YELLOW = 33
|
||||
BLUE = 34
|
||||
MAGENTA = 35
|
||||
CYAN = 36
|
||||
WHITE = 37
|
||||
RESET = 39
|
||||
|
||||
class AnsiBack:
|
||||
_caps_method = 'bg'
|
||||
BLACK = (40,)
|
||||
RED = (41,)
|
||||
GREEN = (42,)
|
||||
YELLOW = (43,)
|
||||
BLUE = (44,)
|
||||
MAGENTA = (45,)
|
||||
CYAN = (46,)
|
||||
WHITE = (47,)
|
||||
RESET = (49,)
|
||||
BLACK = 40
|
||||
RED = 41
|
||||
GREEN = 42
|
||||
YELLOW = 43
|
||||
BLUE = 44
|
||||
MAGENTA = 45
|
||||
CYAN = 46
|
||||
WHITE = 47
|
||||
RESET = 49
|
||||
|
||||
class AnsiStyle:
|
||||
_caps_method = 'style'
|
||||
BRIGHT = (1,)
|
||||
DIM = (2,)
|
||||
NORMAL = (22,)
|
||||
RESET_ALL = (0,)
|
||||
BRIGHT = 1
|
||||
DIM = 2
|
||||
NORMAL = 22
|
||||
RESET_ALL = 0
|
||||
|
||||
Fore = AnsiCodes( AnsiFore )
|
||||
Back = AnsiCodes( AnsiBack )
|
||||
|
||||
@@ -118,12 +118,12 @@ class AnsiToWin32(object):
|
||||
self.wrapped.flush()
|
||||
if self.autoreset:
|
||||
self.reset_all()
|
||||
|
||||
|
||||
|
||||
def reset_all(self):
|
||||
if self.convert:
|
||||
self.call_win32('m', (0,))
|
||||
else:
|
||||
elif is_a_tty(self.wrapped):
|
||||
self.wrapped.write(Style.RESET_ALL)
|
||||
|
||||
|
||||
@@ -173,4 +173,10 @@ class AnsiToWin32(object):
|
||||
args = func_args[1:]
|
||||
kwargs = dict(on_stderr=self.on_stderr)
|
||||
func(*args, **kwargs)
|
||||
elif command in ('H', 'f'): # set cursor position
|
||||
func = winterm.set_cursor_position
|
||||
func(params, on_stderr=self.on_stderr)
|
||||
elif command in ('J'):
|
||||
func = winterm.erase_data
|
||||
func(params, on_stderr=self.on_stderr)
|
||||
|
||||
|
||||
@@ -1,59 +0,0 @@
|
||||
from .conv import xterm256
|
||||
from .conv import ansi
|
||||
|
||||
class _Base(object):
|
||||
"""
|
||||
Base 16-color ANSI.
|
||||
All ANSI terminals as well as Windows support this.
|
||||
"""
|
||||
def fg(self, rgb):
|
||||
(colid, bright) = ansi.from_rgb(rgb)
|
||||
if bright:
|
||||
return (1, 30 + colid)
|
||||
else:
|
||||
return (30 + colid,)
|
||||
|
||||
def bg(self, rgb):
|
||||
(colid, bright) = ansi.from_rgb(rgb)
|
||||
# Ignore brightness for background?
|
||||
return (40 + colid)
|
||||
|
||||
def __str__():
|
||||
return "BASE"
|
||||
|
||||
class _XTerm256(object):
|
||||
"""
|
||||
256-color terminal. Most modern terminal emulators support this.
|
||||
Putty, Gnome Terminal, and even oldies like (m)rxvt, xterm.
|
||||
"""
|
||||
def fg(self, rgb):
|
||||
return (38, 5, xterm256.from_rgb(rgb))
|
||||
|
||||
def bg(self, rgb):
|
||||
return (48, 5, xterm256.from_rgb(rgb))
|
||||
|
||||
def __str__():
|
||||
return "XTERM256"
|
||||
|
||||
class _RGB(object):
|
||||
"""
|
||||
Full 24-bit RGB support.
|
||||
As far as I know, only Konsole can do this.
|
||||
"""
|
||||
def fg(self, rgb):
|
||||
return (38, 2) + rgb
|
||||
|
||||
def bg(self, rgb):
|
||||
return (48, 2) + rgb
|
||||
|
||||
def __str__():
|
||||
return "RGB"
|
||||
|
||||
class ColorCapability:
|
||||
ANSI = _Base() # Base 16-color ANSI support
|
||||
XTERM256 = _XTerm256() # xterm256 (GNOME terminal / Putty)
|
||||
RGB = _RGB() # Full RGB (Konsole)
|
||||
|
||||
# Current terminal caps, default to ANSI
|
||||
capability = ColorCapability.ANSI
|
||||
|
||||
@@ -1,69 +0,0 @@
|
||||
"""
|
||||
ANSI 16-color conversion utilities.
|
||||
"""
|
||||
# Wladimir van der Laan, 2011
|
||||
#
|
||||
# The original ansi colors are defined as follows
|
||||
#
|
||||
# 0 black
|
||||
# 1 red
|
||||
# 2 green
|
||||
# 3 brown (green+red)
|
||||
# 4 blue
|
||||
# 5 magenta (blue+red)
|
||||
# 6 cyan (blue+green)
|
||||
# 7 white
|
||||
#
|
||||
# However, the exact RGB mappings depend on the terminal that
|
||||
# is used.
|
||||
# In addition to these colors, nearly all terminals support 8
|
||||
# colors that are simply brighter versions of the defined colors.
|
||||
|
||||
_colors = [
|
||||
(0,0,0),
|
||||
(128,0,0),
|
||||
(0,128,0),
|
||||
(128,128,0),
|
||||
(0,0,128),
|
||||
(128,0,128),
|
||||
(0,128,128),
|
||||
(192,192,192),
|
||||
(128,128,128),
|
||||
(255,0,0),
|
||||
(0,255,0),
|
||||
(255,255,0),
|
||||
(0,0,255),
|
||||
(255,0,255),
|
||||
(0,255,255),
|
||||
(255,255,255)
|
||||
]
|
||||
|
||||
def to_rgb(color):
|
||||
"""
|
||||
Convert ANSI color tuple (color_id, bright) to RGB tuple.
|
||||
Raise `ValueError` in case of invalid color spec.
|
||||
"""
|
||||
color = (color[1]<<3)|color[0]
|
||||
try:
|
||||
return _colors[color]
|
||||
except IndexError:
|
||||
raise ValueError("Color %i out of range" % color)
|
||||
|
||||
def from_rgb(rgb):
|
||||
"""
|
||||
Convert RGB tuple to ANSI color tuple (color_id, bright).
|
||||
Raise `ValueError` in case of invalid color spec.
|
||||
"""
|
||||
min_d = 1000000
|
||||
closest = None
|
||||
for idx,col in enumerate(_colors):
|
||||
dr = rgb[0]-col[0]
|
||||
dg = rgb[1]-col[1]
|
||||
db = rgb[2]-col[2]
|
||||
d = dr*dr + dg*dg + db*db
|
||||
if d < min_d:
|
||||
min_d = d
|
||||
closest = idx
|
||||
return (closest&7, closest>>3)
|
||||
|
||||
|
||||
@@ -1,184 +0,0 @@
|
||||
"""
|
||||
HTML color conversion utilities.
|
||||
"""
|
||||
# Wladimir van der Laan, 2011
|
||||
|
||||
colors = {
|
||||
'aliceblue': '#F0F8FF',
|
||||
'antiquewhite': '#FAEBD7',
|
||||
'aqua': '#00FFFF',
|
||||
'aquamarine': '#7FFFD4',
|
||||
'azure': '#F0FFFF',
|
||||
'beige': '#F5F5DC',
|
||||
'bisque': '#FFE4C4',
|
||||
'black': '#000000',
|
||||
'blanchedalmond': '#FFEBCD',
|
||||
'blue': '#0000FF',
|
||||
'blueviolet': '#8A2BE2',
|
||||
'brown': '#A52A2A',
|
||||
'burlywood': '#DEB887',
|
||||
'cadetblue': '#5F9EA0',
|
||||
'chartreuse': '#7FFF00',
|
||||
'chocolate': '#D2691E',
|
||||
'coral': '#FF7F50',
|
||||
'cornflowerblue': '#6495ED',
|
||||
'cornsilk': '#FFF8DC',
|
||||
'crimson': '#DC143C',
|
||||
'cyan': '#00FFFF',
|
||||
'darkblue': '#00008B',
|
||||
'darkcyan': '#008B8B',
|
||||
'darkgoldenrod': '#B8860B',
|
||||
'darkgray': '#A9A9A9',
|
||||
'darkgreen': '#006400',
|
||||
'darkgrey': '#A9A9A9',
|
||||
'darkkhaki': '#BDB76B',
|
||||
'darkmagenta': '#8B008B',
|
||||
'darkolivegreen': '#556B2F',
|
||||
'darkorange': '#FF8C00',
|
||||
'darkorchid': '#9932CC',
|
||||
'darkred': '#8B0000',
|
||||
'darksalmon': '#E9967A',
|
||||
'darkseagreen': '#8FBC8F',
|
||||
'darkslateblue': '#483D8B',
|
||||
'darkslategray': '#2F4F4F',
|
||||
'darkslategrey': '#2F4F4F',
|
||||
'darkturquoise': '#00CED1',
|
||||
'darkviolet': '#9400D3',
|
||||
'deeppink': '#FF1493',
|
||||
'deepskyblue': '#00BFFF',
|
||||
'dimgray': '#696969',
|
||||
'dimgrey': '#696969',
|
||||
'dodgerblue': '#1E90FF',
|
||||
'firebrick': '#B22222',
|
||||
'floralwhite': '#FFFAF0',
|
||||
'forestgreen': '#228B22',
|
||||
'fuchsia': '#FF00FF',
|
||||
'gainsboro': '#DCDCDC',
|
||||
'ghostwhite': '#F8F8FF',
|
||||
'gold': '#FFD700',
|
||||
'goldenrod': '#DAA520',
|
||||
'gray': '#808080',
|
||||
'green': '#008000',
|
||||
'greenyellow': '#ADFF2F',
|
||||
'grey': '#808080',
|
||||
'honeydew': '#F0FFF0',
|
||||
'hotpink': '#FF69B4',
|
||||
'indianred ': '#CD5C5C',
|
||||
'indigo ': '#4B0082',
|
||||
'ivory': '#FFFFF0',
|
||||
'khaki': '#F0E68C',
|
||||
'lavender': '#E6E6FA',
|
||||
'lavenderblush': '#FFF0F5',
|
||||
'lawngreen': '#7CFC00',
|
||||
'lemonchiffon': '#FFFACD',
|
||||
'lightblue': '#ADD8E6',
|
||||
'lightcoral': '#F08080',
|
||||
'lightcyan': '#E0FFFF',
|
||||
'lightgoldenrodyellow': '#FAFAD2',
|
||||
'lightgray': '#D3D3D3',
|
||||
'lightgreen': '#90EE90',
|
||||
'lightgrey': '#D3D3D3',
|
||||
'lightpink': '#FFB6C1',
|
||||
'lightsalmon': '#FFA07A',
|
||||
'lightseagreen': '#20B2AA',
|
||||
'lightskyblue': '#87CEFA',
|
||||
'lightslategray': '#778899',
|
||||
'lightslategrey': '#778899',
|
||||
'lightsteelblue': '#B0C4DE',
|
||||
'lightyellow': '#FFFFE0',
|
||||
'lime': '#00FF00',
|
||||
'limegreen': '#32CD32',
|
||||
'linen': '#FAF0E6',
|
||||
'magenta': '#FF00FF',
|
||||
'maroon': '#800000',
|
||||
'mediumaquamarine': '#66CDAA',
|
||||
'mediumblue': '#0000CD',
|
||||
'mediumorchid': '#BA55D3',
|
||||
'mediumpurple': '#9370D8',
|
||||
'mediumseagreen': '#3CB371',
|
||||
'mediumslateblue': '#7B68EE',
|
||||
'mediumspringgreen': '#00FA9A',
|
||||
'mediumturquoise': '#48D1CC',
|
||||
'mediumvioletred': '#C71585',
|
||||
'midnightblue': '#191970',
|
||||
'mintcream': '#F5FFFA',
|
||||
'mistyrose': '#FFE4E1',
|
||||
'moccasin': '#FFE4B5',
|
||||
'navajowhite': '#FFDEAD',
|
||||
'navy': '#000080',
|
||||
'oldlace': '#FDF5E6',
|
||||
'olive': '#808000',
|
||||
'olivedrab': '#6B8E23',
|
||||
'orange': '#FFA500',
|
||||
'orangered': '#FF4500',
|
||||
'orchid': '#DA70D6',
|
||||
'palegoldenrod': '#EEE8AA',
|
||||
'palegreen': '#98FB98',
|
||||
'paleturquoise': '#AFEEEE',
|
||||
'palevioletred': '#D87093',
|
||||
'papayawhip': '#FFEFD5',
|
||||
'peachpuff': '#FFDAB9',
|
||||
'peru': '#CD853F',
|
||||
'pink': '#FFC0CB',
|
||||
'plum': '#DDA0DD',
|
||||
'powderblue': '#B0E0E6',
|
||||
'purple': '#800080',
|
||||
'red': '#FF0000',
|
||||
'rosybrown': '#BC8F8F',
|
||||
'royalblue': '#4169E1',
|
||||
'saddlebrown': '#8B4513',
|
||||
'salmon': '#FA8072',
|
||||
'sandybrown': '#F4A460',
|
||||
'seagreen': '#2E8B57',
|
||||
'seashell': '#FFF5EE',
|
||||
'sienna': '#A0522D',
|
||||
'silver': '#C0C0C0',
|
||||
'skyblue': '#87CEEB',
|
||||
'slateblue': '#6A5ACD',
|
||||
'slategray': '#708090',
|
||||
'slategrey': '#708090',
|
||||
'snow': '#FFFAFA',
|
||||
'springgreen': '#00FF7F',
|
||||
'steelblue': '#4682B4',
|
||||
'tan': '#D2B48C',
|
||||
'teal': '#008080',
|
||||
'thistle': '#D8BFD8',
|
||||
'tomato': '#FF6347',
|
||||
'turquoise': '#40E0D0',
|
||||
'violet': '#EE82EE',
|
||||
'wheat': '#F5DEB3',
|
||||
'white': '#FFFFFF',
|
||||
'whitesmoke': '#F5F5F5',
|
||||
'yellow': '#FFFF00',
|
||||
'yellowgreen': '#9ACD32'}
|
||||
|
||||
def to_rgb(color):
|
||||
"""
|
||||
Parse HTML color specification to RGB tuple.
|
||||
Raise `ValueError` in case of invalid color spec.
|
||||
"""
|
||||
if color in colors:
|
||||
color = colors[color]
|
||||
|
||||
if color.startswith("#"):
|
||||
color = color[1:]
|
||||
|
||||
if len(color) == 6: #RRGGBB
|
||||
rgb = (int(color[0:2],16),
|
||||
int(color[2:4],16),
|
||||
int(color[4:6],16))
|
||||
elif len(color) == 3: #RGB
|
||||
rgb = (int(color[0],16),
|
||||
int(color[1],16),
|
||||
int(color[2],16))
|
||||
rgb = tuple([(x<<4)|x for x in rgb])
|
||||
else:
|
||||
raise ValueError("Invalid color %s" % color)
|
||||
|
||||
return rgb
|
||||
|
||||
def from_rgb(rgb):
|
||||
"""
|
||||
Convert RGB tuple to HTML color specification.
|
||||
"""
|
||||
return "#%02x%02x%02x" % rgb
|
||||
@@ -1,115 +0,0 @@
|
||||
"""
|
||||
256-color xterm conversion utilities.
|
||||
"""
|
||||
# Wladimir van der Laan, 2011
|
||||
|
||||
# whole colortable, filled in later
|
||||
colortable = None
|
||||
|
||||
# the 6 value iterations in the xterm color cube
|
||||
valuerange = [ 0x00, 0x5F, 0x87, 0xAF, 0xD7, 0xFF ]
|
||||
|
||||
# 16 basic ANSI colors + lighter variants
|
||||
# these can be different depending on the terminal settings,
|
||||
# so do not use them when converting from RGB.
|
||||
basic16 = [
|
||||
[ 0x00, 0x00, 0x00 ], # 0
|
||||
[ 0xCD, 0x00, 0x00 ], # 1
|
||||
[ 0x00, 0xCD, 0x00 ], # 2
|
||||
[ 0xCD, 0xCD, 0x00 ], # 3
|
||||
[ 0x00, 0x00, 0xEE ], # 4
|
||||
[ 0xCD, 0x00, 0xCD ], # 5
|
||||
[ 0x00, 0xCD, 0xCD ], # 6
|
||||
[ 0xE5, 0xE5, 0xE5 ], # 7
|
||||
[ 0x7F, 0x7F, 0x7F ], # 8
|
||||
[ 0xFF, 0x00, 0x00 ], # 9
|
||||
[ 0x00, 0xFF, 0x00 ], # 10
|
||||
[ 0xFF, 0xFF, 0x00 ], # 11
|
||||
[ 0x5C, 0x5C, 0xFF ], # 12
|
||||
[ 0xFF, 0x00, 0xFF ], # 13
|
||||
[ 0x00, 0xFF, 0xFF ], # 14
|
||||
[ 0xFF, 0xFF, 0xFF ] # 15
|
||||
]
|
||||
|
||||
# Closest color on RGB color cube (from valuerange)
|
||||
closest6 = [
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
|
||||
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
|
||||
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
|
||||
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
|
||||
1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
|
||||
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
|
||||
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3,
|
||||
3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
|
||||
3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
|
||||
3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
|
||||
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
|
||||
4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5,
|
||||
5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5]
|
||||
|
||||
def _color_dist(c, rgb):
|
||||
"""
|
||||
Return squared Euclidian distance of rgb to color `c`
|
||||
in the color table.
|
||||
"""
|
||||
d0 = colortable[c][0] - rgb[0]
|
||||
d1 = colortable[c][1] - rgb[1]
|
||||
d2 = colortable[c][2] - rgb[2]
|
||||
return d0*d0 + d1*d1 + d2*d2
|
||||
|
||||
def to_rgb(color):
|
||||
"""
|
||||
Convert a xterm256 color value (0-255) to 3 unsigned chars RGB
|
||||
tuple.
|
||||
"""
|
||||
rgb = [0,0,0]
|
||||
if color < 16:
|
||||
# 16 basic colors
|
||||
rgb[0] = basic16[color][0]
|
||||
rgb[1] = basic16[color][1]
|
||||
rgb[2] = basic16[color][2]
|
||||
elif color >= 16 and color <= 231:
|
||||
# color cube color
|
||||
color -= 16
|
||||
rgb[0] = valuerange[(color/36)%6]
|
||||
rgb[1] = valuerange[(color/6)%6]
|
||||
rgb[2] = valuerange[color%6]
|
||||
elif color >= 232 and color <= 255:
|
||||
# gray tone
|
||||
rgb[0] = rgb[1] = rgb[2] = 8+(color-232)*0x0a
|
||||
|
||||
return rgb
|
||||
|
||||
def from_rgb(rgb):
|
||||
"""
|
||||
Convert RGB tuple to xterm256 color value (0-255).
|
||||
"""
|
||||
# Optimized algorithm to find the xterm256 color with Euclidian closest distance to a
|
||||
# provided RGB value: as the palette consists of an independent 6x6x6 RGB cube
|
||||
# and a 24 grey tone scale, determine the closest color in both palettes, then
|
||||
# from these two return the on that's closest to the requested RGB color.
|
||||
|
||||
# Compute closest color on 6x6x6 RGB cube
|
||||
r = [closest6[x] for x in rgb]
|
||||
b = 16 + r[0]*36 + r[1]*6 + r[2]
|
||||
|
||||
# Compute closest point on greyscale line
|
||||
greyscale = sum(rgb)
|
||||
r = (greyscale-9) / 30
|
||||
c = 232 + max(min(r,23),0)
|
||||
|
||||
# Return RGB cube or greyscale color depending
|
||||
# on which is closest.
|
||||
distb = _color_dist(b, rgb)
|
||||
distc = _color_dist(c, rgb)
|
||||
if distb <= distc:
|
||||
color = b
|
||||
else:
|
||||
color = c
|
||||
|
||||
return color
|
||||
|
||||
colortable = [to_rgb(c) for c in xrange(0, 256)]
|
||||
@@ -7,6 +7,9 @@ from .ansitowin32 import AnsiToWin32
|
||||
orig_stdout = sys.stdout
|
||||
orig_stderr = sys.stderr
|
||||
|
||||
wrapped_stdout = sys.stdout
|
||||
wrapped_stderr = sys.stderr
|
||||
|
||||
atexit_done = False
|
||||
|
||||
|
||||
@@ -16,11 +19,14 @@ def reset_all():
|
||||
|
||||
def init(autoreset=False, convert=None, strip=None, wrap=True):
|
||||
|
||||
if wrap==False and (autoreset==True or convert==True or strip==True):
|
||||
if not wrap and any([autoreset, convert, strip]):
|
||||
raise ValueError('wrap=False conflicts with any other arg=True')
|
||||
|
||||
sys.stdout = wrap_stream(orig_stdout, convert, strip, autoreset, wrap)
|
||||
sys.stderr = wrap_stream(orig_stderr, convert, strip, autoreset, wrap)
|
||||
global wrapped_stdout, wrapped_stderr
|
||||
sys.stdout = wrapped_stdout = \
|
||||
wrap_stream(orig_stdout, convert, strip, autoreset, wrap)
|
||||
sys.stderr = wrapped_stderr = \
|
||||
wrap_stream(orig_stderr, convert, strip, autoreset, wrap)
|
||||
|
||||
global atexit_done
|
||||
if not atexit_done:
|
||||
@@ -28,6 +34,16 @@ def init(autoreset=False, convert=None, strip=None, wrap=True):
|
||||
atexit_done = True
|
||||
|
||||
|
||||
def deinit():
|
||||
sys.stdout = orig_stdout
|
||||
sys.stderr = orig_stderr
|
||||
|
||||
|
||||
def reinit():
|
||||
sys.stdout = wrapped_stdout
|
||||
sys.stderr = wrapped_stdout
|
||||
|
||||
|
||||
def wrap_stream(stream, convert, strip, autoreset, wrap):
|
||||
if wrap:
|
||||
wrapper = AnsiToWin32(stream,
|
||||
@@ -36,3 +52,4 @@ def wrap_stream(stream, convert, strip, autoreset, wrap):
|
||||
stream = wrapper.stream
|
||||
return stream
|
||||
|
||||
|
||||
|
||||
@@ -48,8 +48,16 @@ else:
|
||||
("srWindow", SMALL_RECT),
|
||||
("dwMaximumWindowSize", COORD),
|
||||
]
|
||||
def __str__(self):
|
||||
return '(%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d)' % (
|
||||
self.dwSize.Y, self.dwSize.X
|
||||
, self.dwCursorPosition.Y, self.dwCursorPosition.X
|
||||
, self.wAttributes
|
||||
, self.srWindow.Top, self.srWindow.Left, self.srWindow.Bottom, self.srWindow.Right
|
||||
, self.dwMaximumWindowSize.Y, self.dwMaximumWindowSize.X
|
||||
)
|
||||
|
||||
def GetConsoleScreenBufferInfo(stream_id):
|
||||
def GetConsoleScreenBufferInfo(stream_id=STDOUT):
|
||||
handle = handles[stream_id]
|
||||
csbi = CONSOLE_SCREEN_BUFFER_INFO()
|
||||
success = windll.kernel32.GetConsoleScreenBufferInfo(
|
||||
@@ -62,34 +70,54 @@ else:
|
||||
|
||||
def SetConsoleTextAttribute(stream_id, attrs):
|
||||
handle = handles[stream_id]
|
||||
success = windll.kernel32.SetConsoleTextAttribute(handle, attrs)
|
||||
assert success
|
||||
return windll.kernel32.SetConsoleTextAttribute(handle, attrs)
|
||||
|
||||
def SetConsoleCursorPosition(stream_id, position):
|
||||
handle = handles[stream_id]
|
||||
position = COORD(*position)
|
||||
success = windll.kernel32.SetConsoleCursorPosition(handle, position)
|
||||
assert success
|
||||
# If the position is out of range, do nothing.
|
||||
if position.Y <= 0 or position.X <= 0:
|
||||
return
|
||||
# Adjust for Windows' SetConsoleCursorPosition:
|
||||
# 1. being 0-based, while ANSI is 1-based.
|
||||
# 2. expecting (x,y), while ANSI uses (y,x).
|
||||
adjusted_position = COORD(position.Y - 1, position.X - 1)
|
||||
# Adjust for viewport's scroll position
|
||||
sr = GetConsoleScreenBufferInfo(STDOUT).srWindow
|
||||
adjusted_position.Y += sr.Top
|
||||
adjusted_position.X += sr.Left
|
||||
# Resume normal processing
|
||||
handle = handles[stream_id]
|
||||
success = windll.kernel32.SetConsoleCursorPosition(handle, adjusted_position)
|
||||
return success
|
||||
|
||||
def FillConsoleOutputCharacter(stream_id, char, length, start):
|
||||
handle = handles[stream_id]
|
||||
char = TCHAR(char)
|
||||
length = DWORD(length)
|
||||
start = COORD(*start)
|
||||
num_written = DWORD(0)
|
||||
# AttributeError: function 'FillConsoleOutputCharacter' not found
|
||||
# could it just be that my types are wrong?
|
||||
success = windll.kernel32.FillConsoleOutputCharacter(
|
||||
# Note that this is hard-coded for ANSI (vs wide) bytes.
|
||||
success = windll.kernel32.FillConsoleOutputCharacterA(
|
||||
handle, char, length, start, byref(num_written))
|
||||
assert success
|
||||
return num_written.value
|
||||
|
||||
def FillConsoleOutputAttribute(stream_id, attr, length, start):
|
||||
''' FillConsoleOutputAttribute( hConsole, csbi.wAttributes, dwConSize, coordScreen, &cCharsWritten )'''
|
||||
handle = handles[stream_id]
|
||||
attribute = WORD(attr)
|
||||
length = DWORD(length)
|
||||
num_written = DWORD(0)
|
||||
# Note that this is hard-coded for ANSI (vs wide) bytes.
|
||||
success = windll.kernel32.FillConsoleOutputAttribute(
|
||||
handle, attribute, length, start, byref(num_written))
|
||||
return success
|
||||
|
||||
|
||||
if __name__=='__main__':
|
||||
x = GetConsoleScreenBufferInfo(STDOUT)
|
||||
print(x.dwSize)
|
||||
print(x.dwCursorPosition)
|
||||
print(x.wAttributes)
|
||||
print(x.srWindow)
|
||||
print(x.dwMaximumWindowSize)
|
||||
print(x)
|
||||
print('dwSize(height,width) = (%d,%d)' % (x.dwSize.Y, x.dwSize.X))
|
||||
print('dwCursorPosition(y,x) = (%d,%d)' % (x.dwCursorPosition.Y, x.dwCursorPosition.X))
|
||||
print('wAttributes(color) = %d = 0x%02x' % (x.wAttributes, x.wAttributes))
|
||||
print('srWindow(Top,Left)-(Bottom,Right) = (%d,%d)-(%d,%d)' % (x.srWindow.Top, x.srWindow.Left, x.srWindow.Bottom, x.srWindow.Right))
|
||||
print('dwMaximumWindowSize(maxHeight,maxWidth) = (%d,%d)' % (x.dwMaximumWindowSize.Y, x.dwMaximumWindowSize.X))
|
||||
|
||||
|
||||
@@ -22,8 +22,7 @@ class WinStyle(object):
|
||||
class WinTerm(object):
|
||||
|
||||
def __init__(self):
|
||||
self._default = \
|
||||
win32.GetConsoleScreenBufferInfo(win32.STDOUT).wAttributes
|
||||
self._default = win32.GetConsoleScreenBufferInfo(win32.STDOUT).wAttributes
|
||||
self.set_attrs(self._default)
|
||||
self._default_fore = self._fore
|
||||
self._default_back = self._back
|
||||
@@ -67,3 +66,37 @@ class WinTerm(object):
|
||||
handle = win32.STDERR
|
||||
win32.SetConsoleTextAttribute(handle, attrs)
|
||||
|
||||
def set_cursor_position(self, position=None, on_stderr=False):
|
||||
if position is None:
|
||||
#I'm not currently tracking the position, so there is no default.
|
||||
#position = self.get_position()
|
||||
return
|
||||
handle = win32.STDOUT
|
||||
if on_stderr:
|
||||
handle = win32.STDERR
|
||||
win32.SetConsoleCursorPosition(handle, position)
|
||||
|
||||
def erase_data(self, mode=0, on_stderr=False):
|
||||
# 0 (or None) should clear from the cursor to the end of the screen.
|
||||
# 1 should clear from the cursor to the beginning of the screen.
|
||||
# 2 should clear the entire screen. (And maybe move cursor to (1,1)?)
|
||||
#
|
||||
# At the moment, I only support mode 2. From looking at the API, it
|
||||
# should be possible to calculate a different number of bytes to clear,
|
||||
# and to do so relative to the cursor position.
|
||||
if mode[0] not in (2,):
|
||||
return
|
||||
handle = win32.STDOUT
|
||||
if on_stderr:
|
||||
handle = win32.STDERR
|
||||
# here's where we'll home the cursor
|
||||
coord_screen = win32.COORD(0,0)
|
||||
csbi = win32.GetConsoleScreenBufferInfo(handle)
|
||||
# get the number of character cells in the current buffer
|
||||
dw_con_size = csbi.dwSize.X * csbi.dwSize.Y
|
||||
# fill the entire screen with blanks
|
||||
win32.FillConsoleOutputCharacter(handle, ord(' '), dw_con_size, coord_screen)
|
||||
# now set the buffer's attributes accordingly
|
||||
win32.FillConsoleOutputAttribute(handle, self.get_attrs(), dw_con_size, coord_screen );
|
||||
# put the cursor at (0, 0)
|
||||
win32.SetConsoleCursorPosition(handle, (coord_screen.X, coord_screen.Y))
|
||||
|
||||
+1
-1
@@ -115,7 +115,7 @@ class AppDir(object):
|
||||
remove(fn)
|
||||
else:
|
||||
removedirs(fn)
|
||||
except OSError, why:
|
||||
except OSError as why:
|
||||
if why.errno == errno.ENOENT:
|
||||
pass
|
||||
else:
|
||||
|
||||
@@ -12,4 +12,4 @@ This module provides the text output helper system.
|
||||
from . import colored
|
||||
from . import progress
|
||||
|
||||
from core import *
|
||||
from .core import *
|
||||
|
||||
+28
-30
@@ -14,8 +14,9 @@ from __future__ import absolute_import
|
||||
import re
|
||||
import sys
|
||||
|
||||
PY3 = sys.version_info[0] >= 3
|
||||
|
||||
from ..packages import colorama
|
||||
from ..packages.colorama.conv import html
|
||||
|
||||
__all__ = (
|
||||
'red', 'green', 'yellow', 'blue',
|
||||
@@ -24,10 +25,17 @@ __all__ = (
|
||||
)
|
||||
|
||||
COLORS = __all__[:-2]
|
||||
DISABLE_COLOR = False
|
||||
|
||||
if not sys.stdout.isatty():
|
||||
if 'get_ipython' in dir():
|
||||
"""
|
||||
when ipython is fired lot of variables like _oh, etc are used.
|
||||
There are so many ways to find current python interpreter is ipython.
|
||||
get_ipython is easiest is most appealing for readers to understand.
|
||||
"""
|
||||
DISABLE_COLOR = True
|
||||
else:
|
||||
DISABLE_COLOR = False
|
||||
|
||||
|
||||
|
||||
class ColoredString(object):
|
||||
@@ -40,9 +48,8 @@ class ColoredString(object):
|
||||
@property
|
||||
def color_str(self):
|
||||
if sys.stdout.isatty() and not DISABLE_COLOR:
|
||||
colorama.init(autoreset=True)
|
||||
return '%s%s%s' % (
|
||||
colorama.Fore(self.color), self.s, colorama.Fore.RESET)
|
||||
getattr(colorama.Fore, self.color), self.s, colorama.Fore.RESET)
|
||||
else:
|
||||
return self.s
|
||||
|
||||
@@ -53,11 +60,20 @@ class ColoredString(object):
|
||||
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
|
||||
value = self.color_str
|
||||
if isinstance(value, str) and hasattr(value, 'decode'):
|
||||
return value.decode('utf8')
|
||||
return value
|
||||
|
||||
if PY3:
|
||||
__str__ = __unicode__
|
||||
else:
|
||||
def __str__(self):
|
||||
return unicode(self).encode('utf8')
|
||||
|
||||
def __iter__(self):
|
||||
return iter(self.color_str)
|
||||
|
||||
def __add__(self, other):
|
||||
return str(self.color_str) + str(other)
|
||||
@@ -68,8 +84,8 @@ class ColoredString(object):
|
||||
def __mul__(self, other):
|
||||
return (self.color_str * other)
|
||||
|
||||
def split(self, x=' '):
|
||||
return map(self._new, self.s.split(x))
|
||||
def split(self, sep=None):
|
||||
return [self._new(s) for s in self.s.split(sep)]
|
||||
|
||||
def _new(self, s):
|
||||
return ColoredString(self.color, s)
|
||||
@@ -79,7 +95,7 @@ def clean(s):
|
||||
strip = re.compile("([^-_a-zA-Z0-9!@#%&=,/'\";:~`\$\^\*\(\)\+\[\]\.\{\}\|\?\<\>\\]+|[^\s]+)")
|
||||
txt = strip.sub('', str(s))
|
||||
|
||||
strip = re.compile(r'\[[\d;]+m')
|
||||
strip = re.compile(r'\[\d+m')
|
||||
txt = strip.sub('', txt)
|
||||
|
||||
return txt
|
||||
@@ -109,24 +125,6 @@ def cyan(string):
|
||||
def white(string):
|
||||
return ColoredString('WHITE', string)
|
||||
|
||||
def rgb(color, string):
|
||||
"""
|
||||
RGB-colored text.
|
||||
|
||||
The `color` argument can have the following formats:
|
||||
- #3f3f3f
|
||||
- 3f3f3f
|
||||
- ('3f', '3f', '3f')
|
||||
- (100, 24, 244)
|
||||
"""
|
||||
if isinstance(color, basestring):
|
||||
color = html.to_rgb(color)
|
||||
else:
|
||||
if isinstance(color[0], basestring):
|
||||
color = (int(color[0],16), int(color[1],16), int(color[2],16))
|
||||
|
||||
return ColoredString(color, string)
|
||||
|
||||
def disable():
|
||||
"""Disables colors."""
|
||||
global DISABLE_COLOR
|
||||
|
||||
@@ -94,7 +94,6 @@ def columns(*cols, **kwargs):
|
||||
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):
|
||||
|
||||
@@ -15,20 +15,21 @@ import sys
|
||||
STREAM = sys.stderr
|
||||
|
||||
BAR_TEMPLATE = '%s[%s%s] %i/%i\r'
|
||||
BAR_EMPTY_CHAR = '-'
|
||||
BAR_FILLED_CHAR = '='
|
||||
MILL_TEMPLATE = '%s %s %i/%i\r'
|
||||
|
||||
DOTS_CHAR = '.'
|
||||
BAR_FILLED_CHAR = '#'
|
||||
BAR_EMPTY_CHAR = ' '
|
||||
MILL_CHARS = ['|', '/', '-', '\\']
|
||||
|
||||
|
||||
def bar(it, label='', width=32, hide=False):
|
||||
def bar(it, label='', width=32, hide=False, empty_char=BAR_EMPTY_CHAR, filled_char=BAR_FILLED_CHAR):
|
||||
"""Progress iterator. Wrap your iterables with it."""
|
||||
|
||||
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))
|
||||
label, filled_char*x, empty_char*(width-x), _i, count))
|
||||
STREAM.flush()
|
||||
|
||||
count = len(it)
|
||||
@@ -66,3 +67,32 @@ def dots(it, label='', hide=False):
|
||||
STREAM.write('\n')
|
||||
STREAM.flush()
|
||||
|
||||
|
||||
def mill(it, label='', hide=False,):
|
||||
"""Progress iterator. Prints a mill while iterating over the items."""
|
||||
|
||||
def _mill_char(_i):
|
||||
if _i == 100:
|
||||
return ' '
|
||||
else:
|
||||
return MILL_CHARS[_i % len(MILL_CHARS)]
|
||||
|
||||
def _show(_i):
|
||||
if not hide:
|
||||
STREAM.write(MILL_TEMPLATE % (
|
||||
label, _mill_char(_i), _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:
|
||||
STREAM.write('\n')
|
||||
STREAM.flush()
|
||||
|
||||
+29
-5
@@ -11,9 +11,33 @@ Various Python helpers used within clint.
|
||||
from __future__ import absolute_import
|
||||
from __future__ import with_statement
|
||||
|
||||
import sys
|
||||
import errno
|
||||
import os.path
|
||||
from os import makedirs
|
||||
from glob import glob
|
||||
|
||||
try:
|
||||
basestring
|
||||
except NameError:
|
||||
basestring = str
|
||||
|
||||
def expand_path(path):
|
||||
"""Expands directories and globs in given path."""
|
||||
|
||||
paths = []
|
||||
path = os.path.expanduser(path)
|
||||
path = os.path.expandvars(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))
|
||||
|
||||
return paths
|
||||
|
||||
|
||||
|
||||
def is_collection(obj):
|
||||
@@ -29,7 +53,7 @@ def mkdir_p(path):
|
||||
"""Emulates `mkdir -p` behavior."""
|
||||
try:
|
||||
makedirs(path)
|
||||
except OSError, exc: # Python >2.5
|
||||
except OSError as exc: # Python >2.5
|
||||
if exc.errno == errno.EEXIST:
|
||||
pass
|
||||
else:
|
||||
@@ -37,17 +61,17 @@ def mkdir_p(path):
|
||||
|
||||
def tsplit(string, delimiters):
|
||||
"""Behaves str.split but supports tuples of delimiters."""
|
||||
|
||||
|
||||
delimiters = tuple(delimiters)
|
||||
stack = [string,]
|
||||
|
||||
|
||||
for delimiter in delimiters:
|
||||
for i, substring in enumerate(stack):
|
||||
substack = substring.split(delimiter)
|
||||
stack.pop(i)
|
||||
for j, _substring in enumerate(substack):
|
||||
stack.insert(i+j, _substring)
|
||||
|
||||
|
||||
return stack
|
||||
|
||||
def schunk(string, size):
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
@@ -13,4 +15,4 @@ text = 'THIS TEXT IS COLORED %s!'
|
||||
if __name__ == '__main__':
|
||||
|
||||
for color in colored.COLORS:
|
||||
print getattr(colored, color)(text % color.upper())
|
||||
print(getattr(colored, color)(text % color.upper()))
|
||||
|
||||
@@ -1,40 +0,0 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
sys.path.insert(0, os.path.abspath('..'))
|
||||
|
||||
from clint.textui import colored
|
||||
from clint.packages.colorama.conv import html
|
||||
from clint.packages.colorama import caps
|
||||
from clint.textui import columns
|
||||
|
||||
# Force capability to XTERM256 for now, otherwise it's boring
|
||||
# I'm not sure a reliable detection mechanism even exists...
|
||||
caps.capability = caps.ColorCapability.XTERM256
|
||||
|
||||
def colorizer(colors):
|
||||
for color in colors:
|
||||
yield colored.rgb(color, color)
|
||||
|
||||
if __name__ == '__main__':
|
||||
i = iter(colorizer(html.colors))
|
||||
while True:
|
||||
try:
|
||||
a = i.next()
|
||||
except StopIteration:
|
||||
break
|
||||
try:
|
||||
b = i.next()
|
||||
except StopIteration:
|
||||
b = ""
|
||||
try:
|
||||
c = i.next()
|
||||
except StopIteration:
|
||||
c = ""
|
||||
print columns([a,30],[b,30],[c,30])
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,31 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
sys.path.insert(0, os.path.abspath('..'))
|
||||
|
||||
from clint.eng import join
|
||||
from clint.textui import colored, indent, puts
|
||||
|
||||
colors = [
|
||||
colored.blue('blue'),
|
||||
colored.red('red'),
|
||||
colored.yellow('yellow'),
|
||||
colored.green('green'),
|
||||
colored.magenta('magenta')
|
||||
]
|
||||
|
||||
colors = [str(cs) for cs in colors]
|
||||
|
||||
|
||||
puts('Smart:')
|
||||
with indent(4):
|
||||
for i in range(len(colors)):
|
||||
puts(join(colors[:i+1]))
|
||||
puts('\n')
|
||||
puts('Stupid:')
|
||||
with indent(4):
|
||||
for i in range(len(colors)):
|
||||
puts(join(colors[:i+1], im_a_moron=True, conj='\'n'))
|
||||
@@ -17,4 +17,6 @@ if __name__ == '__main__':
|
||||
|
||||
for i in progress.dots(range(100)):
|
||||
sleep(random() * 0.2)
|
||||
|
||||
|
||||
for i in progress.mill(range(100)):
|
||||
sleep(random() * 0.2)
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
@@ -13,16 +15,16 @@ resources.init('kennethreitz', 'clint')
|
||||
lorem = '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.'
|
||||
|
||||
|
||||
print '%s created.' % resources.user.path
|
||||
print('%s created.' % resources.user.path)
|
||||
|
||||
resources.user.write('lorem.txt', lorem)
|
||||
print 'lorem.txt created'
|
||||
print('lorem.txt created')
|
||||
|
||||
assert resources.user.read('lorem.txt') == lorem
|
||||
print 'lorem.txt has correct contents'
|
||||
print('lorem.txt has correct contents')
|
||||
|
||||
resources.user.delete('lorem.txt')
|
||||
print 'lorem.txt deleted'
|
||||
print('lorem.txt deleted')
|
||||
|
||||
assert resources.user.read('lorem.txt') == None
|
||||
print 'lorem.txt deletion confirmed'
|
||||
print('lorem.txt deletion confirmed')
|
||||
|
||||
@@ -19,5 +19,6 @@ if __name__ == '__main__':
|
||||
|
||||
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]))
|
||||
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]))
|
||||
|
||||
@@ -0,0 +1,4 @@
|
||||
{
|
||||
"title": "Bashō's 'old pond'",
|
||||
"text": "古池や蛙飛込む水の音"
|
||||
}
|
||||
@@ -0,0 +1,59 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
import sys
|
||||
import codecs
|
||||
|
||||
sys.path.insert(0, os.path.abspath('..'))
|
||||
|
||||
try:
|
||||
import json
|
||||
except:
|
||||
import simplejson as json
|
||||
|
||||
from clint import args
|
||||
from clint import piped_in
|
||||
from clint.textui import colored, puts, indent
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
puts('Test:')
|
||||
with indent(4):
|
||||
puts('%s Fake test 1.' % colored.green('✔'))
|
||||
puts('%s Fake test 2.' % colored.red('✖'))
|
||||
|
||||
puts('')
|
||||
puts('Greet:')
|
||||
with indent(4):
|
||||
puts(colored.red('Здравствуйте'))
|
||||
puts(colored.green('你好。'))
|
||||
puts(colored.yellow('سلام'))
|
||||
puts(colored.magenta('안녕하세요'))
|
||||
puts(colored.blue('नमस्ते'))
|
||||
puts(colored.cyan('γειά σου'))
|
||||
|
||||
puts('')
|
||||
puts('Arguments:')
|
||||
with indent(4):
|
||||
puts('%s' % colored.red(args[0]))
|
||||
|
||||
puts('')
|
||||
puts('File:')
|
||||
with indent(4):
|
||||
f = args.files[0]
|
||||
puts(colored.yellow('%s:' % f))
|
||||
with indent(2):
|
||||
fd = codecs.open(f, encoding='utf-8')
|
||||
for line in fd:
|
||||
line = line.strip('\n\r')
|
||||
puts(colored.yellow(' %s' % line))
|
||||
fd.close()
|
||||
|
||||
puts('')
|
||||
puts('Input:')
|
||||
with indent(4):
|
||||
in_data = json.loads(piped_in())
|
||||
title = in_data['title']
|
||||
text = in_data['text']
|
||||
puts(colored.blue('Title: %s' % title))
|
||||
puts(colored.magenta('Text: %s' % text))
|
||||
Executable
+3
@@ -0,0 +1,3 @@
|
||||
#!/usr/bin/env sh
|
||||
|
||||
python unicode.py こんにちは。 unicode.json < unicode.json
|
||||
@@ -29,6 +29,10 @@ setup(
|
||||
author='Kenneth Reitz',
|
||||
author_email='me@kennethreitz.com',
|
||||
url='https://github.com/kennethreitz/clint',
|
||||
data_files=[
|
||||
'README.rst',
|
||||
'HISTORY.rst',
|
||||
],
|
||||
packages= [
|
||||
'clint',
|
||||
'clint.textui',
|
||||
@@ -38,14 +42,18 @@ setup(
|
||||
license='ISC',
|
||||
classifiers=(
|
||||
# 'Development Status :: 5 - Production/Stable',
|
||||
'Environment :: Console',
|
||||
'Intended Audience :: Developers',
|
||||
'Natural Language :: English',
|
||||
'License :: OSI Approved :: ISC License (ISCL)',
|
||||
'Programming Language :: Python',
|
||||
'Programming Language :: Python :: 2',
|
||||
'Programming Language :: Python :: 2.5',
|
||||
'Programming Language :: Python :: 2.6',
|
||||
'Programming Language :: Python :: 2.7',
|
||||
# 'Programming Language :: Python :: 3.0',
|
||||
# 'Programming Language :: Python :: 3.1',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Programming Language :: Python :: 3.1',
|
||||
'Programming Language :: Python :: 3.2',
|
||||
'Topic :: Terminals :: Terminal Emulators/X Terminals',
|
||||
),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user