当前位置：首页 > news >正文

Python利用pyobdc和tkinter框架连接数据库2-保存配置文件

news 来源：原创 2025/5/15 1:14:45

为了实现保存用户输入的数据库连接信息，并在下次打开程序时自动加载这些信息，可以使用配置文件（如 .txt 或 .json 文件）来存储这些信息。以下是一个完整的实现方案，结合了动态输入和自动加载配置文件的功能。

实现步骤

保存配置文件：当用户输入连接信息并成功连接后，将这些信息保存到一个配置文件中。
加载配置文件：程序启动时，检查配置文件是否存在。如果存在，则自动加载配置文件中的信息到输入框中。
提供手动输入和保存选项：用户可以选择手动输入新的连接信息，并保存这些信息供下次使用。

示例代码

以下代码展示了如何实现这一功能：

import tkinter as tk
from tkinter import messagebox
import pyodbc
import json
import os

# 配置文件路径
CONFIG_FILE = "db_config.json"

# 保存配置信息到文件
def save_config(config):
    with open(CONFIG_FILE, "w") as file:
        json.dump(config, file)
    messagebox.showinfo("保存成功", "连接信息已保存，下次打开时将自动加载。")

# 加载配置信息
def load_config():
    if os.path.exists(CONFIG_FILE):
        with open(CONFIG_FILE, "r") as file:
            return json.load(file)
    return None

# 连接数据库
def connect_to_database():
    try:
        # 获取用户输入的连接信息
        server = server_entry.get()
        database = database_entry.get()
        username = username_entry.get()
        password = password_entry.get()

        # 构建连接字符串
        connection_string = (
            f"DRIVER={{ODBC Driver 17 for SQL Server}};"
            f"SERVER={server};"
            f"DATABASE={database};"
            f"UID={username};"
            f"PWD={password}"
        )

        # 尝试连接数据库
        connection = pyodbc.connect(connection_string)
        messagebox.showinfo("成功", "数据库连接成功！")

        # 保存连接信息到配置文件
        config = {
            "server": server,
            "database": database,
            "username": username,
            "password": password
        }
        save_config(config)

        # 在这里可以添加后续操作，例如执行查询
        cursor = connection.cursor()
        cursor.execute("SELECT @@VERSION")  # 查询数据库版本
        result = cursor.fetchone()
        messagebox.showinfo("数据库版本", result[0])

        # 关闭连接
        connection.close()
    except pyodbc.Error as e:
        messagebox.showerror("错误", f"连接失败：{e}")

# 创建主窗口
root = tk.Tk()
root.title("动态连接数据库")

# 加载配置文件中的信息
config = load_config()
if config:
    server_default = config.get("server", "")
    database_default = config.get("database", "")
    username_default = config.get("username", "")
    password_default = config.get("password", "")
else:
    server_default = ""
    database_default = ""
    username_default = ""
    password_default = ""

# 创建输入框和标签
tk.Label(root, text="服务器地址:").grid(row=0, column=0, padx=10, pady=5)
server_entry = tk.Entry(root, width=30)
server_entry.grid(row=0, column=1, padx=10, pady=5)
server_entry.insert(0, server_default)  # 加载默认值

tk.Label(root, text="数据库名称:").grid(row=1, column=0, padx=10, pady=5)
database_entry = tk.Entry(root, width=30)
database_entry.grid(row=1, column=1, padx=10, pady=5)
database_entry.insert(0, database_default)  # 加载默认值

tk.Label(root, text="用户名:").grid(row=2, column=0, padx=10, pady=5)
username_entry = tk.Entry(root, width=30)
username_entry.grid(row=2, column=1, padx=10, pady=5)
username_entry.insert(0, username_default)  # 加载默认值

tk.Label(root, text="密码:").grid(row=3, column=0, padx=10, pady=5)
password_entry = tk.Entry(root, show="*", width=30)
password_entry.grid(row=3, column=1, padx=10, pady=5)
password_entry.insert(0, password_default)  # 加载默认值

# 创建连接按钮
connect_button = tk.Button(root, text="连接数据库", command=connect_to_database)
connect_button.grid(row=4, column=0, columnspan=2, pady=10)

# 运行主循环
root.mainloop()

功能说明

保存配置文件：
- 成功连接后，程序会将连接信息保存到 db_config.json 文件中。
- 保存的配置信息包括服务器地址、数据库名称、用户名和密码。
加载配置文件：
- 程序启动时，会检查 db_config.json 文件是否存在。
- 如果存在，会自动将配置文件中的信息加载到输入框中。
手动输入和保存：
- 用户可以选择手动输入新的连接信息。
- 新的连接信息在成功连接后会覆盖原有的配置文件。

注意事项

安全性：
- 配置文件中存储了敏感信息（如用户名和密码）。在实际应用中，建议对密码进行加密处理，或者使用更安全的存储方式（如环境变量或加密的密钥管理工具）。
- 如果不需要保存密码，可以在保存配置时忽略密码字段。
配置文件格式：
- 示例中使用了 .json 文件格式，因为它简单且易于解析。你也可以选择其他格式（如 .txt 或 .ini 文件），但需要相应地调整代码。
错误处理：
- 示例代码中已经包含了基本的错误处理逻辑。你可以根据需要进一步扩展，例如处理文件读写错误。