modules.py

import datetime
import sys
from time import sleep
from subprocess import call

# reset terminal using subprocess.call()
def term_reset(i=1):  # requires fixing some older tutorials due to added spinit
    """reset terminal after given n seconds,
    uses spinit(j) spinning effect, expects int"""
    # two chars about 1s, default
    j = i * 2
    print(f'reset term in {i}s ...',end=' ',sep='')
    # spinner newline
    print(ansicolor.fg.lightcyan)
    spinit(int(j))
    print(ansicolor.reset)
    _ = call('reset')

# default w/ msg
def trst(i=1,msg=''):
    """reset terminal after given n seconds w/ optional msg,
    uses spinit(j) spinning effect, expects int"""
    # two chars about 1s, default
    j = i * 2
    print(f'\n{msg}\n>>> reset term in {i}s ...  ',end='\b',sep='')
    spinit(j)
    print(ansicolor.reset)
    _ = call('reset')

    # spinner end of line
    #print(f'\n{msg}\n>>> reset term in {next(counter)} {n}s ...  ',end='\b',sep='')
    #spinit(j)
    #print(ansicolor.reset)
    #_ = call('reset')

    # counter does not work
    # def count():
    #     for counter in range(1, i):
    #         yield counter
    # # counter + spinner
    # counter = count()
    # sys.stdout.write(next(counter))
    # sys.stdout.flush()
    # sleep(0.5)
    # sys.stdout.write('\b')

# reset only
def treset():
    """reset terminal immediately"""
    _ = call('reset')

# progress spinner using generator w/ some help from [https://stackoverflow.com/a/4995896]
def spinit(n=9):
    # create generator, note keyword 'yield' instead of 'return'
    def spin():
        while True:
            for cursor in '|/-\\':
                yield cursor
    spinner = spin()
    # initial value = 2 spins
    for i in range(1,n):
        sys.stdout.write(next(spinner))
        sys.stdout.flush()
        sleep(0.5)
        sys.stdout.write('\b')

# strings to hex representation
def strhx(i):
    """returns hex value for given input - expects str()"""
    ie = i.encode("utf-8").hex()
    return ie
# print(strhx.__doc__)
# print(strhx('phk'))

def dtformat():
    """format options for datetime objects"""
    print('''\ncustom datetime formatting:

    >>> time
    >>> %Z - timezone name
    >>> %z - UTC offset +HHMM or -HHMM
    >>> %p - locales equivalent am or pm
    >>> %I - 12h hour as zero padded decimal number
    >>> %H - 24h hour as zero padded decimal number
    >>> %M - minute as zero padded decimal number
    >>> %S - second as zero padded decimal number
    >>> %f - microsecond as zero padded decimal number

    >>> day, weekday
    >>> %d - day of month as zero padded decimal number
    >>> %j - day of the year as as zero padded decimal number
    >>> %w - weekday as decimal number 0=Sunday 6=Saturday
    >>> %a - weekday abbreviated name
    >>> %A - weekday full name

    >>> year
    >>> %Y - year w/ century as decimal number
    >>> %y - year w/o century as zero padded decimal number

    >>> month
    >>> %B - months full name
    >>> %b - months abbreviated name
    >>> %m - month as zero padded decimal number\n''')
# print(dtformat())

# now = datetime.datetime.today()
def now():
    return datetime.datetime.today()

# nowf = datetime.datetime.strftime(format, now)
def nowf():
    format = "%Y%m%dT%H%M%SZ"
    return now().strftime(format)

# custom color output
class color:
    PURPLE = '\033[95m'
    CYAN = '\033[96m'
    DARKCYAN = '\033[36m'
    BLUE = '\033[94m'
    GREEN = '\033[92m'
    YELLOW = '\033[93m'
    RED = '\033[91m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'
    END = '\033[0m'
    # 256color foreground [38;5;$idm]
    DARKGRAY = '\033[38;5;240m'
    DARKYELLOW = '\033[38;5;136m'
    # non color ansi codes
    CLS = '\033[2J'
# print(color.DARKYELLOW, '\nfinished:', color.BOLD, color.GREEN, now(), color.CYAN, nowf(), color.END, '\n')

class ansicolor:
    reset = '\033[0m'
    bold = '\033[01m'
    disable = '\033[02m'
    underline = '\033[04m'
    reverse = '\033[07m'
    strikethrough = '\033[09m'
    invisible = '\033[08m'
    cls = '\033[2J'

    class fg:
        black = '\033[30m'
        red = '\033[31m'
        green = '\033[32m'
        orange = '\033[33m'
        blue = '\033[34m'
        purple = '\033[35m'
        cyan = '\033[36m'
        lightgrey = '\033[37m'
        darkgrey = '\033[90m'
        lightred = '\033[91m'
        lightgreen = '\033[92m'
        yellow = '\033[93m'
        lightblue = '\033[94m'
        pink = '\033[95m'
        lightcyan = '\033[96m'
        darkcyan = '\033[36m'
        darkgray = '\033[38;5;240m'
        darkyellow = '\033[38;5;136m'

    class bg:
        black = '\033[40m'
        red = '\033[41m'
        green = '\033[42m'
        orange = '\033[43m'
        blue = '\033[44m'
        purple = '\033[45m'
        cyan = '\033[46m'
        lightgrey = '\033[47m'

class Calc:
    # def __init__(self):
    def add(n1,n2):
        return n1 + n2

    def sub(n1,n2):
        return n1 - n2

    def mul(n1,n2):
        return n1 * n2

    def div(n1,n2):
        return n1 / n2

    def idiv(n1,n2):
        return n1 // n2

    def rdiv(n1,n2):
        return n1 % n2

    def exp(n1,n2):
        return n1 ** n2