Grab part of the screen with pyton devlog part 1

This is my live attempt to grab part of the screen for a little project I have in mind.

Grab the image

We grab the image with ImageGrab from pyscreenshot. With the class Listener we get the mouse position and then we call the grab function.

# grabscreen.py

import pyscreenshot as ImageGrab
import os
from pynput.mouse import Listener
import sys

def grab(x, y, w, h):
    im = ImageGrab.grab(bbox=(x, y, w, h))
    save(im)


def save(im):
    im.save('im.png')
    os.startfile('im.png')

click1 = 0
x1 = 0
y1 = 0
def on_click(x, y, button, pressed):
    global click1, x1, y1
    
    if pressed:
        if click1 == 0:
            x1 = x
            y1 = y
            click1 = 1
        else:
            grab(x1, y1, x, y)
            listener.stop()
            sys.exit()

print("Click once on top left and once on bottom right")
# with Listener(on_move=on_move, on_click=on_click, on_scroll=on_scroll) as listener:
with Listener(on_click=on_click) as listener:
    listener.join()
    listener.stop()
    sys.exit()

See ya.