spincursor.py

来自「Harvestman-最新版本」· Python 代码 · 共 90 行

# -- coding: utf-8
"""
spincursor.py - Module which provides a spin-cursor class.
The spincursor class can be used to indicate progress of
an action on the console.

Author: Anand B Pillai <abpillai at gmail dot com>

Created: Anand B Pillai 31 Oct 2007
    
Copyright (C) 2007, Anand B Pillai.
"""
import threading
import sys, os
import time
import unicodedata

class SpinCursor(threading.Thread):
    """ A console spin cursor class """
    
    def __init__(self, msg='',maxspin=0,minspin=10,speed=5):
        # Count of a spin
        self.count = 0
        self.out = sys.stdout
        self.flag = False
        self.max = maxspin
        self.min = minspin
        # Any message to print first ?
        self.msg = msg
        # Complete printed string
        self.string = ''
        # Speed is given as number of spins a second
        # Use it to calculate spin wait time
        self.waittime = 1.0/float(speed*4)
        # Don't do this for cygwin also!
        if os.name == 'posix' and os.environ.get('TERM','') != 'cygwin':
            self.spinchars = (unicodedata.lookup('FIGURE DASH'),u'\\ ',u'| ',u'/ ')
        else:
            # The unicode dash character does not show
            # up properly in Windows console.
            self.spinchars = (u'-',u'\\ ',u'| ',u'/ ')        
        threading.Thread.__init__(self, None, None, "Spin Thread")
        
    def spin(self):
        """ Perform a single spin """

        for x in self.spinchars:
            self.string = self.msg + "...\t" + x + "\r"
            self.out.write(self.string.encode('utf-8'))
            self.out.flush()
            time.sleep(self.waittime)

    def run(self):

        while (not self.flag) and ((self.count<self.min) or (self.count<self.max)):
            self.spin()
            self.count += 1

        # Clean up display...
        self.out.write(" "*(len(self.string) + 1) + "\n")
        
    def stop(self):
        self.flag = True

class InfiniteSpinCursor(SpinCursor):
    """ A spin cursor that keeps spinning till told to stop"""
    
    def __init__(self, msg=''):
        super(InfiniteSpinCursor, self).__init__(msg)

    def run(self):

        while (not self.flag):
            try:
                self.spin()
                self.count += 1
            except KeyboardInterrupt:
                break

        # Clean up display...
        self.out.write(" "*(len(self.string) + 1) + "\n")        
        
if __name__ == "__main__":
    spin = SpinCursor(msg="Spinning...",minspin=20,maxspin=50,speed=5)
    spin.start()
    spin.join()