screen.py

# Adriel Freud

from tkinter import *
import pyscreenshot as Image
import random

def main():
	global root
	root = Tk()
	root.title("Screenshooter")

	Label(root, text="Nome: ", fg="black", bg="white", font="Arial 11").place(x=5, y=25)
	
	global name_Entry
	name_Entry = Entry(root, fg="black", bg="white", width=15)
	name_Entry.place(x=60, y=30)

	Button(root, text="Shoot", fg="black", bg="white", command=screenshot).place(x=45, y=100)
	Button(root, text="Config", fg="black", bg="white", command=config).place(x=115, y=100)

	root["bg"] = "white"
	root.geometry("225x150")
	root.mainloop()


def generate_string():
	return "screnn_%s.png"%random.randint(1, 10000)

def config():
	win = Tk()
	win.title("Config")

	Label(win, text="X1", fg="black", bg="white", font="Arial 11").place(x=5,y=20)
	X1 = Entry(win, fg="black", bg="white", width=7)
	X1.place(x=50, y=20)

	Label(win, text="X2", fg="black", bg="white", font="Arial 11").place(x=5,y=40)
	X2 = Entry(win, fg="black", bg="white", width=7)
	X2.place(x=50, y=40)

	Label(win, text="Y1", fg="black", bg="white", font="Arial 11").place(x=5,y=60)
	Y1 = Entry(win, fg="black", bg="white", width=7)
	Y1.place(x=50, y=60)

	Label(win, text="Y2", fg="black", bg="white", font="Arial 11").place(x=5,y=80)
	Y2 = Entry(win, fg="black", bg="white", width=7)
	Y2.place(x=50, y=80)

	def save_config():
		data = (int(X1.get()), int(X2.get()), int(Y1.get()), int(Y2.get()))
		img = Image.grab(bbox=data)
		img.save(generate_string())
		img.close()

	Button(win, text="SAVE", fg="black", bg="white", font="Arial 11", command=save_config).place(x=45, y=130)

	win["bg"] = "white"
	win.geometry("150x170")
	win.mainloop()

def screenshot():
	img = Image.grab()

	if(name_Entry.get()):
		img.save(name_Entry.get())
	else:
		img.save(generate_string())
	img.close()

main()