Added base daun functionality
This commit is contained in:
31
modules/process.py
Normal file
31
modules/process.py
Normal 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()
|
||||
Reference in New Issue
Block a user