【机试题(实现语言：python3)】字符串加密-字符串+列表

题目描述
有一种技巧可以对数据进行加密，它使用一个单词作为它的密匙。下面是它的工作原理：首先，选择一个单词作为密匙，如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。

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

本题有多组输入数据。

输入描述:

先输入key和要加密的字符串

输出描述:

返回加密后的字符串

示例1
输入

nihao
ni

输出

le

解题思路：
需要先根据入参key，推导出密文对照码list2
input_str入参要加密就根据list1和list2对照找出对应的字符

代码实现如下：

import string
def func():
    while True:
        try:
            key = input()
            str1 = string.ascii_uppercase
            list1 = list(str1)
            input_str = input()
            list2 = []
            for i in key:
                if str.upper(i) not in list2:
                    list2.append(str.upper(i))
            for i in list1:
                if i not in list2:
                    list2.append(i)
            res = ''
            for i in input_str:
                if i.isupper():
                    v = i
                    index = list1.index(v)
                    res += list2[index]
                else:
                    v = str.upper(i)
                    index = list1.index(v)
                    res += str.lower(list2[index])
            print(res)
        except Exception as e:
            #print(e)
            break
if __name__ == '__main__':
    func()