[BaekJoon] - 10935 base64 인코딩

 

10935번: BASE64 인코딩

 

#!/bin/python3
# base64 인코딩

import string

BIT = 6

def padding(binary):
    return binary + '0' * (BIT - len(binary) % 6)

if __name__ == "__main__":
    b64_table = dict()
    count = 0
    
    for char in string.ascii_uppercase + string.ascii_lowercase + string.digits + "+/":
        b64_table[str(count)] = char
        count += 1
    
    plaintext = input("")
    binary = ''
    for char in plaintext:
        binary += bin(ord(char))[2:].zfill(8) 
    
    if len(binary) % BIT != 0:
        binary = padding(binary)
        
    count = int(len(binary) / BIT)
    encode = ''
    for i in range(0, count):
        bit6 = binary[i * BIT : (i+1) * BIT]
        encode += b64_table[str((int(bit6,2)))]
    
    if len(plaintext) % 3 != 0:
        encode += "=" * (3 - len(plaintext) % 3)
        
    print(encode)