from termcolor import colored
import sys, time
def demo_basic_colors():
    print(colored("This is red text", "red"))
    print(colored("This is green text", "green"))
    print(colored("This is yellow text", "yellow"))
    print(colored("This is blue text", "blue"))
    print(colored("This is magenta text", "magenta"))
    print(colored("This is cyan text", "cyan"))
 
def demo_text_attributes():
    print(colored("Bold text", "white", attrs=["bold"]))
    print(colored("Underlined text", "cyan", attrs=["underline"]))
    print(colored("Reversed text", "yellow", attrs=["reverse"]))
    print(colored("Dark text", "magenta", attrs=["dark"]))
 
def demo_background_colors():
    print(colored("White on red", "white", "on_red"))
    print(colored("Blue on yellow", "blue", "on_yellow"))
    print(colored("Green on magenta", "green", "on_magenta"))
 
def status_msg(msg, status):
    color = {
        "success": "green",
        "error": "red",
        "warning": "yellow",
        "info": "cyan"
    }.get(status, "white")
    return colored(msg, color, attrs=["bold"])
 
def demo_status_messages():
    print(status_msg("✓ Operation completed successfully!", "success"))
    print(status_msg("⚠ Warning: Disk almost full!", "warning"))
    print(status_msg("✖ Error: Failed to connect!", "error"))
    print(status_msg("ℹ Info: Update available.", "info"))
 
def demo_loop_with_colors():
    for i in range(1, 6):
        print(colored(f"Step {i}/5", "cyan"), colored("✔", "green", attrs=["bold"]))
 
def color_progress_bar(progress, total, bar_length=30):
    percent = float(progress) / total
    filled_len = int(round(bar_length * percent))
    
    bar = colored("█" * filled_len, "green") + colored("-" * (bar_length - filled_len), "white")
    percent_text = colored(f"{int(percent * 100)}%", "cyan", attrs=["bold"])
    
    sys.stdout.write(f"\r[{bar}] {percent_text}")
    sys.stdout.flush()
 
def demo_progress():
    total = 50
    for i in range(total + 1):
        color_progress_bar(i, total)
        time.sleep(0.05)
    print()  # 换行
if __name__ == "__main__":
    print(colored("=== Basic Colors ===", "white", attrs=["bold", "underline"]))
    demo_basic_colors()
    print()
 
    print(colored("=== Text Attributes ===", "white", attrs=["bold", "underline"]))
    demo_text_attributes()
    print()
 
    print(colored("=== Background Colors ===", "white", attrs=["bold", "underline"]))
    demo_background_colors()
    print()
 
    print(colored("=== Status Messages ===", "white", attrs=["bold", "underline"]))
    demo_status_messages()
    print()
 
    print(colored("=== Loop with Colored Steps ===", "white", attrs=["bold", "underline"]))
    demo_loop_with_colors()
 
    print(colored("=== Colored Progress Bar Demo ===", "white", attrs=["bold", "underline"]))
    demo_progress()