screenshare/screen.py
2021-10-30 14:23:49 -05:00

69 lines
2.0 KiB
Python
Executable File

#Copyright (C) 2021 Qijun Gu
#
#This file is part of Screenshare.
#
#Screenshare is free software: you can redistribute it and/or modify
#it under the terms of the GNU General Public License as published by
#the Free Software Foundation, either version 3 of the License, or
#(at your option) any later version.
#
#Screenshare is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#GNU General Public License for more details.
#
#You should have received a copy of the GNU General Public License
#along with Screenshare. If not, see <https://www.gnu.org/licenses/>.
import threading, time, base64, sys
ver = sys.version_info.major
if ver==2:
import StringIO as io
elif ver==3:
import io
if sys.platform in ["win32", "darwin"]:
from PIL import ImageGrab as ig
else:
import pyscreenshot as ig
bkend = "pygdk3"
class Screen():
def __init__(self):
self.FPS = 10
self.screenbuf = ""
self.password = ""
if ver==2:
self.screenfile = io.StringIO()
elif ver==3:
self.screenfile = io.BytesIO()
threading.Thread(target=self.getframes).start()
def __del__(self):
self.screenfile.close()
def getframes(self):
while True:
if sys.platform in ["win32", "darwin"]:
im = ig.grab()
else:
im = ig.grab(childprocess=False,backend=bkend)
self.screenfile.seek(0)
self.screenfile.truncate(0)
im_converted = im.convert("RGB")
im_converted.save(self.screenfile, format="jpeg", quality=75, progressive=True)
self.screenbuf = base64.b64encode(self.screenfile.getvalue())
time.sleep(1.0/self.FPS)
def gen(self):
s = ''
if ver==2:
s = self.screenbuf
elif ver==3:
s = self.screenbuf.decode()
return s
screenlive = Screen()