"""
#*********************************#
#* Grail *#
#*********************************#
#* Author: Lukas Mahler *#
#* Copyright: (C)2018-2020 *#
#* Version: 0.5.0 *#
#* Date: 25.12.2020 *#
#*********************************#
# Instant Upload Screenshot-Tool #
# based on the idea of Gyazoo. #
# -> MS Windows only! #
#*********************************#
"""
__author__ = "Lukas Mahler"
__copyright__ = "Copyright 2018-2021"
__version__ = "0.5.1"
__date__ = "16.01.2021"
__email__ = "lm@ankerlab.de"
__status__ = "Development"
try:
# Defaults
import os
import sys
import webbrowser
from io import BytesIO
from shutil import rmtree
from getpass import getpass
from tempfile import gettempdir
from socket import create_connection
from configparser import ConfigParser
from time import gmtime, strftime, sleep
# GUI
from PyQt4 import QtGui, QtCore, Qt
from PyQt4.QtCore import QProcess, QPoint
from fillscreen import Ui_View
# Customs
import win32gui
import win32clipboard
import paramiko
from desktopmagic.screengrab_win32 import getRectAsImage
except ImportError as e:
print("ERROR: Missing Module |||", e)
sleep(10)
os.system("PAUSE>NUL")
sys.exit()
###################################################################################################
###################################################################################################
class Fillscreen(QtGui.QWidget, Ui_View):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
###############################################################################################
###############################################################################################
# Bastelstüble wegen Transparentem Window welches aber den Input auf unterliegendes blockieren soll
# self.setAttribute(QtCore.Qt.WA_TranslucentBackground, True)
# self.setBackgroundRole(QtGui.QPalette.Base)
# self.setAttribute(Qt.Qt.WA_NoSystemBackground, True)
self.setWindowFlags(
QtCore.Qt.WindowStaysOnTopHint |
QtCore.Qt.FramelessWindowHint |
QtCore.Qt.X11BypassWindowManagerHint
)
# self.istransparent = True
self.setupUi(self)
# self.setWindowOpacity(0.3)
# Qcolor = "rgba(255, 255, 255, 0)"
# self.setStyleSheet("background-color:{0}".format(Qcolor))
###############################################################################################
###############################################################################################
# Change Cursor to select tool on Startup
self.change_cursor()
self.begin = QtCore.QPoint()
self.end = QtCore.QPoint()
self.startingpoint = QtCore.QPoint()
self.endpoint = QtCore.QPoint()
def showFullscreen(self):
self.showMaximized()
# https://stackoverflow.com/a/44468898/5593051
def paintEvent(self, event):
qp = QtGui.QPainter(self)
# br = QtGui.QBrush(QtGui.QColor(100, 10, 10, 40))
qp.setPen(QtGui.QPen(QtCore.Qt.red, 3))
qp.drawRect(QtCore.QRect(self.begin, self.end))
def mousePressEvent(self, event):
if event.button() == QtCore.Qt.RightButton:
print("DEBUG: Exiting on Right Click Request")
app.quit()
else:
self.begin = event.pos()
self.end = event.pos()
# self.startingpoint = event.pos()
self.startingpoint = get_real_pos()
# print("Startpoint:",self.startingpoint)
self.update()
def mouseMoveEvent(self, event):
self.end = event.pos()
self.update()
def mouseReleaseEvent(self, event):
self.begin = event.pos()
self.end = event.pos()
self.endpoint = event.pos()
self.endpoint = get_real_pos()
# print("Endpoint:",self.endpoint)
self.update()
do_screen(self.startingpoint, self.endpoint)
app.quit()
@staticmethod
def change_cursor():
# http://doc.qt.io/archives/qt-4.8/qcursor.html
QtGui.QApplication.setOverrideCursor(QtCore.Qt.CrossCursor)
# QtGui.QApplication.restoreOverrideCursor()
###################################################################################################
###################################################################################################
def read_ini(rootdir):
myini = {}
config = ConfigParser()
try:
config.read_file(open(rootdir+'settings.ini'))
except Exception as e:
print("ERROR: Couldn't locate a 'settings.ini' |||", e)
# Creating new blank ini
with open(rootdir+'settings.ini', 'w') as f:
txt = "[MAIN]\n" \
"# MODE EITHER ONLINE OR OFFLINE\nMODE = OFFLINE\n\n\n" \
"# IF MODE IS ONLINE PROVIDE THE SFTP INFO BELOW\n" \
"[ONLINE]\n\n" \
"# URL EITHER IS AN IP OR A FQDN\nURL = \n\n" \
"# PORT IS THE SSH PORT\nPORT = 22\n\n" \
"# THE TIME IN SECONDS TO TRY TO CONNECT\nTIMEOUT = 5\n\n" \
"# YOUR SSH USER, THIS CAN BE BLANK TO GET ASKED INTERACTIVELY\nUSR = \nPASS = \n\n" \
"# THE LOCAL SERVER PATH WHERE TO PUT THE SCREENSHOT\nPTH = ./Screens/\n\n" \
"# THIS SHOULD BE THE FULL ONLINE URL\nWHERE = \n\n\n" \
"# IF MODE IS OFFLINE YOU MAY CHANGE THE SETTINGS BELOW\n" \
"[OFFLINE]\n\n" \
"WHERE = "
f.write(txt)
mode = config.get('MAIN', 'MODE')
if mode == "ONLINE":
url = config.get('ONLINE', 'URL')
port = int(config.get('ONLINE', 'PORT'))
timeout = int(config.get('ONLINE', 'TIMEOUT'))
usr = config.get('ONLINE', 'USR')
passw = config.get('ONLINE', 'PASS')
pth = config.get('ONLINE', 'PTH')
where = config.get('ONLINE', 'WHERE')
myini.update(
mode = mode,
URL = url,
PORT = port,
TIMEOUT = timeout,
USR = usr,
PASS = passw,
PTH = pth,
WHERE = where
)
elif mode == "OFFLINE":
where = config.get('OFFLINE', 'WHERE')
myini.update(
mode = mode,
WHERE = where
)
else:
print("ERROR: you seem to be having a corrupt or broken 'settings.ini' please recheck!")
return myini
def get_real_pos():
flags, hcursor, (x, y) = win32gui.GetCursorInfo()
return {"x": x, "y": y}
def calc_frame(startinspot, endspot):
# If dragging began from top
if startinspot['x'] < endspot['x']:
left = startinspot['x']
right = endspot['x']
top = startinspot['y']
bottom = endspot['y']
# If dragging began from bottom
else:
left = endspot['x']
right = startinspot['x']
top = endspot['y']
bottom = startinspot['y']
# print("DEBUG: Start:",left,"x",top)
# print("DEBUG: End :",right,"x",bottom)
if abs(left-right) <= 10 and abs(top-bottom) <= 10:
coordinates = None
else:
coordinates = (left, top, right+1, bottom+1)
return coordinates
def do_screen(start, end):
""" This Function somehow is the Brain now """
# QtGui.QPixmap.grabWindow(QtGui.QApplication.desktop().winId()).save('screenshot.jpg', 'jpg')
# Hide all windows to take a Screenshot of the underlying interface
for window in windows:
window.hide()
frame = calc_frame(start, end)
print("Took Snapshot on Coordinates:", frame)
# https://github.com/python-pillow/Pillow/issues/1547
# https://github.com/ludios/Desktopmagic/blob/master/desktopmagic/screengrab_win32.py
# im = ImageGrab.grab(bbox=frame)
if not frame:
print("ERROR: Screenshots below 5x5 px not allowed, exiting...")
sleep(5)
return
im = getRectAsImage(frame)
send_to_clipboard(im)
timestamp = strftime("%Y-%m-%d_%H.%M.%S")
tempdir = gettempdir()+"\\Screens"
# print("DEBUG:",tempdir)
if not os.path.exists(tempdir):
os.makedirs(tempdir)
tempsave = '{0}\\{1}.png'.format(tempdir, 'Screen_'+timestamp)
im.save(tempsave)
if settings['MODE'] == "ONLINE":
upload_to_url(tempsave)
else:
save_to_local(tempsave)
clean_temp(tempdir)
# https://stackoverflow.com/questions/34322132/copy-image-to-clipboard-in-python3
def send_to_clipboard(im=None):
if im:
output = BytesIO()
im.convert("RGB").save(output, "BMP")
data = output.getvalue()[14:]
output.close()
win32clipboard.OpenClipboard()
win32clipboard.EmptyClipboard()
win32clipboard.SetClipboardData(win32clipboard.CF_DIB, data)
win32clipboard.CloseClipboard()
print("Copied to Clipboard")
else:
win32clipboard.EmptyClipboard()
print("Cleared Clipboard")
# https://stackoverflow.com/questions/432385/sftp-in-python-platform-independent
def upload_to_url(tempsave):
# SFTP INFO
url = settings['URL']
port = settings['PORT']
timeout = settings['TIMEOUT']
usr = settings['USR']
passw = settings['PASS']
pth = settings['PTH'] + os.path.basename(tempsave)
where = settings['WHERE'] + os.path.basename(tempsave)
# Test Connection first or go into fallback
try:
s = create_connection((url, port), timeout)
s.close()
except Exception as e:
print("ERROR: Connection failed. |||", e)
print("\nFalling back to local save...")
save_to_local(tempsave, fallback=1)
return
# If not provided ask for Username & Password
if not usr:
usr = input("Please provide your Username: ")
if not passw:
passw = getpass("Please provide the password for {0}: ".format(usr))
try:
transport = paramiko.Transport((url, port))
transport.connect(username=usr, password=passw)
except Exception as e:
print("ERROR: SSH2 negotiation or authentication failed. |||", e)
print("\nFalling back to local save...")
save_to_local(tempsave, fallback=1)
return
print("Starting to Upload...")
sftp = paramiko.SFTPClient.from_transport(transport)
sftp.put(tempsave, pth)
sftp.close()
transport.close()
print("Screen was Uploaded: URL: " + where)
open_url(where)
def save_to_local(tempsave, fallback=None):
if not fallback:
where = settings['WHERE']
else:
print(" -> Did fallback!")
config = ConfigParser()
config.read_file(open(root+'_settings.ini'))
where = config.get('OFFLINE', 'WHERE')
if not where:
print("No local Screens folder provided, creating one...")
where = os.path.dirname(os.path.abspath(__file__))+"\\Screens"
print("OFFLINE MODE: Screenshot Folder:", where)
if not os.path.exists(where):
os.makedirs(where)
src = tempsave
dst = where + "\\" + os.path.basename(tempsave)
# print("Source:",src,"\nDestination:",dst)
os.rename(src, dst)
def open_url(where):
webbrowser.open(where, new=0, autoraise=True)
def clean_temp(tempdir):
if os.path.exists(tempdir):
print("Cleaning up the tempdir [", tempdir, "]...")
rmtree(tempdir, ignore_errors=True)
###################################################################################################
###################################################################################################
if __name__ == '__main__':
root = os.path.dirname(os.path.abspath(__file__))+"\\"
settings = read_ini(root)
if os.path.exists(root+"qt.conf"):
try:
os.remove(root+"qt.conf")
except Exception as e:
print("DEBUG:", e)
app = QtGui.QApplication(sys.argv)
windows = []
# https://stackoverflow.com/questions/51058236/create-qt-windows-based-on-number-of-list-items
for i in range(QtGui.QApplication.desktop().screenCount()):
topLeft = QtGui.QApplication.desktop().screenGeometry(i).topLeft()
window = Fillscreen()
window.move(topLeft)
window.showFullscreen()
windows.append(window)
app.exec_()