Mirroring/progress
1
0
Fork 0
mirror of https://github.com/verigak/progress.git synced 2025-12-08 19:33:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #69 from frasern/atexit

Browse Source 
Fixed #64 -- ensure hidden cursor is reshown at exit.
This commit is contained in:
Georgios Verigakis
2019-07-02 15:12:13 +03:00
committed by GitHub
parent 80a91b9c78 f6390b76f2
commit 1ed414290f
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
progress/__init__.py
View File

@@ -14,6 +14,7 @@
from __future__ import division, print_function from __future__ import division, print_function
import atexit
from collections import deque from collections import deque
from datetime import timedelta from datetime import timedelta
from math import ceil from math import ceil
@@ -52,6 +53,7 @@ class Infinite(object):
if self.file and self.is_tty(): if self.file and self.is_tty():
if self.hide_cursor: if self.hide_cursor:
print(HIDE_CURSOR, end='', file=self.file) print(HIDE_CURSOR, end='', file=self.file)
atexit.register(self.finish)
print(self.message, end='', file=self.file) print(self.message, end='', file=self.file)
self.file.flush() self.file.flush()
@@ -107,6 +109,7 @@ class Infinite(object):
print(file=self.file) print(file=self.file)
if self.hide_cursor: if self.hide_cursor:
print(SHOW_CURSOR, end='', file=self.file) print(SHOW_CURSOR, end='', file=self.file)
atexit.unregister(self.finish)
def is_tty(self): def is_tty(self):
try: try: