九宫格键盘

class Solution:
    def NineGridInputMethod(self, input_str):
        dic = {'1': ',.', '2': 'abc', '3': 'def', '4': 'ghi', '5': 'jkl',
               '6': 'mno', '7': 'pqrs', '8': 'tuv', '9': 'wxyz'}
        num_mode = True  
        length = len(input_str)
        i = 0
        result = ""
        while i < length:
            s: str = input_str[i]
            if s == '#':
                num_mode = not num_mode
                i += 1
                continue
            if s == '/':
                i += 1
                continue
            if num_mode:
                result += s
                i += 1
                continue
            if s == '0':
                result += " "
                i += 1
                continue
            cnt = 0
            while i < length and input_str[i] == s:
                cnt += 1
                i += 1
            string = dic[s]
            result += string[cnt % len(string) - 1]
        return result
 
if __name__ == "__main__":
    line = str(input().strip())
    function = Solution()
    results = function.NineGridInputMethod(line)
    print(results)