Phonetically correct word scrambler

wordmap.csv

wordscramble.py

#!/usr/bin/python3

import random

class Scrambler:

    @staticmethod
    def getDict(fn:str = "wordmap.csv"): 
        d = {}
        with open(fn) as f:
            for row in f:
                s = row.split(',')
                if len(s) >= 2:
                    key = s[0].upper()
                    if( d.get(key) ):
                        d[key].append(s[1].upper().strip())
                    else:
                        d[s[0].upper()] = [s[1].upper().strip()] # assign the index before the comma to the element after
        return d

    def __init__(self, wordmap = None):
        self.wordmap = wordmap

        if self.wordmap == None:
            self.wordmap = self.getDict()

        self.maxLen = 1;
        for k in self.wordmap.keys(): # get the max length of the keys
            if len(k) > self.maxLen:
                self.maxLen = len(k)

    def scramble(self, s:str):
        sarr = s.split(' ')
        newarr = []
        for word in sarr:
            newword = ""
            currIdx = 0

            while(currIdx < len(word)):
                for i in range(self.maxLen, 0, -1):
                    translation = self.wordmap.get( word[currIdx:currIdx+i].upper() )

                    if( translation ): # if there is a valid translation
                        newword += random.choice( translation )
                        currIdx += i
                        break
                    elif( i == 1 ): # if there's no translation for a single letter, do not translate it
                        newword += word[currIdx].upper()
                        currIdx += i
                        break

            newarr.append(newword)

        return ' '.join(newarr)


def main():

    sc = Scrambler()

    while(1):
        s = input("Input a phrase: ")
        print(sc.scramble(s))


if __name__ == '__main__':
    main()