fileAssocUtil.py

# file assoc util
# Copyright (C) 2020 Hiroki Fujii <hfujii@hisystron.com>

import wx, itertools, sys, winreg, logging
import globalVars, shellapi, constants

def setAssoc(extension, associate=None, mimetype=None):
	if not hasattr(sys,"frozen"): return False
	if extension[0] != ".": extension = "." + extension
	if associate == None: associate = "actlab.%s" % constants.APP_NAME
	return _registerFileAssociation(extension, sys.executable, associate, mimetype)

def unsetAssoc(associate=None):
	if associate == None: associate = "actlab.%s" % constants.APP_NAME
	return _unregisterFileAssociation(associate)

def _registerFileAssociation(extension, exePath, associate, mimetype):
	try:
		with winreg.CreateKeyEx(winreg.HKEY_CURRENT_USER, "SOFTWARE\\Classes\\%s" % associate, 0, winreg.KEY_WRITE) as k:
			#winreg.SetValueEx(k, None, 0, winreg.REG_SZ, "")
			with winreg.CreateKeyEx(k, "DefaultIcon", 0, winreg.KEY_WRITE) as k2:
				winreg.SetValueEx(k2, None, 0, winreg.REG_SZ, "@{exePath},1".format(exePath=exePath))
			with winreg.CreateKeyEx(k, "shell\\open\\command", 0, winreg.KEY_WRITE) as k2:
				winreg.SetValueEx(k2, None, 0, winreg.REG_SZ, u"\"{exePath}\" \"%1\"".format(exePath=exePath))
		with winreg.CreateKeyEx(winreg.HKEY_CURRENT_USER, "SOFTWARE\\Classes\\%s" % extension, 0, winreg.KEY_WRITE) as k:
			winreg.SetValueEx(k, None, 0, winreg.REG_SZ, associate)
			if mimetype != None: winreg.SetValueEx(k, "Content Type", 0, winreg.REG_SZ, mimetype)
			k2 = winreg.CreateKeyEx(k, "OpenWithProgids\\%s" % associate, 0, winreg.KEY_WRITE)
			winreg.CloseKey(k2)
		shellapi.SHChangeNotify(shellapi.SHCNE_ASSOCCHANGED, shellapi.SHCNF_IDLIST, None, None)
		return True
	except WindowsError as e:
		log = logging.getLogger("%s.fileAssoc" % (constants.LOG_PREFIX))
		log.error("file Association faild - %s" %(str(e)))
		return False

def _unregisterFileAssociation(associate):
	try:
		_deleteKeyAndSubkeys(winreg.HKEY_CURRENT_USER, "Software\\Classes\\%s" % associate)
	except WindowsError as e:
		log = logging.getLogger("%s.fileAssoc" % (constants.LOG_PREFIX))
		log.error("Unset File Association faild - %s" %str(e))
		return False
	shellapi.SHChangeNotify(shellapi.SHCNE_ASSOCCHANGED, shellapi.SHCNF_IDLIST, None, None)
	return True

def _deleteKeyAndSubkeys(key, subkey):
	isFinished = False
	tryCount = 0
	while isFinished == False and tryCount <= 10:
		try:
			with winreg.OpenKey(key, subkey, 0, winreg.KEY_WRITE|winreg.KEY_READ) as k:
				for i in itertools.count():
					try:
						subkeyName = winreg.EnumKey(k, i)
					except WindowsError as e:
						break
					_deleteKeyAndSubkeys(k, subkeyName)
				winreg.DeleteKey(k, "")
			isFinished = True
		except: tryCount += 1