Skip to content

Instantly share code, notes, and snippets.

@DanEdens

DanEdens/vitrual_desktop_script_windows_outdated

Created February 11, 2025 13:40
Show Gist options
  • Save DanEdens/b9885dfa2fe71afb8adf3582c9c84bd0 to your computer and use it in GitHub Desktop.
Save DanEdens/b9885dfa2fe71afb8adf3582c9c84bd0 to your computer and use it in GitHub Desktop.
Download ZIP
vitrual_desktop_script_windows_outdated
Raw
vitrual_desktop_script_windows_outdated
from ctypes.wintypes import HRESULT, HWND, BOOL, POINTER, DWORD, INT, UINT, LPVOID, ULONG
import comtypes
import ctypes
from comtypes import helpstring, COMMETHOD
from comtypes.GUID import GUID
REFGUID = POINTER(GUID)
REFIID = REFGUID
ENUM = INT
IID = GUID
INT32 = ctypes.c_int32
INT64 = ctypes.c_int64
CLSID_ImmersiveShell = GUID(
'{C2F03A33-21F5-47FA-B4BB-156362A2F239}'
)
CLSID_IVirtualNotificationService = GUID(
'{A501FDEC-4A09-464C-AE4E-1B9C21B84918}'
)
class HSTRING__(ctypes.Structure):
_fields_ = [
('unused', INT),
]
HSTRING = POINTER(HSTRING__)
class EventRegistrationToken(ctypes.Structure):
_fields_ = [
('value', INT64)
]
class AdjacentDesktop(ENUM):
LeftDirection = 3
RightDirection = 4
class ApplicationViewOrientation(ENUM):
ApplicationViewOrientation_Landscape = 0
ApplicationViewOrientation_Portrait = 1
class TrustLevel(ENUM):
BaseTrust = 0
PartialTrust = BaseTrust + 1
FullTrust = PartialTrust + 1
IID_IInspectable = GUID(
'{AF86E2E0-B12D-4C6A-9C5A-D7AA65101E90}'
)
class IInspectable(comtypes.IUnknown):
_case_insensitive_ = True
_idlflags_ = []
_iid_ = IID_IInspectable
_methods_ = [
COMMETHOD(
[helpstring('Method GetIids')],
HRESULT,
'GetIids',
(['out'], POINTER(ULONG), 'iidCount'),
(['out'], POINTER(POINTER(IID)), 'iids'),
),
COMMETHOD(
[helpstring('Method GetRuntimeClassName')],
HRESULT,
'GetRuntimeClassName',
(['out'], POINTER(HSTRING), 'className'),
),
COMMETHOD(
[helpstring('Method GetTrustLevel')],
HRESULT,
'GetTrustLevel',
(['out'], POINTER(TrustLevel), 'trustLevel'),
),
]
IID_IApplicationViewConsolidatedEventArgs = GUID(
'{514449EC-7EA2-4DE7-A6A6-7DFBAAEBB6FB}'
)
class IApplicationViewConsolidatedEventArgs(IInspectable):
_case_insensitive_ = True
_iid_ = IID_IApplicationViewConsolidatedEventArgs
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method get_IsUserInitiated')],
HRESULT,
'get_IsUserInitiated',
(['retval', 'out'], POINTER(BOOL), 'value'),
),
]
IID_IApplicationView = GUID(
'{D222D519-4361-451E-96C4-60F4F9742DB0}'
)
class IApplicationView(IInspectable):
_case_insensitive_ = True
_iid_ = IID_IApplicationView
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method get_Orientation')],
HRESULT,
'get_Orientation',
(['retval', 'out'], POINTER(ApplicationViewOrientation), 'value'),
),
COMMETHOD(
[helpstring('Method get_AdjacentToLeftDisplayEdge')],
HRESULT,
'get_AdjacentToLeftDisplayEdge',
(['retval', 'out'], POINTER(BOOL), 'value'),
),
COMMETHOD(
[helpstring('Method get_AdjacentToRightDisplayEdge')],
HRESULT,
'get_AdjacentToRightDisplayEdge',
(['retval', 'out'], POINTER(BOOL), 'value'),
),
COMMETHOD(
[helpstring('Method get_IsFullScreen')],
HRESULT,
'get_IsFullScreen',
(['retval', 'out'], POINTER(BOOL), 'value'),
),
COMMETHOD(
[helpstring('Method get_IsOnLockScreen')],
HRESULT,
'get_IsOnLockScreen',
(['retval', 'out'], POINTER(BOOL), 'value'),
),
COMMETHOD(
[helpstring('Method get_IsScreenCaptureEnabled')],
HRESULT,
'get_IsScreenCaptureEnabled',
(['retval', 'out'], POINTER(BOOL), 'value'),
),
COMMETHOD(
[helpstring('Method put_IsScreenCaptureEnabled')],
HRESULT,
'put_IsScreenCaptureEnabled',
(['in'], BOOL, 'value'),
),
COMMETHOD(
[helpstring('Method put_Title')],
HRESULT,
'put_Title',
(['in'], HSTRING, 'value'),
),
COMMETHOD(
[helpstring('Method get_Title')],
HRESULT,
'get_Title',
(['retval', 'out'], POINTER(HSTRING), 'value'),
),
COMMETHOD(
[helpstring('Method get_Id')],
HRESULT,
'get_Id',
(['retval', 'out'], POINTER(INT32), 'value'),
),
COMMETHOD(
[helpstring('Method add_Consolidated')],
HRESULT,
'add_Consolidated',
(['in'], POINTER(IApplicationViewConsolidatedEventArgs), 'handler'),
(['retval', 'out'], POINTER(EventRegistrationToken), 'token'),
),
COMMETHOD(
[helpstring('Method remove_Consolidated')],
HRESULT,
'remove_Consolidated',
(['in', ], EventRegistrationToken, 'EventRegistrationToken'),
),
]
IID_IServiceProvider = GUID(
'{6D5140C1-7436-11CE-8034-00AA006009FA}'
)
class IServiceProvider(comtypes.IUnknown):
_case_insensitive_ = True
_idlflags_ = []
_iid_ = IID_IServiceProvider
_methods_ = [
COMMETHOD(
[helpstring('Method QueryService'), 'local', 'in'],
HRESULT,
'QueryService',
(['in'], REFGUID, 'guidService'),
(['in'], REFIID, 'riid'),
(['out'], POINTER(LPVOID), 'ppvObject'),
),
]
IID_IObjectArray = GUID(
"{92CA9DCD-5622-4BBA-A805-5E9F541BD8C9}"
)
class IObjectArray(comtypes.IUnknown):
"""
Unknown Object Array
"""
_case_insensitive_ = True
_idlflags_ = []
_iid_ = None
_methods_ = [
COMMETHOD(
[helpstring('Method GetCount')],
HRESULT,
'GetCount',
(['out'], POINTER(UINT), 'pcObjects'),
),
COMMETHOD(
[helpstring('Method GetAt')],
HRESULT,
'GetAt',
(['in'], UINT, 'uiIndex'),
(['in'], REFIID, 'riid'),
(['out', 'iid_is'], POINTER(LPVOID), 'ppv'),
),
]
IID_IVirtualDesktop = GUID(
'{FF72FFDD-BE7E-43FC-9C03-AD81681E88E4}'
)
class IVirtualDesktop(comtypes.IUnknown):
_case_insensitive_ = True
_iid_ = IID_IVirtualDesktop
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method IsViewVisible')],
HRESULT,
'IsViewVisible',
(['out'], POINTER(IApplicationView), 'pView'),
(['out'], POINTER(INT), 'pfVisible'),
),
COMMETHOD(
[helpstring('Method GetID')],
HRESULT,
'GetID',
(['out'], POINTER(GUID), 'pGuid'),
)
]
IID_IVirtualDesktopManager = GUID(
'{A5CD92FF-29BE-454C-8D04-D82879FB3F1B}'
)
class IVirtualDesktopManager(comtypes.IUnknown):
_case_insensitive_ = True
_iid_ = IID_IVirtualDesktopManager
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method IsWindowOnCurrentVirtualDesktop')],
HRESULT,
'IsWindowOnCurrentVirtualDesktop',
(['in'], HWND, 'topLevelWindow'),
(['out'], POINTER(BOOL), 'onCurrentDesktop'),
),
COMMETHOD(
[helpstring('Method GetWindowDesktopId')],
HRESULT,
'GetWindowDesktopId',
(['in'], HWND, 'topLevelWindow'),
(['out'], POINTER(GUID), 'desktopId'),
),
COMMETHOD(
[helpstring('Method MoveWindowToDesktop')],
HRESULT,
'MoveWindowToDesktop',
(['in'], HWND, 'topLevelWindow'),
(['in'], REFGUID, 'desktopId'),
),
]
CLSID_VirtualDesktopManagerInternal = GUID(
'{C5E0CDCA-7B6E-41B2-9FC4-D93975CC467B}'
)
IID_IVirtualDesktopManagerInternal = GUID(
'{F31574D6-B682-4CDC-BD56-1827860ABEC6}'
)
# IID_IVirtualDesktopManagerInternal = GUID(
# '{AF8DA486-95BB-4460-B3B7-6E7A6B2962B5}'
# )
# IID_IVirtualDesktopManagerInternal = GUID(
# '{EF9F1A6C-D3CC-4358-B712-F84B635BEBE7}'
# )
class IVirtualDesktopManagerInternal(comtypes.IUnknown):
_case_insensitive_ = True
_iid_ = IID_IVirtualDesktopManagerInternal
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method GetCount')],
HRESULT,
'GetCount',
(['out'], POINTER(UINT), 'pCount'),
),
COMMETHOD(
[helpstring('Method MoveViewToDesktop')],
HRESULT,
'MoveViewToDesktop',
(['out'], POINTER(IApplicationView), 'pView'),
(['out'], POINTER(IVirtualDesktop), 'pDesktop'),
),
COMMETHOD(
[helpstring('Method CanViewMoveDesktops')],
HRESULT,
'CanViewMoveDesktops',
(['out'], POINTER(IApplicationView), 'pView'),
(['out'], POINTER(INT), 'pfCanViewMoveDesktops'),
),
COMMETHOD(
[helpstring('Method GetCurrentDesktop')],
HRESULT,
'GetCurrentDesktop',
(['out'], POINTER(POINTER(IVirtualDesktop)), 'desktop'),
),
COMMETHOD(
[helpstring('Method GetDesktops')],
HRESULT,
'GetDesktops',
(['out'], POINTER(POINTER(IObjectArray)), 'ppDesktops'),
),
COMMETHOD(
[helpstring('Method GetAdjacentDesktop')],
HRESULT,
'GetAdjacentDesktop',
(['out'], POINTER(IVirtualDesktop), 'pDesktopReference'),
(['in'], AdjacentDesktop, 'uDirection'),
(['out'], POINTER(POINTER(IVirtualDesktop)), 'ppAdjacentDesktop'),
),
COMMETHOD(
[helpstring('Method SwitchDesktop')],
HRESULT,
'SwitchDesktop',
(['in'], POINTER(IVirtualDesktop), 'pDesktop'),
),
COMMETHOD(
[helpstring('Method CreateDesktopW')],
HRESULT,
'CreateDesktopW',
(['out'], POINTER(POINTER(IVirtualDesktop)), 'ppNewDesktop'),
),
COMMETHOD(
[helpstring('Method RemoveDesktop')],
HRESULT,
'RemoveDesktop',
(['in'], POINTER(IVirtualDesktop), 'pRemove'),
(['in'], POINTER(IVirtualDesktop), 'pFallbackDesktop'),
),
COMMETHOD(
[helpstring('Method FindDesktop')],
HRESULT,
'FindDesktop',
(['in'], POINTER(GUID), 'desktopId'),
(['out'], POINTER(POINTER(IVirtualDesktop)), 'ppDesktop'),
),
]
IID_IVirtualDesktopNotification = GUID(
'{C179334C-4295-40D3-BEA1-C654D965605A}'
)
class IVirtualDesktopNotification(comtypes.IUnknown):
_case_insensitive_ = True
_iid_ = IID_IVirtualDesktopNotification
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method VirtualDesktopCreated')],
HRESULT,
'VirtualDesktopCreated',
(['in'], POINTER(IVirtualDesktop), 'pDesktop'),
),
COMMETHOD(
[helpstring('Method VirtualDesktopDestroyBegin')],
HRESULT,
'VirtualDesktopDestroyBegin',
(['in'], POINTER(IVirtualDesktop), 'pDesktopDestroyed'),
(['in'], POINTER(IVirtualDesktop), 'pDesktopFallback'),
),
COMMETHOD(
[helpstring('Method VirtualDesktopDestroyFailed')],
HRESULT,
'VirtualDesktopDestroyFailed',
(['in'], POINTER(IVirtualDesktop), 'pDesktopDestroyed'),
(['in'], POINTER(IVirtualDesktop), 'pDesktopFallback'),
),
COMMETHOD(
[helpstring('Method VirtualDesktopDestroyed')],
HRESULT,
'VirtualDesktopDestroyed',
(['in'], POINTER(IVirtualDesktop), 'pDesktopDestroyed'),
(['in'], POINTER(IVirtualDesktop), 'pDesktopFallback'),
),
COMMETHOD(
[helpstring('Method ViewVirtualDesktopChanged')],
HRESULT,
'ViewVirtualDesktopChanged',
(['in'], POINTER(IApplicationView), 'pView'),
),
COMMETHOD(
[helpstring('Method CurrentVirtualDesktopChanged')],
HRESULT,
'CurrentVirtualDesktopChanged',
(['in'], POINTER(IVirtualDesktop), 'pDesktopOld'),
(['in'], POINTER(IVirtualDesktop), 'pDesktopNew'),
),
]
IID_IVirtualDesktopNotificationService = GUID('{0CD45E71-D927-4F15-8B0A-8FEF525337BF}')
class IVirtualDesktopNotificationService(comtypes.IUnknown):
_case_insensitive_ = True
_iid_ = IID_IVirtualDesktopNotificationService
_idlflags_ = []
_methods_ = [
COMMETHOD(
[helpstring('Method Register')],
HRESULT,
'Register',
(['in'], POINTER(IVirtualDesktopNotification), 'pNotification'),
(['out'], POINTER(DWORD), 'pdwCookie'),
),
COMMETHOD(
[helpstring('Method Unregister')],
HRESULT,
'Unregister',
(['in'], DWORD, 'dwCookie'),
),
]
comtypes.CoInitialize()
pServiceProvider = comtypes.CoCreateInstance(
CLSID_ImmersiveShell,
IServiceProvider,
comtypes.CLSCTX_LOCAL_SERVER,
)
pDesktopManagerInternal = comtypes.cast(
pServiceProvider.QueryService(
CLSID_VirtualDesktopManagerInternal,
IID_IVirtualDesktopManagerInternal
),
ctypes.POINTER(IVirtualDesktopManagerInternal)
)
pObjectArray = POINTER(IObjectArray)()
pDesktopManagerInternal.GetDesktops(ctypes.byref(pObjectArray))
count = UINT()
pObjectArray.GetCount(ctypes.byref(count))
for i in range(count):
pDesktop = POINTER(IVirtualDesktop)()
pObjectArray.GetAt(i, IID_IVirtualDesktop, ctypes.byref(pDesktop))
id = GUID()
pDesktop.GetID(ctypes.byref(id))
print(id)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment