|
|
import mysql.connector
|
|
|
from mysql.connector import Error
|
|
|
from math import *
|
|
|
import tkinter as t
|
|
|
import psutil
|
|
|
import os
|
|
|
import time
|
|
|
import threading
|
|
|
import datetime
|
|
|
|
|
|
|
|
|
db_config = {
|
|
|
'host': 'localhost',
|
|
|
'user': 'root',
|
|
|
'password': '2015wocani',
|
|
|
'database': 'ProcessData'
|
|
|
}
|
|
|
|
|
|
def init_db_connection():
|
|
|
global db_connection, cursor
|
|
|
try:
|
|
|
db_connection = mysql.connector.connect(**db_config)
|
|
|
cursor = db_connection.cursor()
|
|
|
print("Database connection successful")
|
|
|
except Error as e:
|
|
|
print(f"Error connecting to MySQL: {e}")
|
|
|
|
|
|
# 关闭数据库连接
|
|
|
def close_db_connection():
|
|
|
if db_connection.is_connected():
|
|
|
cursor.close()
|
|
|
db_connection.close()
|
|
|
print("MySQL connection is closed")
|
|
|
|
|
|
|
|
|
def get_db_cursor(connection):
|
|
|
"""获取并返回数据库游标"""
|
|
|
return connection.cursor()
|
|
|
|
|
|
|
|
|
|
|
|
db_connection = create_db_connection(**db_config)
|
|
|
|
|
|
# 定义获取游标的函数
|
|
|
def get_db_cursor(connection):
|
|
|
return connection.cursor()
|
|
|
|
|
|
# 使用上面创建的连接来获取游标
|
|
|
db_cursor = get_db_cursor(db_connection)
|
|
|
|
|
|
|
|
|
root= t.Tk() #主窗口
|
|
|
root.title('任务管理器') #标题
|
|
|
root.geometry('730x630') #窗口尺寸
|
|
|
|
|
|
root.resizable(width=True, height=True)# 设置窗口宽度，高度可变
|
|
|
root.attributes('-alpha',0.9) #设置窗口透明度
|
|
|
root.wm_attributes('-topmost',1) #实现root窗口的置顶显示
|
|
|
|
|
|
sb=t.Scrollbar(root)
|
|
|
sb.pack(side='left',fill='y')
|
|
|
text=t.Text(root,width=100,height=40)
|
|
|
text.place(x=10,y=36)
|
|
|
sb.config(command=text.yview)
|
|
|
text.config(yscrollcommand=sb.set)
|
|
|
sb.pack(side='right',fill='y')
|
|
|
t1=t.Label(text='')
|
|
|
t2=t.Label(text='')
|
|
|
t3=t.Label(text='')
|
|
|
t1.place(x=10,y=580,width=120)
|
|
|
t2.place(x=150,y=580,width=120)
|
|
|
t3.place(x=300,y=580,width=120)
|
|
|
|
|
|
def create_db_connection():
|
|
|
connection = pymysql.connect(host='localhost', user='root', password='2015wocani', database='ProcessData')
|
|
|
return connection
|
|
|
|
|
|
db_connection = create_db_connection(**db_config)
|
|
|
db_cursor = get_db_cursor(db_connection)
|
|
|
def insert_process(pid, name, path, mem_usage, cpu_usage):
|
|
|
"""插入进程数据到数据库"""
|
|
|
try:
|
|
|
sql = """
|
|
|
INSERT INTO ProcessData
|
|
|
(pid, process_name, path, memory_usage, cpu_usage)
|
|
|
VALUES
|
|
|
(%s, %s, %s, %s, %s)
|
|
|
"""
|
|
|
val = (pid, name, path, mem_usage, cpu_usage)
|
|
|
db_cursor.execute(sql, val)
|
|
|
db_connection.commit() # 提交事务
|
|
|
except Exception as e:
|
|
|
print(f"Error inserting process: {e}")
|
|
|
def yy():
|
|
|
text.delete(1.0, 'end')
|
|
|
text.insert('insert', '进程号 ' + '进程名 ' + ' 进程文件路径' + '\n')
|
|
|
|
|
|
for y in psutil.pids():
|
|
|
a = psutil.Process(y)
|
|
|
|
|
|
if a.name() == 'System Idle Process':
|
|
|
continue
|
|
|
else:
|
|
|
# 确保这里的缩进与'else:'对齐，通常使用四个空格
|
|
|
mem_info = a.memory_info()
|
|
|
mem_usage = mem_info.rss / 1024 # 单位转换为KB，可根据需要调整
|
|
|
|
|
|
# 插入数据到数据库
|
|
|
insert_process(a.pid, a.name(), a.exe(), mem_usage, psutil.cpu_percent(interval=None))
|
|
|
|
|
|
text.insert('insert', f"{y} {a.name()} {a.exe()}\n\n")
|
|
|
|
|
|
# 确保root.update()与yy()函数体对齐，不在else块内
|
|
|
root.update()
|
|
|
|
|
|
|
|
|
def jc():
|
|
|
text.delete(1.0,'end')
|
|
|
mm=os.popen('tasklist')
|
|
|
text.insert('insert',mm.read())
|
|
|
root.update()
|
|
|
|
|
|
def fw():
|
|
|
text.delete(1.0,'end')
|
|
|
mm=os.popen('sc query type= service')
|
|
|
text.insert('insert',mm.read())
|
|
|
root.update()
|
|
|
|
|
|
def xn():
|
|
|
text.delete(1.0,'end')
|
|
|
l1=t.Label(root,text='开机时间：')
|
|
|
tm=datetime.datetime.fromtimestamp(psutil.boot_time()).strftime("%Y-%m-%d %H:%M:%S")
|
|
|
l2=t.Label(root,text=str(tm))
|
|
|
l3=t.Label(root,text='当前时间：')
|
|
|
l4=t.Label(root,text='')
|
|
|
dq=time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
|
|
|
l4.configure(text=str(dq))
|
|
|
l5=t.Label(root,text='物理内存使用情况(MB)：')
|
|
|
l6=t.Label(root,text='')
|
|
|
jh=psutil.virtual_memory()
|
|
|
tt=int((jh.total)/1024/1024)
|
|
|
us=int((jh.used)/1024/1024)
|
|
|
fr=int((jh.free)/1024/1024)
|
|
|
l6.configure(text='总量：' + str(tt) +'\n'+'使用：'+str(us) +'\n'+'剩余：'+str(fr))
|
|
|
l7=t.Label(root,text='交换内存使用情况(MB)：')
|
|
|
l8=t.Label(root,text='')
|
|
|
hj=psutil.swap_memory()
|
|
|
ht=int((hj.total)/1024/1024)
|
|
|
hu=int((hj.used)/1024/1024)
|
|
|
hf=int((hj.free)/1024/1024)
|
|
|
l8.configure(text='总量：' + str(ht) + ' '+'使用：'+str(hu) +' '+'剩余：'+str(hf))
|
|
|
text.window_create('insert',window=l1)
|
|
|
text.window_create('insert',window=l2)
|
|
|
text.insert('insert','\n\n')
|
|
|
text.window_create('insert',window=l3)
|
|
|
text.window_create('insert',window=l4)
|
|
|
text.insert('insert','\n\n')
|
|
|
text.window_create('insert',window=l5)
|
|
|
text.window_create('insert',window=l6)
|
|
|
text.insert('insert','\n\n')
|
|
|
text.window_create('insert',window=l7)
|
|
|
text.window_create('insert',window=l8)
|
|
|
root.update()
|
|
|
|
|
|
def lw():
|
|
|
text.delete(1.0,'end')
|
|
|
n = psutil.net_io_counters()
|
|
|
r=str(float(n.bytes_recv / 1024 / 1024))+'MB'
|
|
|
s= str(float(n.bytes_sent / 1024 / 1024))+'MB'
|
|
|
text.insert('insert','网卡接收流量: '+str(r)+'\n'+'网卡发送流量：'+str(s)+'\n')
|
|
|
root.update()
|
|
|
|
|
|
def yh():
|
|
|
text.delete(1.0,'end')
|
|
|
use=' 用户'+' '+' 状态'+'\n'
|
|
|
text.insert('insert',use)
|
|
|
for y in psutil.users():
|
|
|
text.insert('2.0',str(y.name)+' '+'运行中。。。。'+'\n')
|
|
|
root.update()
|
|
|
|
|
|
def end_process():
|
|
|
"""根据用户输入的PID结束进程"""
|
|
|
pid = pid_entry.get()
|
|
|
try:
|
|
|
pid = int(pid) # 将输入转换为整型
|
|
|
process = psutil.Process(pid)
|
|
|
process.terminate() # 尝试结束进程
|
|
|
text.insert('insert', f"尝试结束PID为{pid}的进程...\n")
|
|
|
root.update()
|
|
|
except psutil.NoSuchProcess:
|
|
|
text.insert('insert', f"没有找到PID为{pid}的进程。\n")
|
|
|
root.update()
|
|
|
except ValueError:
|
|
|
text.insert('insert', "请输入有效的数字PID。\n")
|
|
|
root.update()
|
|
|
except Exception as e:
|
|
|
text.insert('insert', f"结束进程时发生错误: {e}\n")
|
|
|
root.update()
|
|
|
|
|
|
|
|
|
def search_processes():
|
|
|
"""根据用户输入的关键词搜索进程并更新显示"""
|
|
|
keyword = search_entry.get().strip().lower()
|
|
|
if not keyword:
|
|
|
text.insert('insert', "请输入搜索关键词...\n")
|
|
|
return
|
|
|
|
|
|
found_processes = []
|
|
|
for proc in psutil.process_iter(['pid', 'name']):
|
|
|
if keyword in proc.info['name'].lower():
|
|
|
found_processes.append(proc.info)
|
|
|
|
|
|
if found_processes:
|
|
|
text.delete(1.0, 'end')
|
|
|
for proc_info in found_processes:
|
|
|
text.insert('insert', f"{proc_info['pid']} | {proc_info['name']}\n")
|
|
|
else:
|
|
|
text.insert('insert', f"未找到包含'{keyword}'的进程。\n")
|
|
|
root.update()
|
|
|
|
|
|
|
|
|
# 确保窗口大小适应新的组件
|
|
|
search_frame = t.Frame(root)
|
|
|
search_frame.pack(side='right', anchor='ne') # 定位在右侧上角
|
|
|
|
|
|
# 通过grid布局管理器，可以更好地控制组件之间的对齐和间距
|
|
|
search_frame.columnconfigure(0, weight=1) # 让输入框所在的列自动填充空白
|
|
|
search_frame.rowconfigure(0, weight=1)
|
|
|
|
|
|
search_label = t.Label(search_frame, text="搜索进程:")
|
|
|
search_label.grid(row=0, column=0, sticky='w') # 保持左侧对齐，但不额外增加宽度
|
|
|
|
|
|
search_entry = t.Entry(search_frame)
|
|
|
search_entry.grid(row=0, column=1, sticky='ew') # 让输入框填充可用空间
|
|
|
search_entry.config(width=20) # 可选：限制输入框的初始宽度
|
|
|
|
|
|
search_button = t.Button(search_frame, text="搜索", command=search_processes)
|
|
|
search_button.grid(row=0, column=2, padx=(0, 5), pady=(0, 5), sticky='e') # 靠右对齐，可加微小间隔
|
|
|
|
|
|
|
|
|
m=t.Menu(root)
|
|
|
#文件菜单
|
|
|
file=t.Menu(m,tearoff=False)
|
|
|
m.add_cascade(label='文件', menu=file)
|
|
|
file.add_command(label='新建任务',accelerator='(N)')
|
|
|
file.add_command(label='退出任务栏管理器',command=root.quit,accelerator='(x)')
|
|
|
|
|
|
|
|
|
|
|
|
pid_end_frame = t.Frame(root)
|
|
|
pid_end_frame.pack(side=t.RIGHT, anchor=t.SE, padx=10, pady=10) # 添加padx和pady以留出空间
|
|
|
|
|
|
pid_label = t.Label(pid_end_frame, text="PID:")
|
|
|
pid_label.pack(side=t.LEFT)
|
|
|
|
|
|
pid_entry = t.Entry(pid_end_frame, width=5)
|
|
|
pid_entry.pack(side=t.LEFT, padx=(10, 0))
|
|
|
|
|
|
end_process_button = t.Button(pid_end_frame, text="结束进程", command=end_process)
|
|
|
end_process_button.pack(side=t.LEFT, padx=(10, 0)) # 可能需要根据实际调整padx以保持布局美观
|
|
|
|
|
|
root.geometry("400x600")
|
|
|
#选项菜单
|
|
|
ii=t.IntVar()
|
|
|
ii.set(1)
|
|
|
o=t.Menu(m,tearoff=False)
|
|
|
m.add_cascade(label='选项',menu=o)
|
|
|
o.add_radiobutton(label='前端显示',variable=ii, value=0)
|
|
|
o.add_radiobutton(label='使用时最小化',variable=ii, value=1)
|
|
|
o.add_radiobutton(label='最小化时隐藏',variable=ii, value=2)
|
|
|
|
|
|
|
|
|
#查看菜单
|
|
|
v=t.Menu(m,tearoff=False)
|
|
|
m.add_cascade(label='查看',menu=v)
|
|
|
v.add_command(label='立即刷新')
|
|
|
#二级菜单
|
|
|
iv=t.IntVar()
|
|
|
iv.set(1)
|
|
|
s=t.Menu(v,tearoff=False)
|
|
|
v.add_cascade(label='更新速度',menu=s)
|
|
|
s.add_radiobutton(label='高',variable=iv, value=0)
|
|
|
s.add_radiobutton(label='普通',variable=iv, value=1)
|
|
|
s.add_radiobutton(label='低',variable=iv, value=2)
|
|
|
s.add_radiobutton(label='暂停',variable=iv, value=3)
|
|
|
v.add_command(label='选项列')
|
|
|
|
|
|
|
|
|
#帮助菜单
|
|
|
h=t.Menu(m,tearoff=False)
|
|
|
m.add_cascade(label='帮助',menu=h)
|
|
|
h.add_command(label='任务管理器帮助主体')
|
|
|
h.add_command(label='关于任务管理器')
|
|
|
|
|
|
b1=t.Button(root,text='应用程序',command=yy)
|
|
|
b2=t.Button(root,text='进程',command=jc)
|
|
|
b3=t.Button(root,text='服务',command=fw)
|
|
|
b4=t.Button(root,text='性能',command=xn)
|
|
|
b5=t.Button(root,text='联网',command=lw)
|
|
|
b6=t.Button(root,text='用户',command=yh)
|
|
|
b1.place(x=10,y=15,height=20,width=60)
|
|
|
b2.place(x=70,y=15,height=20,width=60)
|
|
|
b3.place(x=130,y=15,height=20,width=60)
|
|
|
b4.place(x=190,y=15,height=20,width=60)
|
|
|
b5.place(x=250,y=15,height=20,width=60)
|
|
|
b6.place(x=310,y=15,height=20,width=60)
|
|
|
|
|
|
|
|
|
|
|
|
def jcs():
|
|
|
t1.configure(text='进程数：'+str(len(psutil.pids())))
|
|
|
root.after(3000,jcs)
|
|
|
def cpu():
|
|
|
pp=str(ceil(psutil.cpu_percent(1)))
|
|
|
t2.configure(text='CPU 使用率：'+pp+'%')
|
|
|
root.after(1500,cpu)
|
|
|
def wlnc():
|
|
|
f= psutil.virtual_memory().free #剩余内存
|
|
|
t=psutil.virtual_memory().total#总内存
|
|
|
wl= float(t-f)/float(t) #为使得最后值更精确，必须用float
|
|
|
t3.configure(text='物理内存：'+str(floor(wl*100))+'%')
|
|
|
root.after(2000,wlnc)
|
|
|
root.bind('<Visibility>',yy) #当打开窗口时显示第一个按钮选项
|
|
|
root.bind('<Visibility>',jcs())
|
|
|
root.bind('<Visibility>',cpu())
|
|
|
root.bind('<Visibility>',wlnc())
|
|
|
root.configure(menu=m)
|
|
|
root.mainloop() #主窗口循环显示 |
