import pymysql
# 连接数据库 创建连接对象 con
con = pymysql.connect(
    host="localhost",  # 主机名，使用默认即可（localhost）
    user="root",  # 用户名，一般是root
    password="924604223",  # 密码，登录数据库的密码，这个是自己设置的
    database="atm"  # 要连接的数据库
)
print("数据库连接成功！！！")


# 使用 cursor() 方法创建一个游标对象 cursor
cursor = con.cursor()
# 创建用户表(users)，包含账户（account）、密码(pwd)、金额（money）、账户记录（record)等
create_table_sql = """
    create table if not exists records(
        id INT AUTO_INCREMENT PRIMARY KEY,
        account VARCHAR(20) ,  
        time VARCHAR(25) NOT NULL,
        content VARCHAR(50),       
        money float(10,2)
        )
"""
# 使用execute()方法执行数据库命令
cursor.execute(create_table_sql)
print("记录表创建成功！！！")


def insert_data(account, time, content,money):
    try:
        # 插入数据的SQL语句
        insert_sql = "INSERT INTO records (account, time,content,money) VALUES (%s, %s, %s, %s)"
        data = (account,time, content, money)
        # 执行SQL语句
        cursor.execute(insert_sql, data)
        # 提交事务
        con.commit()
    except:
        print("该账号已存在")

a =[]
def select_data():
    # 查询数据的SQL语句
    select_sql = "SELECT * FROM records"
    # 执行SQL语句
    cursor.execute(select_sql)
    # 获取查询结果
    results = cursor.fetchall()
    for row in results:
        a.append(list(row))
        print(row)

def select_dataA(account):
    # 查询数据的SQL语句
    select_sql = f"SELECT * FROM records where account = {account}"
    # 执行SQL语句
    cursor.execute(select_sql)
    # 获取查询结果
    results = cursor.fetchall()
    for row in results:
        a.append(list(row))
        print(row)


if __name__ == '__main__':
    select_data()
    print("1001的流水记录如下：")
    select_dataA('1001')
    print("1002的流水记录的类型：")