Wordles with Dads

def checkguess(answer, guess_in):
    if len(guess_in) != len(answer):
        print("Not the right guess length")
        return False
    if not alphacheck(guess_in):
        print("Invalid characters A-Z only")
        return False

    # histomaker returns the histogram of the alphabet positions
    # i.e. histomaker('ABA') will return
    #    {'A': [0, 2], 'B': [1], 'C': [], 'D': [], ..., 'Z': []}
    truth = histomaker(answer)
    guess = histomaker(guess_in)
    correct = []
    position = []
    for ltr in alphabet:
        tmp = guess[ltr]
        truetmp = truth[ltr]
        counter = 0
        for i in tmp:
            if i in truetmp:
                counter += 1
                correct.append(i)
        for i in tmp:
            if not i in correct:
                if len(truetmp) > counter:
                    counter += 1
                    position.append(i)
    correct.sort()
    position.sort()
    return {"correct": correct, "position": position}

import requests
search_url = "https://icanhazdadjoke.com/search"

# creates a jokes.txt file and insert all jokes scraped
# ... all the jokes are formatted the same way as `wordleswithdads.py`
def scrape():
    for page in range(22): # hard-coding 22 because there are 22 pages of jokes
        response = requests.get(search_url,
                                headers={"Accept": "application/json"},
                                params={"limit": 30, "page": page + 1}) # API only allows max 30 jokes per query
        joke_request = response.json()
        for joke in joke_request["results"]:
            # write to file
            with open("jokes.txt", "a") as joke_file:
                joke_file.write(joke["joke"] + "\n")
scrape()

# opens our existing jokes.txt and sanitize the jokes
# store the resulting jokes (sorted by length) into jokes_format.txt
def format():
    output = []
    with open("jokes.txt", "r") as joke_file:
        joke_list = joke_file.readlines()
        for joke in joke_list:
            # remove space
            s = joke.replace(" ", "")
            # capitalize
            s = s.upper()
            # remove non-alphabetic characters
            s = ''.join([i for i in s if i.isalpha()])
            output.append(s)

    # sort array by length
    output.sort(key=len, reverse=True)

    # write output to file
    with open("jokes_format.txt", "w") as joke_file:
        for joke in output:
            joke_file.write(joke + "\n")
format()

def search(length, start_with=''):
    search = []
    # load our database of jokes
    with open("jokes_format.txt", "r") as joke_file:
        # read the jokes line by line
        joke_list = joke_file.readlines()
        # iterate through the jokes
        for joke in joke_list:
            # get rid of white spaces
            joke = joke.strip()
            # if we get the right joke candidate, we add it to search
            if len(joke) == length and joke.startswith(start_with):
                search.append(joke)
    # return the list of jokes of given length & start with given characters
    return search

# this will print out list of likely candidates of length 44 and starts with 'WH'
print(search(44, 'WH'))

def play_one_round():
    # read game start message (which includes joke length)
    welcome_to_dad_joke_msg = str(r.recvline())

    # get the joke length (it's stored in the 8th word)
    length = int(welcome_do_dad_joke_msg.split(' ')[8])

    # search for all jokes of this length
    db = search(length)

    # read away useless line
    r.recvuntil('Guess? >')

    # print out what we are guessing
    print(f"{'Guess #1:':<30}{db[0]:<40}")

    # send our first guess to game server
    r.sendline(db[0])

    # read away useless line
    r.recvline()

    # response will return msg that hints to us if correct/wrong
    response = str(r.recvline())

    # print out what our response is
    print(f"{'Response 1:':<30}{response:<40}")

    # guess is not correct if we can find 'position' and 'correct' in the response (both are arrays)
    if 'position' in response and 'correct' in response:
        print('[Guess is Wrong]')

        # get ready for next guess
        r.recvuntil("Guess? >")

        # parseStats() will return the arrays correct, position from the response string
        correct, position = parseStats(response)

        # get the most likely candidate satisfying the given correct & position arrays
        candidates = getCandidates(correct, position, db[0], db)

        # no candidates found :( which means that sth went wrong so go into interactive mode to debug
        if len(candidates) == 0:
            print("[No candidates]")
            r.interactive()

        # at least one candidate found, so let's guess with that
        else:
            # print out what we are guessing
            print(f"{'Guess #2:':<30}{candidates[0]:<40}")

            # send our second guess to game server
            r.sendline(candidates[0])

            # receive response
            response = str(r.recvline())

            # print out what our response is
            print(f"{'Response 2:':<30}{response:<40}")

            # sad
            if str(r.recvline()) == "You lose":
                print('[Guess 2 Wrong]')
                # we failed :(
                return False

    # guess is correct!!!
    else:
        print('[Guess 1 Correct]')
        # yay!
        return True

    # if we reach here it probably means we didn't guess right
    return False

from pwn import *

# connect to the game server
r = remote("0.cloud.chals.io", 33282)

# read away the first useless line
r.recvline()

def restart_game():
    r.close()
    r = remote("0.cloud.chals.io", 33282)
    r.recvline()

def main():
    # start playing!
    games_won = 0
    while True:
        if play_one_round():
            games_win += 1
        else:
            games_won = 0
            restart_game()
            print("It's okay. We try again")
            continue

        if games_won >= 10:
            print('-------------------------------------------')
            print('We did it..!')
            print('-------------------------------------------')
            r.interactive()
            break

main()