windows 覆盖图标实现角标 - firemail

ForceRemove ' MyOverlayIcon' = s '{81539FE6-33C7-4CE7-90C7-1C7B8F2F2D40}'

复制代码

STDMETHODIMP CShellExt::GetOverlayInfo(LPWSTR pwszIconFile, int cchMax, int* pIndex, DWORD* pdwFlags)
{
if (!pwszIconFile)
return E_POINTER;
if (!pIndex)
return E_POINTER;
if (!pdwFlags)
return E_POINTER;
if(cchMax < 1)
return E_INVALIDARG;
// Set "out parameters" since we return S_OK later.
*pwszIconFile = 0;
*pIndex = 0;
*pdwFlags = 0;
// Now here's where we can find out if due to lack of enough overlay
// slots some of our overlays won't be shown.
// To do that we have to mark every overlay handler that's successfully
// loaded, so we can later check if some are missing
switch (m_State)
{
case FileStateVersioned : g_normalovlloaded = true; break;
case FileStateModified : g_modifiedovlloaded = true; break;
case FileStateConflict : g_conflictedovlloaded = true; break;
case FileStateDeleted : g_deletedovlloaded = true; break;
case FileStateReadOnly : g_readonlyovlloaded = true; break;
case FileStateLockedOverlay : g_lockedovlloaded = true; break;
case FileStateAddedOverlay : g_addedovlloaded = true; break;
case FileStateIgnoredOverlay : g_ignoredovlloaded = true; break;
case FileStateUnversionedOverlay : g_unversionedovlloaded = true; break;
}
// we don't have to set the icon file and/or the index here:
// the icons are handled by the TortoiseOverlays dll.
return S_OK;
};

复制代码

The binary files in this directory are directly taken from the TortoiseSVN project.
The source code and the original binary files are available in the TortoiseSVN Subversion repository:
SourceForge Web-SVN-Browser: http://sourceforge.net/p/tortoisesvn/code/HEAD/tree/trunk/src/TortoiseOverlays/
Official binaries: http://sourceforge.net/p/tortoisesvn/code/HEAD/tree/TortoiseOverlays/

复制代码

import os
import winreg
import sqlite3
import winerror
import win32con
// Pay attention to the import path here, the path of the old version of pywin32 is different
from win32comext.shell import shell, shellcon
class IconOverlayOk:
"""
Icon overlay component
The IShellIconOverlayIdentifier interface under the windows shell is used to implement the icon overlay function
The following steps are required to implement functions in python
1. Use pywin32 to write python class to implement windows com components
2. Register the component to the windows system
3. Add the icon overlay component item in the registry
4. Restart explorer.exe
When the windows file explorer browses the file
1. Explorer will call the IsMemberOf method of IShellIconOverlayIdentifier for each file currently to be displayed to determine whether the file needs to be processed
2. If IsMemberOf returns winerror.S_OK, the system will continue to call the GetOverlayInfo method to obtain the icon file to be superimposed
3. The operating system is responsible for superimposing the icon returned by GetOverlayInfo onto the original icon of the file
_reg_remove_keys_: When uninstalling the component, the registry keys that need to be deleted at the same time. Here, add a space prefix to the last registry key name to ensure the priority of the icon overlay component and avoid other icon overlay programs such as oneDriver, Nut Cloud Unsuccessful competition causes the icon overlay to fail
"""
_reg_clsid_ = '{EA258179-A91D-45F9-A237-FC92A5290423}'
_reg_progid_ = 'SUN.PythonPackagesOverlayHandlerOK'
_reg_desc_ = 'Icon Overlay Handler to indicate Python packages'
_public_methods_ = ['GetOverlayInfo', 'GetPriority', 'IsMemberOf']
_com_interfaces_ = [shell.IID_IShellIconOverlayIdentifier]
_reg_remove_keys_ = [(r'Software\Microsoft\Windows\CurrentVersion\Explorer\ShellIconOverlayIdentifiers\ PyPackageOverlay1',
win32con.HKEY_LOCAL_MACHINE)]
def GetOverlayInfo(self):
"""
Get the location of the icon file to be superimposed
Obtain the icon file path iconPath from the registry
:return:
"""
key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r'Software\Dachengyun')
iconPath, index = winreg.QueryValueEx(key, 'iconPath')
key.Close()
return os.path.join(iconPath, 'Ok.ico'), 0, shellcon.ISIOI_ICONFILE
def GetPriority(self):
return 1
def IsMemberOf(self, fname, attributes):
"""
Determine whether to overwrite the file icon
Read the synchronization folder path syncPath, sqlite database file path dbPath from the registry
If the current file is under syncPath, go to db to query the file synchronization status
Determine whether icon overlay is needed according to the file synchronization status
:param fname:
:param attributes:
:return:
"""
key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r'Software\Dachengyun')
syncPath, index = winreg.QueryValueEx(key, 'syncPath')
dbPath, index = winreg.QueryValueEx(key, 'dbPath')
key.Close()
if fname.startswith(syncPath):
filePath = fname[len(os.path.dirname(syncPath)):]
conn = sqlite3.connect(dbPath)
c = conn.cursor()
c.execute('SELECT state FROM file_sync_state WHERE path=:filePath', {'filePath': filePath})
r = c.fetchone()
c.close()
conn.close()
if r is not None and r[0] == 1:
return winerror.S_OK
return winerror.E_FAIL
class IconOverlaySync:
_reg_clsid_ = '{EA258179-A92D-45F9-A237-FC92A5290423}'
_reg_progid_ = 'SUN.PythonPackagesOverlayHandlerSync'
_reg_desc_ = 'Icon Overlay Handler to indicate Python packages'
_public_methods_ = ['GetOverlayInfo', 'GetPriority', 'IsMemberOf']
_com_interfaces_ = [shell.IID_IShellIconOverlayIdentifier]
_reg_remove_keys_ = [(r'Software\Microsoft\Windows\CurrentVersion\Explorer\ShellIconOverlayIdentifiers\ PyPackageOverlay2',
win32con.HKEY_LOCAL_MACHINE)]
def GetOverlayInfo(self):
key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r'Software\Dachengyun')
iconPath, index = winreg.QueryValueEx(key, 'iconPath')
key.Close()
return os.path.join(iconPath, 'Sync.ico'), 0, shellcon.ISIOI_ICONFILE
def GetPriority(self):
return 2
def IsMemberOf(self, fname, attributes):
key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r'Software\Dachengyun')
syncPath, index = winreg.QueryValueEx(key, 'syncPath')
dbPath, index = winreg.QueryValueEx(key, 'dbPath')
key.Close()
if fname.startswith(syncPath):
filePath = fname[len(os.path.dirname(syncPath)):]
conn = sqlite3.connect(dbPath)
c = conn.cursor()
c.execute('SELECT state FROM file_sync_state WHERE path=:filePath',
{'filePath': filePath})
r = c.fetchone()
c.close()
conn.close()
if r is not None and r[0] == 2:
return winerror.S_OK
return winerror.E_FAIL

复制代码

// Pay attention to the introduction here, the introduction path of the old version of pywin32 seems to be different, which is prone to errors
from win32comext.shell import shell, shellcon
// syncPath is the absolute path of the file to be refreshed
shell.SHChangeNotify(shellcon.SHCNE_ATTRIBUTES,
shellcon.SHCNF_PATH | shellcon.SHCNF_FLUSHNOWAIT,
bytes(syncPath, 'gbk'), None)

复制代码

import os
import win32api
import win32con
from win32com.server import register
import iconOverlay
// Multiple icons need to copy the following code, and register multiple classes after modifying the class name
register.UseCommandLine(iconOverlay.IconOverlayOk)
win32api.RegSetValue(win32api.RegCreateKey(iconOverlay.IconOverlayOk._reg_remove_keys_[0][1],iconOverlay.IconOverlayOk._reg_remove_keys_[0][0]),
None, win32con.REG_SZ,
iconOverlay.IconOverlayOk._reg_clsid_)
register.UseCommandLine(iconOverlay.IconOverlaySync)
win32api.RegSetValue(win32api.RegCreateKey(iconOverlay.IconOverlaySync._reg_remove_keys_[0][1],iconOverlay.IconOverlaySync._reg_remove_keys_[0][0]),
None, win32con.REG_SZ,
iconOverlay.IconOverlaySync._reg_clsid_)
// Explorer needs to be restarted after component registration is complete.exe makes the component effective
os.system('taskkill /f /im explorer.exe & start explorer.exe')

复制代码

import win32com.server.register
win32com.server.register.UnregisterClasses(iconOverlay.IconOverlayOk)
win32com.server.register.UnregisterClasses(iconOverlay.IconOverlaySync)
os.system('taskkill /f /im explorer.exe & start explorer.exe')

复制代码