143 lines
5.3 KiB
Python
143 lines
5.3 KiB
Python
#!/usr/bin/env python
|
|
import PySimpleGUI as sg
|
|
import os
|
|
import psutil
|
|
import win32api
|
|
import win32con
|
|
import win32gui
|
|
import win32process
|
|
import cv2
|
|
from PIL import ImageGrab
|
|
import numpy as np
|
|
|
|
|
|
def convert_string_to_tuple(string):
|
|
"""
|
|
Converts a string that represents a tuple. These strings have the format:
|
|
"('item 1', 'item 2')"
|
|
The desired return value is ('item 1', 'item 2')
|
|
:param string:
|
|
:return:
|
|
"""
|
|
parts = string[1:-1].split(',')
|
|
part1 = parts[0][1:-1]
|
|
part2 = parts[1][2:-1]
|
|
return part1, part2
|
|
|
|
|
|
def show_list_by_name(window, output_key, python_only):
|
|
process_list = get_window_list()
|
|
|
|
title_list = []
|
|
for proc in process_list:
|
|
names = convert_string_to_tuple(proc)
|
|
if python_only and names[0] == 'python.exe':
|
|
title_list.append(names[1])
|
|
elif not python_only:
|
|
title_list.append(names[1])
|
|
title_list.sort()
|
|
window[output_key].update(title_list)
|
|
return title_list
|
|
|
|
|
|
def get_window_list():
|
|
titles = []
|
|
t = []
|
|
pidList = [(p.pid, p.name()) for p in psutil.process_iter()]
|
|
|
|
def enumWindowsProc(hwnd, lParam):
|
|
""" append window titles which match a pid """
|
|
if (lParam is None) or ((lParam is not None) and (win32process.GetWindowThreadProcessId(hwnd)[1] == lParam)):
|
|
text = win32gui.GetWindowText(hwnd)
|
|
if text:
|
|
wStyle = win32api.GetWindowLong(hwnd, win32con.GWL_STYLE)
|
|
if wStyle & win32con.WS_VISIBLE:
|
|
t.append("%s" % (text))
|
|
return
|
|
|
|
def enumProcWnds(pid=None):
|
|
win32gui.EnumWindows(enumWindowsProc, pid)
|
|
|
|
for pid, pName in pidList:
|
|
enumProcWnds(pid)
|
|
if t:
|
|
for title in t:
|
|
titles.append("('{0}', '{1}')".format(pName, title))
|
|
t = []
|
|
titles = sorted(titles, key=lambda x: x[0].lower())
|
|
return titles
|
|
|
|
|
|
def save_win(filename=None, title=None, crop=True):
|
|
"""
|
|
Saves a window with the title provided as a file using the provided filename.
|
|
If one of them is missing, then a window is created and the information collected
|
|
|
|
:param filename:
|
|
:param title:
|
|
:return:
|
|
"""
|
|
C = 7 if crop else 0 # pixels to crop
|
|
if filename is None or title is None:
|
|
layout = [[sg.T('Choose window to save', font='Any 18')],
|
|
[sg.T('The extension you choose for filename will determine the image format')],
|
|
[sg.T('Window Title:', size=(12, 1)), sg.I(title if title is not None else '', key='-T-')],
|
|
[sg.T('Filename:', size=(12, 1)), sg.I(filename if filename is not None else '', key='-F-')],
|
|
[sg.Button('Ok', bind_return_key=True), sg.Button('Cancel')]]
|
|
event, values = sg.Window('Choose Win Title and Filename', layout).read(close=True)
|
|
if event != 'Ok': # if cancelled or closed the window
|
|
print('Cancelling the save')
|
|
return
|
|
filename, title = values['-F-'], values['-T-']
|
|
try:
|
|
fceuxHWND = win32gui.FindWindow(None, title)
|
|
rect = win32gui.GetWindowRect(fceuxHWND)
|
|
rect_cropped = (rect[0] + C, rect[1], rect[2] - C, rect[3] - C)
|
|
frame = np.array(ImageGrab.grab(bbox=rect_cropped), dtype=np.uint8)
|
|
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
|
cv2.imwrite(filename, frame)
|
|
sg.cprint('Wrote image to file:', filename)
|
|
except Exception as e:
|
|
sg.popup('Error trying to save screenshot file', e, keep_on_top=True)
|
|
|
|
|
|
def main():
|
|
layout = [[sg.Text('Window Snapshot', key='-T-', font='Any 20', justification='c')],
|
|
[sg.Listbox(values=[' '], size=(50, 20), select_mode=sg.SELECT_MODE_EXTENDED, font=('Courier', 12), key='-PROCESSES-')],
|
|
[sg.Checkbox('Show only Python programs', default=True, key='-PYTHON ONLY-')],
|
|
[sg.Checkbox('Crop image', default=True, key='-CROP-')],
|
|
[sg.Multiline(size=(63, 10), font=('Courier', 10), key='-ML-')],
|
|
[sg.Text('Output folder:'), sg.In(os.path.dirname(__file__), key='-FOLDER-'), sg.FolderBrowse()],
|
|
[sg.Button('Refresh'),
|
|
sg.Button('Snapshot', button_color=('white', 'DarkOrange2')),
|
|
sg.Exit(button_color=('white', 'sea green'))]]
|
|
|
|
window = sg.Window('Window Snapshot', layout, keep_on_top=True, auto_size_buttons=False, default_button_element_size=(12, 1), finalize=True)
|
|
|
|
window['-T-'].expand(True, False, False) # causes text to center by expanding the element
|
|
|
|
sg.cprint_set_output_destination(window, '-ML-')
|
|
show_list_by_name(window, '-PROCESSES-', True)
|
|
|
|
# ---------------- main loop ----------------
|
|
while True:
|
|
# --------- Read and update window --------
|
|
event, values = window.read()
|
|
if event in (sg.WIN_CLOSED, 'Exit'):
|
|
break
|
|
|
|
# --------- Do Button Operations --------
|
|
if event == 'Refresh':
|
|
show_list_by_name(window, '-PROCESSES-', values['-PYTHON ONLY-'])
|
|
elif event == 'Snapshot':
|
|
for title in values['-PROCESSES-']:
|
|
sg.cprint('Saving: ', end='', c='white on red')
|
|
sg.cprint(title)
|
|
output_filename = os.path.join(values['-FOLDER-'], f'{title}.png')
|
|
save_win(output_filename, title, values['-CROP-'])
|
|
window.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|