import base64
from Crypto.Cipher import AES
import pandas as pd


#  明文aes加密

def triple_aes_encrypt(text):
    key = 'abcdefgh12345678'  # 16字节密钥
    iv = '12345678abcdefgh'  # 16字节初始向量
    cipher = AES.new(key.encode(), AES.MODE_CBC, iv.encode())
    pad_text = text.encode('utf-8')
    padded_text = pad_text + (16 - len(pad_text) % 16) * chr(16 - len(pad_text) % 16).encode('utf-8')
    encrypted = cipher.encrypt(padded_text)
    return base64.b64encode(encrypted).decode()


# 读取到的数据转换为字典列表
def get_list(df,Keylist):
    DataList = []
    for index, row in df.iterrows():
        DataDict = {}
        for key in Keylist:
            DataDict[key] = str(row[key])
        DataList.append(DataDict)
    return DataList


# 读取Excel文件
def read_excel(filename):
    try:
        # 读取Excel文件
        df = pd.read_excel(filename)
        return df
    except Exception as e:
        print(f"读取发生错误:{e}")


# 写入Excel文件
def write_excel(filename,raw,column,content):
    try:
        # 读取Excel文件
        df = pd.read_excel(filename)
        # 确保列为字符串类型
        df[column] = df[column].astype(str)
        # 可以按条件更新其他行，比如按索引更新
        df.at[raw, column] = content
        # 将修改后的 DataFrame 保存回 Excel 文件
        df.to_excel(filename, index=False)
        print("成功写入Excel")
    except Exception as e:
        print(f"发生错误：{e}")


if __name__ == '__main__':
    print(triple_aes_encrypt("Baoyi@1341"))