init

screen.py

@@ -0,0 +1,60 @@
+#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
+
+from PIL import ImageGrab as ig
+
+class Screen():
+    def __init__(self):
+        self.FPS = 30
+        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:
+            im = ig.grab(bbox=(0, 0, 1920, 1080), include_layered_windows=False, all_screens=False, xdisplay=":0")
+            self.screenfile.seek(0)
+            self.screenfile.truncate(0)
+            im_converted = im.convert("RGB")
+            im_converted.save(self.screenfile, format="jpeg", quality=60, 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()