summaryrefslogtreecommitdiff
path: root/lib/python2.7/site-packages/pip/_vendor/progress/helpers.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/python2.7/site-packages/pip/_vendor/progress/helpers.py')
-rw-r--r--lib/python2.7/site-packages/pip/_vendor/progress/helpers.py91
1 files changed, 91 insertions, 0 deletions
diff --git a/lib/python2.7/site-packages/pip/_vendor/progress/helpers.py b/lib/python2.7/site-packages/pip/_vendor/progress/helpers.py
new file mode 100644
index 0000000..9ed90b2
--- /dev/null
+++ b/lib/python2.7/site-packages/pip/_vendor/progress/helpers.py
@@ -0,0 +1,91 @@
+# Copyright (c) 2012 Giorgos Verigakis <verigak@gmail.com>
+#
+# Permission to use, copy, modify, and distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+
+from __future__ import print_function
+
+
+HIDE_CURSOR = '\x1b[?25l'
+SHOW_CURSOR = '\x1b[?25h'
+
+
+class WriteMixin(object):
+ hide_cursor = False
+
+ def __init__(self, message=None, **kwargs):
+ super(WriteMixin, self).__init__(**kwargs)
+ self._width = 0
+ if message:
+ self.message = message
+
+ if self.file.isatty():
+ if self.hide_cursor:
+ print(HIDE_CURSOR, end='', file=self.file)
+ print(self.message, end='', file=self.file)
+ self.file.flush()
+
+ def write(self, s):
+ if self.file.isatty():
+ b = '\b' * self._width
+ c = s.ljust(self._width)
+ print(b + c, end='', file=self.file)
+ self._width = max(self._width, len(s))
+ self.file.flush()
+
+ def finish(self):
+ if self.file.isatty() and self.hide_cursor:
+ print(SHOW_CURSOR, end='', file=self.file)
+
+
+class WritelnMixin(object):
+ hide_cursor = False
+
+ def __init__(self, message=None, **kwargs):
+ super(WritelnMixin, self).__init__(**kwargs)
+ if message:
+ self.message = message
+
+ if self.file.isatty() and self.hide_cursor:
+ print(HIDE_CURSOR, end='', file=self.file)
+
+ def clearln(self):
+ if self.file.isatty():
+ print('\r\x1b[K', end='', file=self.file)
+
+ def writeln(self, line):
+ if self.file.isatty():
+ self.clearln()
+ print(line, end='', file=self.file)
+ self.file.flush()
+
+ def finish(self):
+ if self.file.isatty():
+ print(file=self.file)
+ if self.hide_cursor:
+ print(SHOW_CURSOR, end='', file=self.file)
+
+
+from signal import signal, SIGINT
+from sys import exit
+
+
+class SigIntMixin(object):
+ """Registers a signal handler that calls finish on SIGINT"""
+
+ def __init__(self, *args, **kwargs):
+ super(SigIntMixin, self).__init__(*args, **kwargs)
+ signal(SIGINT, self._sigint_handler)
+
+ def _sigint_handler(self, signum, frame):
+ self.finish()
+ exit(0)