字符串加密

题目：

有一种技巧可以对数据进行加密，它使用一个单词作为它的密匙。下面是它的工作原理：首先，选择一个单词作为密匙，如TRAILBLAZERS。
如果单词中包含有重复的字母，只保留第1个，其余几个丢弃。现在，修改过的那个单词属于字母表的下面，如下所示：

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

T R A I L B Z E S C D F G H J K M N O P Q U V W X Y

上面其他用字母表中剩余的字母填充完整。在对信息进行加密时，信息中的每个字母被固定于顶上那行，
并用下面那行的对应字母一一取代原文的字母(字母字符的大小写状态应该保留)。因此，使用这个密匙，Attack AT DAWN(黎明时攻击)
就会被加密为Tpptad TP ITVH。

请实现下述接口，通过指定的密匙和明文得到密文。

代码：

while True:
    try:
        str1 = input().upper()
        s = input()
        # str1 = 'nihao'.upper()
        # s = 'ni'
        list1 = []  #存储生成的26个大写字母
        list2 = []  #存储输入去重后的字符串内容
        res = '' #新建一个空字符串
        for i in range(97,123):  #生成26个大写字母的列表
            list1.append(chr(i).upper())
        #print(list1)
        #str1 = input()

        for i in str1:
            if i not in list2:
                list2.append(i)
        #print(list2)    打印list2
        for i in list1:
            if i not in list2 and len(list2) <= 26:
                list2.append(i)
        #print(list2)
        for i in s:
            if i in list1:  # 说明i是大写字母，因为alpha字母表默认值是大写字母
                res = res + list2[list1.index(i)]   # 索引值相同，根据字母表索引值在新字母表中取值后追加给返回结果
            else:  # i为小写字母
                res = res + list2[list1.index(i.upper())].lower()  # 将小写字母转换为大写字母取索引值后，追加返回结果之前将字母还原为小写字母

        print(res)
    except :
        break