add windows screenshot

This commit is contained in:
Qijun Gu 2021-10-11 09:55:57 -05:00
parent 961b9f434f
commit 7b48ba5a6b
2 changed files with 15 additions and 8 deletions

View File

@ -26,7 +26,7 @@ This tool can run on Linux, Windows and MAC.
## Install ## Install
1. pip install flask-bootstrap pyscreenshot pyopenssl 1. pip install flask-bootstrap pyscreenshot pyopenssl pillow
2. In a directory, run "**git clone https://github.com/qijungu/screenshare.git**". You will have a new directory "screenshare" with code inside. 2. In a directory, run "**git clone https://github.com/qijungu/screenshare.git**". You will have a new directory "screenshare" with code inside.

View File

@ -15,24 +15,28 @@
#You should have received a copy of the GNU General Public License #You should have received a copy of the GNU General Public License
#along with Screenshare. If not, see <https://www.gnu.org/licenses/>. #along with Screenshare. If not, see <https://www.gnu.org/licenses/>.
import threading, time, base64 import threading, time, base64, sys
import pyscreenshot as ig
import sys
ver = sys.version_info.major ver = sys.version_info.major
if ver==2: if ver==2:
import StringIO import StringIO as io
elif ver==3: elif ver==3:
import io import io
if sys.platform in ["win32", "darwin"]:
from PIL import ImageGrab as ig
else:
import pyscreenshot as ig
bkend = "pygdk3"
class Screen(): class Screen():
def __init__(self): def __init__(self):
self.FPS = 10 self.FPS = 10
self.screenbuf = "" self.screenbuf = ""
self.password = "" self.password = ""
if ver==2: if ver==2:
self.screenfile = StringIO.StringIO() self.screenfile = io.StringIO()
elif ver==3: elif ver==3:
self.screenfile = io.BytesIO() self.screenfile = io.BytesIO()
threading.Thread(target=self.getframes).start() threading.Thread(target=self.getframes).start()
@ -42,7 +46,10 @@ class Screen():
def getframes(self): def getframes(self):
while True: while True:
im = ig.grab(childprocess=False,backend="pygdk3") if sys.platform in ["win32", "darwin"]:
im = ig.grab()
else:
im = ig.grab(childprocess=False,backend=bkend)
self.screenfile.seek(0) self.screenfile.seek(0)
self.screenfile.truncate(0) self.screenfile.truncate(0)
im.save(self.screenfile, format="jpeg", quality=75, progressive=True) im.save(self.screenfile, format="jpeg", quality=75, progressive=True)