py

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
# pip3 install PyMySQL
import tkinter as tk
import pymysql
import csv
import os
window = tk.Tk()
# 标题
window.title('天眼查小工具')
# 窗口大小
window.geometry('500x500')
# 请输入关键字提示文字
label1 = tk.Label(window, text="请输入关键字:")
# 下方提示文字
label2 = tk.Label(window, text="输入关键字点击查询按钮即可搜索，点击导出数据即可导出为excel文件。")
# 关键字输入框
company_name_input = tk.Entry(window, show=None, font=('微软雅黑', 14))
# 查询结果显示文本框
text = tk.Text(window, width=39, height=14, font=('微软雅黑', 15))
# 连接数据库
db = pymysql.connect("localhost", "root", "", "holder")
cursor = db.cursor()
success_con = "程序启动成功。\n数据库连接成功。\n输入关键字开始查询吧..."
text.insert("insert", success_con)
export_res = ()
# 清空
def reset():
company_name_input.delete(0, "end")
text.delete(1.0, "end")
# 查询
def search():
text.delete(1.0, "end")
key_word = company_name_input.get()
key_word = str(key_word).lstrip().rstrip()
do_search(key_word)
# 导出
def export():
if len(export_res) == 0:
text.delete(1.0, "end")
text.insert("insert", '内容为空，无法导出！')
file_path = 'data.csv'
if os.path.exists(file_path):
os.remove(file_path)
csv_file = open(file_path, 'a+', newline='')
try:
writer = csv.writer(csv_file)
header = ('股东', '公司', 'Account UserName', 'Primary UserName', 'Account Segment', 'AM Name')
writer.writerow(header)
for res in export_res:
writer.writerow(res)
path = os.path.abspath(file_path)
text.insert("insert", '---------------\n文件导出完成！\n文件地址：' + path)
finally:
csv_file.close()
# 查询按钮
button_search = tk.Button(window, text="查询", bg="LightBlue", font=('微软雅黑', 12), width=6, height=1, command=search)
# 清空按钮
button_reset = tk.Button(window, text="清空", bg="LightCyan", font=('微软雅黑', 12), width=6, height=1, command=reset)
# 导出按钮
button_export = tk.Button(window, text="导出数据", font=('微软雅黑', 12), width=8, height=1, command=export)
def do_search(key_word):
content = "未查询到任何与" + key_word + "相关的结果！"
if len(key_word) != 0:
cursor.execute(
'select share,name,account_username,primary_username,account_segment,am_name from holder where share in (select share from holder where name like \'%' + key_word + '%\')')
result = cursor.fetchall()
global export_res
export_res = result
if len(result) != 0:
content = "共查询到" + str(len(result)) + "条与" + key_word + "相关的信息：\n"
for res in result:
content += res[0] + " - " + res[1] + " - " + res[2] + " - " + res[3] + " - " + res[
4] + " - " + res[5] + "\n"
else:
content = "请输入关键字！"
text.insert("insert", content)
db.commit()
def main():
# 请输入关键字提示文字坐标
label1.place(x=10, y=15)
# 关键字输入框坐标
company_name_input.place(x=100, y=15)
# 查询按钮坐标
button_search.place(x=340, y=12)
# 清空按钮坐标
button_reset.place(x=420, y=12)
# 查询结果显示文本框坐标
text.place(x=12, y=60)
# 导出按钮坐标
button_export.place(x=400, y=455)
# 下方提示文字坐标
label2.place(x=10, y=465)
window.mainloop()
if __name__ == '__main__':
main()

本文地址：https://blog.csdn.net/King__Cheung/article/details/110917604