Base64

from string import ascii_uppercase, ascii_lowercase, digits

base64_table = ascii_uppercase + ascii_lowercase + digits + '+/'


def encode(words: str) -> str:  # base64 encode
    strings = ''.join([bin(ord(_)).replace('0b', '').rjust(8, '0') for _ in words])
    strings += '0' * ((len(strings) // 6 + 1) * 6 - len(strings)) if len(strings) % 6 != 0 else ''
    return ''.join([base64_table[int(strings[6 * i: 6 * (i + 1)], 2)] for i in range(len(strings) // 6)])


def decode(base64: str) -> str:   # base64 decode
    strings = ''.join([bin(base64_table.index(_)).replace('0b', '').rjust(6, '0') for _ in base64])
    decoded = ''.join([chr(int(strings[8 * _: 8 * (_ + 1)], 2)) for _ in range(len(strings) // 8)])
    return decoded

def base_steg(base64: list) -> str:  # base64 steg
    steg_string, steg_msg = '', ''
    for i in base64:
        i = i.replace('=', '')
        strings = ''.join([bin(base64_table.index(_)).replace('0b', '').rjust(6, '0') for _ in i])
        steg_string += strings[(len(strings) - len(strings) // 8 * 8) * -1:] if len(strings) % 8 != 0 else ''
    for i in range(len(steg_string) // 8):
        cha = chr(int(steg_string[i * 8: (i + 1) * 8], 2))
        steg_msg += cha if cha in base64_table else ''
    return steg_msg