Python使用一个进度条

淩亂°似流年 2022-09-30 04:58 213阅读 0赞

由pyskydrive工程([http://code.google.com/p/pyskydrive/][http_code.google.com_p_pyskydrive])中发现的进度条，加了一点修改！

import os import sys import cmd import unicodedata from threading import Thread from time import sleep class ProgressBar(Thread): """A simple text progress bar control.""" def \_\_init\_\_(self): self.running = True self.percent = -1 self.msg = "" Thread.\_\_init\_\_(self) def run(self): i = 1 while self.running and self.percent < 100: display = '/r' if self.msg != "": display += self.msg + ' ' if self.percent >= 0: display += str(self.percent) + '% ' display += '-' \* (40 \* self.percent / 100) if i + 1 + len(display) > 80: i = 1 display += '-' \* i display += '>' display += ' ' \* (80 - len(display)) sys.stderr.write(display) i += 1 sleep(0.5) sys.stderr.write('/r' + ' ' \* 79 + '/r') sys.stderr.flush() def stop(self): """Stop displaying progress bar. Note: there may be latency to stop. You'd better wait for the thread stops. See \_stop\_progress(t\_bar). """ self.running = False def set\_percent(self, percent, msg = ""): """Call back method for owner of progress bar.""" self.percent = int(percent) self.msg = msg def is\_alive(self): return self.isAlive() def \_start\_progress(): """Display a progress bar""" t\_bar = ProgressBar() t\_bar.start() return t\_bar def \_stop\_progress(t\_bar): """Hide the progress bar""" if hasattr(t\_bar, "is\_alive"): alive = t\_bar.is\_alive() else: alive = t\_bar.isAlive() if t\_bar and alive: t\_bar.stop() t\_bar.join() def \_set\_progress(t\_bar,percent,msg = ""): t\_bar.set\_percent(percent) def \_print\_msg(t\_bar, \*args): """Hide the progress bar and print message""" assert t\_bar is None or isinstance(t\_bar, ProgressBar) if t\_bar and t\_bar.is\_alive(): \_stop\_progress(t\_bar) print ' '.join(args) def \_unicode\_str\_width(text): """Calculate the exact width of unicode string.""" width = 0 for char in text: if isinstance(char, unicode): if unicodedata.east\_asian\_width(char) in ('F', 'W', 'A'): width += 2 else: width += 1 else: width += 1 return width \#Test the progress bar if \_\_name\_\_ == "\_\_main\_\_": t\_bar = \_start\_progress() count = 0; while count<100: \_set\_progress(t\_bar,count,'%d'%(count)) sleep(0.1) count = count+1 \_stop\_progress(t\_bar) print "Done!"

\*其实几个控制函数还可以封装到一个类中使用，那就更为方便了!

[http_code.google.com_p_pyskydrive]: http://code.google.com/p/pyskydrive/