# 使用標準程式庫中 random 的 randint()
from random import randint

# 定義 Encrypt 類別
class Encrypt:
    # 定義建構子
    def __init__(self):
        # 建立密碼表
        self.set_code()

    # _code 的 getter
    @property
    def code(self):
        return self.__code

    # _code 的 setter
    def set_code(self):
        # 取得 a 、 b 值
        a = 0
        b = 0
        while a % 2 == 0:
            a = randint(0, 9)
            b = randint(0, 9)
        # 利用公式建立密碼表
        self.__code = ""
        c = "a"
        i = 0
        while i < 26:
            x = c
            y = ord(x) * a + b
            m = y % 26
            self.__code += chr(m + 97)
            c = chr(ord(c) + 1)
            i += 1

    # 編碼的方法
    def toEncode(self, str):
        # 暫存編碼結果的字串
        result = ""

        # 利用迴圈走完參數字串的所有字元
        for c in str:
            # 判斷該字元是否為英文小寫字母
            # 若是英文小寫字母就進行編碼轉換
            c1 = ord(c) >= 97
            c2 = ord(c) <= 122
            if c1 and c2:
                m = ord(c) - 97
                result += self.code[m]
            else:
                result += c

        # 結束回傳編碼過的字串
        return result

    # 解碼的方法
    def toDecode(self, str):
        # 暫存解碼結果的字串
        result = ""

        i = 0
        # 第一層迴圈逐一取得每一個字元
        while i < len(str):
            # 判斷該字元是否為英文小寫字母
            # 若是英文小寫字母就進行解碼轉換
            i1 = ord(str[i]) >= 97
            i2 = ord(str[i]) <= 122
            if i1 and i2:
                j = 0
                # 第二層迴圈尋找該字元在密碼表中
                # 的索引值，加上 DIFF 就可轉換
                # 回原本的字元
                while j < len(self.code):
                    if str[i] == self.code[j]:
                        result += chr(j + 97)
                    j += 1
            else:
                result += str[i]

            i += 1

        # 結束回傳解碼過的字串
        return result

# 測試部分
if __name__ == '__main__':
    e = Encrypt()
    print()
    print(e.code)
    s1 = "There is no spoon."
    print("Input : " + s1)
    s2 = e.toEncode(s1)
    print("Encode: " + s2)
    s3 = e.toDecode(s2)
    print("Decode: " + s3)
    print()

# 檔名: encrypt07.py
# 說明:《Python入門指南》的範例程式
# 網站: http://kaiching.org
# 作者: 張凱慶
# 時間: 2023 年 5 月

if __name__ == '__main__':
    e = Encrypt()
    print()
    print(e.code)
    s1 = "There is no spoon."
    print("Input : " + s1)
    s2 = e.toEncode(s1)
    print("Encode: " + s2)
    s3 = e.toDecode(s2)
    print("Decode: " + s3)
    print()