loading_bars.py

# Jacobus Burger (2022)
# Info:
#   Varous random CLI loading animations done to get an idea of how they make
#   stuff update in place on the command line.
from time import sleep
from os import get_terminal_size as term_size
from string import ascii_letters
from itertools import cycle
from random import choices, rand_int, choice


def percent(speed):
    for i in range(101):
        print(f"{i}%", end='')
        print('\r', end='')
        sleep(1 / speed)
    print()


def loading(speed):
    for i in range(101):
        print("Loading{}".format('.' * (i % 4)), end='\r', flush=True)
        sleep(1 / speed)
    print()


def spin(speed):
    phase = ["\\", "|", "/", "-"]
    for i in range(101):
        print(phase[i % 4], end='\r')
        sleep(1 / speed)
    print()


def orb(speed):
    phase = ['.', 'o', 'O']
    for i in range(101):
        print(phase[i % 3], end='\r')
        sleep(1 / speed)
    print()


def sweep(speed):
    phase = ["*..", ".*.", "..*", "..."]
    for i in range(0, 101, 3):
        print(phase[i % 4], end='\r')
        sleep(1 / speed)
    print()


def bar(speed):
    for i in range(0, 101):
        if i < term_size()[0]:
            print(('=' * i) + '>', end='\r')
        sleep(1 / speed)
    print()


def bounce(speed):
    phase = [".", "o"]
    for i in range(0, 101):
        print(phase[i % 2], end='\r')
        sleep(1 / speed)
    print()


def glitch(speed):
    for i in range(0, 101):
        print(''.join(choices(ascii_letters, k=term_size()[0]//4)), end='\r')
        sleep(1 / speed)
    print()


def scroll(speed):
    for i in range(0, 101):
        print(('_' * (i % 3)) + ' '.join('='*(term_size()[0]//4)), end='\r')
        sleep(1 / speed)
    print()


def rotate(speed):
    message = "Loading...          "
    for i in range(0, 101):
        print(message, end='\r') 
        sleep(1 / speed)
        # rotate text by 1
        message = message[-1] + message[:-1]
    print()


def fira(speed):
    """note: this requires fira code font to work"""
    message = '\uee00' + ('\uee01' * 8) + '\uee02' + ' '
    message = list(message)
    offset = 0
    for i in range(0, 101):
        if i % 10 == 0:
            message[offset] = chr(ord(message[offset]) + 3)
            offset += 1
        print(''.join(message), end='\r')
        sleep(1 / speed)
    print()


def alphabet(speed):
    letters = cycle(ascii_letters)
    for i in range(0, 101):
        print(next(letters), end='\r')
        sleep(1 / speed)
    print()


# a cheap tqdm imitation 😃
def tqdm(speed):
    for i in range(0, 101):
        print("{:<3}% |{:<10}|".format(i, ('█'*(i//10) + ' '*(10 - (i//10)))), end='\r')
        sleep(1 / speed)
    print()


# show a bunch of animals randomly
def animals(speed):
    animals = ['🦄', '🐏', '🦩', '🐧', '🐍', '🐢', '🐙', '🦋']
    for i in range(0, 101):
        print(''.join(choices(animals, k=term_size()[0]//4)), end='\r')
        sleep(1 / speed)
    print()


# waddle animal in a parade across the terminal
def parade(speed, animal='🐧'):
    for i in range(0, 101):
        print(' ' * (term_size()[0] - (i + 1)) + animal * (i // 2), end='\r')
        sleep(1 / speed)
    print()


# float a banana across the screen
def banana(speed):
    for i in range(0, 101):
        print(' ' * (term_size()[0] - (i + 1)) + animal, end='\r')
        sleep(1 / speed)
    print()


# randomness across the screen
def noise(speed):
    arr = ' ' * term_size()[0]
    for i in range(0, 101):
        arr[rand_int(0, len(arr) - 1)] = choice(['@', '.', '*'])
        sleep(1 / speed)
    print()


if __name__ == '__main__':
    while True:
        choice = int(input(
                """Choose animation:
                    (1)  percent
                    (2)  loading
                    (3)  spin
                    (4)  orb
                    (5)  sweep
                    (6)  progress bar
                    (7)  bounce
                    (8)  glitch
                    (9)  scroll
                    (10) rotate
                    (11) fira code progress bar (need font)
                    (12) alphabet soup
                    (13) not tqdm
                    (14) animals
                    (15) parade
                    (16) banana
                """
        ))
        speed = int(input("What speed? "))
        if choice == 1:
            percent(speed)
        if choice == 2:
            loading(speed)
        if choice == 3:
            spin(speed)
        if choice == 4:
            orb(speed)
        if choice == 5:
            sweep(speed)
        if choice == 6:
            bar(speed)
        if choice == 7:
            bounce(speed)
        if choice == 8:
            glitch(speed)
        if choice == 9:
            scroll(speed)
        if choice == 10:
            rotate(speed)
        if choice == 11:
            fira(speed)
        if choice == 12:
            alphabet(speed)
        if choice == 13:
            tqdm(speed)
        if choice == 14:
            animals(speed)
        if choice == 15:
            parade(speed)
        if choice == 16:
            banana(speed)