1.背景：

最近公司项目的用户中心模块经常出现线上问题，测试人员需要将线上真实数据导入到测试环境中去复现Bug。公司有3套测试环境，来回切换并校验数据比较麻烦，于是就有了如下的数据库同步神器。

2.界面：

3.源代码：

# -*- coding: utf-8 -*-
"""
@date:  2022/01/16 22:00
@author: Anker
@python：v3.6
"""
import json
import tkinter as tk
import tkinter.filedialog
import tkinter.messagebox
import pymysql as pymysql
from tkinter import END
from pymysql.cursors import DictCursor# 线上数据库
online_connect = "erdshggadkjhg2wrtyuic.sfgh.aliyuncs.com"
online_user = "Anker"
online_password = "123456"
online_port = "3306"
online_db_name = "user_center"
online_table = "uuc_business_user"# 测试数据库
test_connect = "bm-cp134dfhdh5e7uho.mysql.rds.aliyuncs.com"
test_user = "root"
test_port = "3306"
test_password = "Vr9ybte9hQJMghjkljh"
test_table = "uuc_business_user"# 创建一个窗口
window = tk.Tk()
window.title("数据库同步工具")
window.geometry('600x600')# 将new_user_data变量初始化
new_user_data = Nonedef data_synchronization():# 定义一个全局变量new_user_dataglobal new_user_data# 每次触发操作之前，清空多行文本框的内容text.delete("1.0", "end")# 定义两个变量name和phone，用来获取用户在文本框输入的内容name = entry1.get()phone = entry2.get()# 连接线上数据库数据库conn_online_sql = pymysql.connect(host=online_connect, user=online_user, cursorclass=DictCursor,password=online_password, database=online_db_name)curs_online = conn_online_sql.cursor()# 通过字典去读取不同的测试环境db_config = radio.get()config = {"0": "0", "test1环境": "user_center_ding", "test2环境": "user_center_ding_test2", "test3环境": "test3"}test_db_name = config[db_config]try:if not (entry1.get() and entry2.get()):tk.messagebox.showinfo(title="提示", message="用户姓名、用户手机为必填项！")elif radio.get() == "0":tk.messagebox.showinfo(title="提示", message="请选择数据库")elif radio.get() == "test3环境":tk.messagebox.showinfo(title="提示", message="目前没有该测试数据库")else:# 连接测试数据库数据库"conn_test2_sql = pymysql.connect(host=test_connect, user=test_user, password=test_password,database=test_db_name)curs2 = conn_test2_sql.cursor()# 读取线上数据库的用户信息read_online_sql = f'''select * from `{online_db_name}`.`{online_table}` where name = "{name}" and phone = "{phone}" '''# 读取测试数据库的用户信息read_test2_sql = f'''select * from `{test_db_name}`.`{test_table}` where name = "{name}" and phone = "{phone}" '''# 执行sql查询，查询线上数据库，将查询出来的结果保存在online_user_data中curs_online.execute(read_online_sql)online_user_data = curs_online.fetchone()print("线上数据库该用户数据为：" + '\n' + str(online_user_data) + '\n')# 判断线上数据库是否有该用户if not online_user_data:text.insert("insert", "线上数据库没有该用户！" + '\n')else:# 执行下面这句后data为{"content": "x", "created_time": "2019-06-19 13:28:30"}，是str类型data = json.dumps(online_user_data, default=str, ensure_ascii=False)# 把data转换为字典类型data_result = json.loads(data)print("转换为dict类型后的数据为：", '\n', data_result, '\n')# 把data再次转为json类型json_result = json.dumps(data_result)print("转换为json类型后的数据为：" + '\n' + json_result + '\n')# 获取返回字典data_result值的所有value的列表new_user_data = tuple(data_result.values())print(new_user_data)# 执行sql查询，查询测试数据库，将查询出来的结果保存在test_user_data中curs2.execute(read_test2_sql)test2_user_data = curs2.fetchone()# 提交数据至线上数据库conn_online_sql.commit()# 判断测试数据库是否有该用户if test2_user_data:text.insert("insert", "测试数据库已存在该用户数据！" + '\n')# 如果线上数据库有该用户，测试数据库没有该用户，则执行同步操作if online_user_data and (not test2_user_data):try:# 插入的SQL语句insert_sql = f'''insert into `{test_db_name}`.`{test_table}` VALUES {new_user_data}'''# 执行SQL，向测试数据库中插入数据curs2.execute(insert_sql)print(insert_sql)text.insert("insert", "数据同步成功！" + '\n')except IOError:text.insert("insert", "数据同步失败！" + '\n')conn_test2_sql.commit()curs_online.close()curs2.close()except IOError:print("数据插入失败")conn_online_sql.rollback()tkinter.filedialog.Directory()conn_online_sql.close()# 清空按钮函数：用来清空输入文本框中的内容
def clear():entry1.delete(0, END)  # 清空Entry文本框中的内容entry2.delete(0, END)  # 清空Entry文本框中的内容text.delete("1.0", "end")  # 清空text多行文本框中的内容radio.set("0")  # 清空Radiobutton标签选项框def selection():text.delete("1.0", "end")choose = str(radio.get())text.insert("insert", "您选择的是：" + choose + '\n')# 创建一个标签
tk.Label(window, text="线上用户姓名：").place(x=10, y=50)
tk.Label(window, text="线上用户手机：").place(x=10, y=120)
# 创建一个Entry文本框：用来输入一行文本字符串
entry1 = tk.Entry(window, bd=3, width=62, highlightcolor='red')
entry1.place(x=120, y=50)
entry2 = tk.Entry(window, bd=3, width=62, highlightcolor='green')
entry2.place(x=120, y=120)
# 创建一个可以输入多行文字的文本框
text = tk.Text(window, height=20, width=75, bg='white', highlightcolor='blue', highlightthickness=1)
text.place(x=35, y=240)
# 创建两个按钮：选择文件夹、退出窗口
tk.Button(window, text='开始同步', bg='grey', fg='white', command=data_synchronization).place(x=130, y=520)
tk.Button(window, text='清空', bg='grey', fg='white', command=clear).place(x=290, y=520)
tk.Button(window, text='退出', bg='grey', fg='white', command=window.quit).place(x=430, y=520)radio = tk.StringVar()
radio.set("0")
tk.Label(window, text='请选择数据库：', bg='green', fg='white', width=12).place(x=10, y=180)
# 创建3个radiobutton选项
radio1 = tk.Radiobutton(window, text='test1环境', variable=radio, command=selection, value="test1环境")
radio1.place(x=140, y=180)
radio2 = tk.Radiobutton(window, text='test2环境', variable=radio, command=selection, value="test2环境")
radio2.place(x=290, y=180)
radio3 = tk.Radiobutton(window, text='test3环境', variable=radio, command=selection, value="test3环境")
radio3.place(x=440, y=180)# 主循环
window.mainloop()