base64 인코딩, 디코딩 툴 ( Webhacking.kr 6번 문제를 위한 )

파이썬으로 만들어져있고, Webhacking.kr 6번문제를 쭉 수동으로 돌리기가 뭐하여 만들어봄 ..

import base64
 
def main():
    mode = int(input("Select Mode ( 1: encode, 2. decode ) :"))
    if mode == 1:
        encode_mod()
    elif mode == 2:
        decode_mod()
    else:
        pass
 
def encode_mod():
    encode_text = input("encodeing String : ")
    encode_text = str(encode_text)
    count = input("count : ")
    count = int(count)
    replace_key = [ ['1', '!'],\
                    ['2', '@'],\
                    ['3', '$'],\
                    ['4', '^'],\
                    ['5', '&'],\
                    ['6', '*'],\
                    ['7', '('],\
                    ['8', ')']]
    
    for i in range(0,8):
        encode_text = encode_text.replace(replace_key[i][0], replace_key[i][1])
 
 
    encode_text = encode_text.encode()
    for i in range(0,count):
        encode_text = base64.b64encode(encode_text)
    print(str(encode_text))
 
 
def decode_mod():
    decode_text = input("Decodeing String : ")
    decode_text = str(decode_text)
    count = input("count : ")
    count = int(count)
    replace_key = [ ['1', '!'],\
                    ['2', '@'],\
                    ['3', '$'],\
                    ['4', '^'],\
                    ['5', '&'],\
                    ['6', '*'],\
                    ['7', '('],\
                    ['8', ')']]
    
    for i in range(0,8):
        decode_text = decode_text.replace(replace_key[i][1], replace_key[i][0])
 
    decode_text = decode_text.encode()
    for i in range(0,count):
        decode_text = base64.b64decode(decode_text)
    print(str(decode_text))
 
if __name__ == '__main__':
    main()