Talk to xkcd checkbox

xkcd_2445.py

# https://xkcd.com/2445

import requests

BASE_URL = 'https://xkcd.com/2445/morse/.../-.._-..._...--_...--_....-_.-_.-_-..._-....-_----._..---_.-_.----_-....-_.----_.----_._-..._-....-_---.._-----_-----_.----_-....-_---.._-.-._.----_-...._....-_....._....-_..-._-..._-----_..---_.-/'

MORSE_CODE_DICT = { 'A':'.-', 'B':'-...',
                    'C':'-.-.', 'D':'-..', 'E':'.',
                    'F':'..-.', 'G':'--.', 'H':'....',
                    'I':'..', 'J':'.---', 'K':'-.-',
                    'L':'.-..', 'M':'--', 'N':'-.',
                    'O':'---', 'P':'.--.', 'Q':'--.-',
                    'R':'.-.', 'S':'...', 'T':'-',
                    'U':'..-', 'V':'...-', 'W':'.--',
                    'X':'-..-', 'Y':'-.--', 'Z':'--..',
                    '1':'.----', '2':'..---', '3':'...--',
                    '4':'....-', '5':'.....', '6':'-....',
                    '7':'--...', '8':'---..', '9':'----.',
                    '0':'-----', ', ':'--..--', '.':'.-.-.-',
                    '?':'..--..', '/':'-..-.', '-':'-....-',
                    '(':'-.--.', ')':'-.--.-', ' ':' / '
                }

REVERSE_MORSE_CODE_DICT = {v: k for k, v in MORSE_CODE_DICT.items()}

def from_morsecode(code: str) -> str:
    text = ''
    for words in code.split(' / ')[1:]:
        word = ''.join(REVERSE_MORSE_CODE_DICT.get(i, '?') for i in words.split())
        text += '  ' + word
    return text

def to_morsecode(text: str) -> str:
    code = '_'.join(MORSE_CODE_DICT[c.upper()] for c in text)
    return code

def call(code: str) -> str:
    url = f'{BASE_URL}{code}'
    r = requests.get(url)
    r.raise_for_status()
    return r.text

def send_and_receive(msg: str, debug = False) -> str:
    if debug: print(f'===> Sending msg={msg}')
    code = to_morsecode(msg)
    if debug: print(f'===> Encoded msg={msg} to code={code}')
    response_code = call(code)
    if debug: print(f'===> Received response={response_code}')
    response = from_morsecode(response_code)
    if debug: print(f'===> Decoded code={response_code} to msg={response}')
    return response

def main():
    while True:
        msg = input('Enter message: ')
        response = send_and_receive(msg)
        print(f'Received: {response}')

def verify():
    s = '.. -... ...-- ...-- ....- .- .- -... -....- ----. ..--- .- .---- -....- .---- .---- . -... -....- ---.. ----- ----- .---- -....- ---.. -.-. .---- -.... ....- ..... ....- ..-. -... ----- ..--- .- / .... . .-.. .-.. --- -.-.-- / .- -. -.-- -... --- -.. -.-- / --- ..- - / - .... . .-. . ..--..'
    msg = from_morsecode(s)
    print(msg)

if __name__ == '__main__':
    main()
    # verify()