Added base daun functionality

This commit is contained in:
BarsTiger
2022-04-08 21:27:26 +03:00
parent da79421bc1
commit 57087f2607
10 changed files with 206 additions and 1 deletions

31
modules/process.py Normal file
View File

@@ -0,0 +1,31 @@
# Module for windows process related functions
import psutil
def get_pid(process_name: str) -> int:
"""
Get the PID by name
"""
if process_name:
for proc in psutil.process_iter():
if process_name == proc.name():
return proc.pid
def get_location(process_name: str = None, pid: int = None) -> str:
"""
Get the location of a process
Use either process_name or pid
"""
if process_name:
pid = get_pid(process_name)
return psutil.Process(pid).exe() if psutil.Process(pid).exe() != psutil.Process().exe() else None
def kill(process_name: str = None, pid: int = None) -> None:
"""
Stop a process by name or PID
"""
if process_name:
pid = get_pid(process_name)
psutil.Process(pid).kill()