mirror of
https://github.com/dw-0/kiauh.git
synced 2025-12-25 16:53:36 +05:00
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import sys
|
|
import threading
|
|
import time
|
|
from typing import List, Literal
|
|
|
|
from core.constants import (
|
|
COLOR_GREEN,
|
|
COLOR_RED,
|
|
COLOR_WHITE,
|
|
COLOR_YELLOW,
|
|
RESET_FORMAT,
|
|
)
|
|
|
|
SpinnerColor = Literal["white", "red", "green", "yellow"]
|
|
|
|
|
|
class Spinner:
|
|
def __init__(
|
|
self,
|
|
message: str = "Loading",
|
|
color: SpinnerColor = "white",
|
|
interval: float = 0.2,
|
|
) -> None:
|
|
self.message = f"{message} ..."
|
|
self.interval = interval
|
|
self._stop_event = threading.Event()
|
|
self._thread = threading.Thread(target=self._animate)
|
|
self._color = ""
|
|
self._set_color(color)
|
|
|
|
def _animate(self) -> None:
|
|
animation: List[str] = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
|
|
while not self._stop_event.is_set():
|
|
for char in animation:
|
|
sys.stdout.write(f"\r{self._color}{char}{RESET_FORMAT} {self.message}")
|
|
sys.stdout.flush()
|
|
time.sleep(self.interval)
|
|
if self._stop_event.is_set():
|
|
break
|
|
sys.stdout.write("\r" + " " * (len(self.message) + 1) + "\r")
|
|
sys.stdout.flush()
|
|
|
|
def _set_color(self, color: SpinnerColor) -> None:
|
|
if color == "white":
|
|
self._color = COLOR_WHITE
|
|
elif color == "red":
|
|
self._color = COLOR_RED
|
|
elif color == "green":
|
|
self._color = COLOR_GREEN
|
|
elif color == "yellow":
|
|
self._color = COLOR_YELLOW
|
|
|
|
def start(self) -> None:
|
|
self._stop_event.clear()
|
|
if not self._thread.is_alive():
|
|
self._thread = threading.Thread(target=self._animate)
|
|
self._thread.start()
|
|
|
|
def stop(self) -> None:
|
|
self._stop_event.set()
|
|
self._thread.join()
|