feat: add loading spinner

Signed-off-by: Dominik Willner <th33xitus@gmail.com>
This commit is contained in:
dw-0
2024-06-30 16:01:11 +02:00
parent 956666605c
commit 7c9dcea359

33
kiauh/utils/spinner.py Normal file
View File

@@ -0,0 +1,33 @@
import sys
import threading
import time
class Spinner:
def __init__(self, message="Loading", delay=0.2):
self.message = f"{message} ..."
self.delay = delay
self._stop_event = threading.Event()
self._thread = threading.Thread(target=self._animate)
def _animate(self):
animation = ["", "", "", ""]
while not self._stop_event.is_set():
for char in animation:
sys.stdout.write(f"\r{char} {self.message}")
sys.stdout.flush()
time.sleep(self.delay)
if self._stop_event.is_set():
break
sys.stdout.write("\r" + " " * (len(self.message) + 1) + "\r")
sys.stdout.flush()
def start(self):
self._stop_event.clear()
if not self._thread.is_alive():
self._thread = threading.Thread(target=self._animate)
self._thread.start()
def stop(self):
self._stop_event.set()
self._thread.join()