添加禁用窗口截图的功能

Author: zhiyiYo

README.md

@@ -35,6 +35,7 @@
 * Win11 mica blur
 * Win7 Aero blur
 * MacOS blur
+* Disable screen capture
 
 ## Install
 To install use pip:

examples/screen_capture_filter.py

@@ -0,0 +1,34 @@
+# coding:utf-8
+import sys
+
+from PyQt5.QtCore import Qt
+from PyQt5.QtWidgets import QApplication
+
+from qframelesswindow import FramelessWindow
+from qframelesswindow.utils import ScreenCaptureFilter
+
+
+
+class Window(FramelessWindow):
+
+    def __init__(self, parent=None):
+        super().__init__(parent=parent)
+        self.setWindowTitle("PyQt-Frameless-Window")
+
+        # disable screen capture
+        self.installEventFilter(ScreenCaptureFilter(self))
+
+
+
+if __name__ == "__main__":
+    # enable dpi scale
+    QApplication.setHighDpiScaleFactorRoundingPolicy(
+        Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
+    QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
+    QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps)
+
+    # run app
+    app = QApplication(sys.argv)
+    demo = Window()
+    demo.show()
+    sys.exit(app.exec_())

qframelesswindow/__init__.py

@@ -12,7 +12,7 @@
 :license: GPLv3, see LICENSE for more details.
 """
 
-__version__ = "0.5.1"
+__version__ = "0.6.0"
 __author__ = "zhiyiYo"
 
 import sys

qframelesswindow/utils/__init__.py

@@ -4,12 +4,15 @@
 if sys.platform == "win32":
     from .win32_utils import WindowsMoveResize as MoveResize
     from .win32_utils import getSystemAccentColor
+    from .win32_utils import WindowsScreenCaptureFilter as ScreenCaptureFilter
 elif sys.platform == "darwin":
     from .mac_utils import MacMoveResize as MoveResize
     from .mac_utils import getSystemAccentColor
+    from .mac_utils import MacScreenCaptureFilter as ScreenCaptureFilter
 else:
     from .linux_utils import LinuxMoveResize as MoveResize
     from .linux_utils import getSystemAccentColor
+    from .linux_utils import LinuxScreenCaptureFilter as ScreenCaptureFilter
 
 
 def startSystemMove(window, globalPos):

qframelesswindow/utils/linux_utils.py

@@ -3,7 +3,7 @@
 
 import xcffib as xcb
 from PyQt5 import sip
-from PyQt5.QtCore import QPointF, Qt, QEvent, QPoint
+from PyQt5.QtCore import QPointF, Qt, QEvent, QPoint, QObject
 from PyQt5.QtGui import QMouseEvent, QColor
 from PyQt5.QtWidgets import QWidget, QApplication
 from PyQt5.QtX11Extras import QX11Info
@@ -176,4 +176,23 @@ def getSystemAccentColor():
     color: QColor
         accent color
     """
-    return QColor()
+    return QColor()
+
+
+class LinuxScreenCaptureFilter(QObject):
+    """ Filter for screen capture """
+
+    def __init__(self, parent: QWidget):
+        super().__init__(parent)
+        self.setScreenCaptureEnabled(False)
+
+    def eventFilter(self, watched, event):
+        if watched == self.parent():
+            if event.type() == QEvent.Type.WinIdChange:
+                self.setScreenCaptureEnabled(self.isScreenCaptureEnabled)
+
+        return super().eventFilter(watched, event)
+
+    def setScreenCaptureEnabled(self, enabled: bool):
+        """ Set screen capture enabled """
+        self.isScreenCaptureEnabled = enabled

qframelesswindow/utils/mac_utils.py

@@ -3,7 +3,7 @@
 
 import Cocoa
 import objc
-from PyQt5.QtCore import QT_VERSION_STR
+from PyQt5.QtCore import QT_VERSION_STR, QEvent, QObject
 from PyQt5.QtGui import QColor
 from PyQt5.QtWidgets import QWidget
 from Quartz.CoreGraphics import (CGEventCreateMouseEvent,
@@ -86,4 +86,29 @@ def getSystemAccentColor():
     r = int(color.redComponent() * 255)
     g = int(color.greenComponent() * 255)
     b = int(color.blueComponent() * 255)
-    return QColor(r, g, b)
+    return QColor(r, g, b)
+
+
+class MacScreenCaptureFilter(QObject):
+    """ Filter for screen capture """
+
+    def __init__(self, parent: QWidget):
+        super().__init__(parent)
+        self.setScreenCaptureEnabled(False)
+
+    def eventFilter(self, watched, event):
+        if watched == self.parent():
+            if event.type() == QEvent.Type.WinIdChange:
+                self.setScreenCaptureEnabled(self.isScreenCaptureEnabled)
+
+        return super().eventFilter(watched, event)
+
+    def setScreenCaptureEnabled(self, enabled: bool):
+        """ Set screen capture enabled """
+        self.isScreenCaptureEnabled = enabled
+
+        nsWindow = getNSWindow(self.parent().winId())
+        if nsWindow:
+            NSWindowSharingNone = 0
+            NSWindowSharingReadOnly = 1
+            nsWindow.setSharingType_(NSWindowSharingReadOnly if enabled else NSWindowSharingNone)

qframelesswindow/utils/win32_utils.py

@@ -1,5 +1,5 @@
 # coding:utf-8
-from ctypes import Structure, byref, sizeof, windll, c_int, c_ulong, c_bool, POINTER
+from ctypes import Structure, byref, sizeof, windll, c_int, c_ulong, c_bool, POINTER, WinDLL, wintypes
 from ctypes.wintypes import DWORD, HWND, LPARAM, RECT, UINT
 from platform import platform
 import sys
@@ -9,8 +9,9 @@
 import win32con
 import win32gui
 import win32print
-from PyQt5.QtCore import QOperatingSystemVersion
+from PyQt5.QtCore import QOperatingSystemVersion, QObject, QEvent
 from PyQt5.QtGui import QGuiApplication, QColor
+from PyQt5.QtWidgets import QWidget
 from win32comext.shell import shellcon
 
 
@@ -351,3 +352,31 @@ def starSystemResize(cls, window, globalPos, edges):
             window edges
         """
         pass
+
+
+class WindowsScreenCaptureFilter(QObject):
+    """ Filter for screen capture """
+
+    def __init__(self, parent: QWidget):
+        super().__init__(parent)
+        self.setScreenCaptureEnabled(False)
+
+    def eventFilter(self, watched, event):
+        if watched == self.parent():
+            if event.type() == QEvent.Type.WinIdChange:
+                self.setScreenCaptureEnabled(self.isScreenCaptureEnabled)
+
+        return super().eventFilter(watched, event)
+
+    def setScreenCaptureEnabled(self, enabled: bool):
+        """ Set screen capture enabled """
+        self.isScreenCaptureEnabled = enabled
+        WDA_NONE = 0x00000000
+        WDA_EXCLUDEFROMCAPTURE = 0x00000011
+
+        user32 = WinDLL('user32', use_last_error=True)
+        SetWindowDisplayAffinity = user32.SetWindowDisplayAffinity
+        SetWindowDisplayAffinity.argtypes = (wintypes.HWND, wintypes.DWORD)
+        SetWindowDisplayAffinity.restype = wintypes.BOOL
+
+        SetWindowDisplayAffinity(int(self.parent().winId()), WDA_NONE if enabled else WDA_EXCLUDEFROMCAPTURE)

setup.py

@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name="PyQt5-Frameless-Window",
-    version="0.5.1",
+    version="0.6.0",
     keywords="pyqt frameless",
     author="zhiyiYo",
     author_email="[email protected]",