| 44 | | |
| 45 | | class Percent : |
| 46 | | """A class to display progress.""" |
| 47 | | def __init__(self, app, size=None) : |
| 48 | | """Initializes the engine.""" |
| 49 | | self.isatty = sys.stdout.isatty() |
| 50 | | self.app = app |
| 51 | | self.size = None |
| 52 | | if size : |
| 53 | | self.setSize(size) |
| 54 | | self.previous = None |
| 55 | | self.before = time.time() |
| 56 | | |
| 57 | | def setSize(self, size) : |
| 58 | | """Sets the total size.""" |
| 59 | | self.number = 0 |
| 60 | | self.size = size |
| 61 | | if size : |
| 62 | | self.factor = 100.0 / float(size) |
| 63 | | |
| 64 | | def display(self, msg) : |
| 65 | | """Displays the value.""" |
| 66 | | if self.isatty : |
| 67 | | self.app.display(msg) |
| 68 | | sys.stdout.flush() |
| 69 | | |
| 70 | | def oneMore(self) : |
| 71 | | """Increments internal counter.""" |
| 72 | | if self.size : |
| 73 | | self.number += 1 |
| 74 | | percent = "%.02f" % (float(self.number) * self.factor) |
| 75 | | if percent != self.previous : # optimize for large number of items |
| 76 | | self.display("\r%s%%" % percent) |
| 77 | | self.previous = percent |
| 78 | | |
| 79 | | def done(self) : |
| 80 | | """Displays the 'done' message.""" |
| 81 | | after = time.time() |
| 82 | | if self.size : |
| 83 | | try : |
| 84 | | speed = self.size / ((after - self.before) + 0.00000000001) # adds an epsilon to avoid an user's problem I can't reproduce... |
| 85 | | except ZeroDivisionError : |
| 86 | | speed = 1 # Fake value in case of division by zero, shouldn't happen anyway with the epsilon above... |
| 87 | | self.display("\r100.00%%\r \r%s. %s : %.2f %s.\n" \ |
| 88 | | % (_("Done"), _("Average speed"), speed, _("entries per second"))) |
| 89 | | else : |
| 90 | | self.display("\r100.00%%\r \r%s.\n" % _("Done")) |
