def main(argv):
assert not argv
if not isUserAdmin():
printf("Current user is not an admin!")
runAsAdmin("cmd /k cd /d %s" % str(gs.GitDir / "bin"),
wait=False,
show=True)
def setupPython():
folder = AppVersionDict["python"]
printf("Setup %s..." % folder)
newPathList = []
# Scan the user path
userPath = regQuery(r"HKCU\Environment", "Path")[0]
userPathList = [p.strip() for p in userPath.split(';')]
foundPython3 = False
for p in userPathList:
testPath = Path(p)/"python.exe"
if testPath.is_file():
ret = call([str(testPath), "--version"])
if ret[2] == 0 and ret[0].split()[1].startswith("3."):
foundPython3 = True
break
# Scan the sys path
sysPath = regQuery(r"HKLM\SYSTEM\CurrentControlSet\Control\Session Manager\Environment", "Path")[0]
sysPathList = [p.strip() for p in sysPath.split(';')]
for p in sysPathList:
testPath = Path(p)/"python.exe"
if testPath.is_file():
ret = call([str(testPath), "--version"])
if ret[2] == 0 and ret[0].split()[1].startswith("3."):
foundPython3 = True
break
if not foundPython3:
newPathList.append(str(gs.GitDir/"app"/folder))
# Add the bin path too
binDir = str(gs.GitDir/"bin")
if binDir not in userPathList + sysPathList:
newPathList.append(binDir)
if newPathList:
regAdd(r"HKCU\Environment", "Path", ";".join(newPathList + userPathList))
notifyEnvChange()
def backup():
printf("Backup the notepad++ settings")
srcDir = Path(os.environ["appdata"])/"Notepad++"
dstDir = gs.GitDir/"backup/notepad++"
if srcDir.is_dir():
copyPath(srcDir, dstDir)
removePath(gs.GitDir/"backup/Notepad++/session.xml") # No need to back up session
printf("Backup the ConEmu settings")
folder = AppVersionDict["conemu"]
srcPath = gs.GitDir/"app"/folder/"ConEmu.xml"
dstDir = gs.GitDir/"backup/conemu"
if srcPath.is_file():
copyPath(srcPath, dstDir)
printf("Backup the Sumatra PDF settings")
folder = AppVersionDict["sumatra"]
srcPath = gs.GitDir/"app"/folder/"SumatraPDF-settings.txt"
dstDir = gs.GitDir/"backup/sumatra"
if srcPath.is_file():
copyPath(srcPath, dstDir)
printf("Backup the Listary settings")
srcPath = Path(os.environ["appdata"])/"Listary/UserData/Preferences.json"
dstPath = gs.GitDir/"backup/Listary/UserData/Preferences.json"
if srcPath.is_file():
copyPath(srcPath, dstPath)
printf("Backup done!")
def setupNpp():
folder = AppVersionDict["notepad++"]
printf("Setup %s" % folder)
appDir = gs.GitDir/"app"/folder
nppPath = appDir/"notepad++.exe"
clsid = "B298D29A-A6ED-11DE-BA8C-A68E55D89593"
regAdd(r"HKCR\CLSID\{%s}" % clsid, '@', "ANotepad++64")
regAddList(r"HKCR\CLSID\{%s}\InprocServer32" % clsid, [
('@', str(appDir/"NppShell_06.dll")),
("ThreadingModel", "Apartment")])
regAddList(r"HKCR\CLSID\{%s}\Settings" % clsid, [
("Title", "Edit with &Notepad++"),
("Path", str(nppPath)),
("Custom", ''),
("ShowIcon", 1),
("Dynamic", 1),
("Maxtext", 25)])
regAdd(r"HKCR\*\shellex\ContextMenuHandlers\ANotepad++64", '@', "{%s}" % clsid)
# Setup txt, log
setOpenWith("txt", str(nppPath), icon=1, regKeyName="Notepad++_file")
setOpenWith("log", str(nppPath), icon=1, regKeyName="Notepad++_file")
srcDir = gs.GitDir/"backup/Notepad++"
dstDir = Path(os.environ["appdata"])/"Notepad++"
if srcDir.is_dir():
copyPath(srcDir, dstDir)
def setupAutoHotKey():
folder = AppVersionDict["ahk"]
printf("Setup %s..." % folder)
exePath = gs.GitDir/"app"/folder/"AutoHotkeyU64.exe"
setOpenWith("ahk", str(exePath), icon=1, regKeyName="AutoHotkeyScript")
startupCmd = 'cmd /c "%s"' % str(gs.GitDir/"app"/folder/"runAhk.bat")
regAdd(r"HKCU\SOFTWARE\Microsoft\Windows\CurrentVersion\Run",
"AutoHotKeyScript", startupCmd)
call(startupCmd)
def setup7z():
# Setup file extension with 7z
folder = AppVersionDict["7z"]
printf("Setup %s..." % folder)
clsid = "23170F69-40C1-278A-1000-000100020000" # The clsid may be changed with the version
iconDict = {
"001": 9,
"7z": 0,
"arj": 4,
"bz2": 2,
"bzip2": 2,
"cpio": 12,
"deb": 11,
"dmg": 17,
"fat": 21,
"gz": 14,
"gzip": 14,
"hfs": 18,
"lha": 6,
"lzh": 6,
"lzma": 16,
"ntfs": 22,
"rar": 3,
"rpm": 10,
"squashfs": 24,
"swm": 15,
"tar": 13,
"taz": 5,
"tbz": 2,
"tbz2": 2,
"tgz": 14,
"tpz": 14,
"txz": 23,
"wim": 15,
"xar": 19,
"xz": 23,
"z": 5,
"zip": 1
}
appDir = gs.GitDir/"app"/folder
for ext in iconDict:
setOpenWith(ext, str(appDir/"7zFM.exe"),
icon="%s,%d" % (str(appDir/"7z.dll"), iconDict[ext]),
regKeyName="7-Zip.%s" % ext)
# Setup the context menu
regAdd(r"HKCR\CLSID\{%s}" % clsid, '@', "7-Zip Shell Extension")
regAdd(r"HKCR\CLSID\{%s}\InprocServer32" % clsid, '@', str(appDir/"7-zip.dll"))
regAdd(r"HKCR\CLSID\{%s}\InprocServer32" % clsid, "ThreadingModel", "Apartment")
regAdd(r"HKCR\*\shellex\ContextMenuHandlers\7-Zip", '@', "{%s}" % clsid)
regAdd(r"HKCR\Folder\shellex\ContextMenuHandlers\7-Zip", '@', "{%s}" % clsid)
regAdd(r"HKCR\Directory\shellex\ContextMenuHandlers\7-Zip", '@', "{%s}" % clsid)
regAdd(r"HKCR\Directory\shellex\DragDropHandlers\7-Zip", '@', "{%s}" % clsid)
regAdd(r"HKCR\Drive\shellex\DragDropHandlers\7-Zip", '@', "{%s}" % clsid)
regAdd(r"HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\Shell Extensions\Approved")
def setupGit():
printf("Setup git... (involves persional info)")
nppFolder = AppVersionDict["notepad++"]
nppPath = gs.GitDir/"app"/nppFolder/"notepad++.exe"
editorCmd = "'%s' -multiInst -notabbar -nosession -noPlugin" % str(nppPath)
cmd = 'git config --global core.editor "%s"' % editorCmd
call(cmd)
cmd = 'git config --global user.email [email protected]'
call(cmd)
cmd = 'git config --global user.name "Yuhe Wang"'
call(cmd)
def setupConEmu():
folder = AppVersionDict["conemu"]
printf("Setup %s..." % folder)
# Copy the config
srcDir = gs.GitDir/"backup/conemu"
dstDir = gs.GitDir/"app"/folder
if srcDir.is_dir():
copyPath(srcDir, dstDir)
# Add cmd context menu
conEmuExe = str(dstDir/"ConEmu64.exe")
contextCmd = '"%s" -Single -run cmd.exe /s /k pushd "%%V"' % conEmuExe
addContextMenu("cmdhere", "Open CMD here (&Q)", contextCmd, icon="cmd.exe")
def setupSumatraPdf():
folder = AppVersionDict["sumatra"]
printf("Setup %s..." % folder)
exePath = gs.GitDir/"app"/folder/"SumatraPDF.exe"
# Setup open with. Need special handling for pdf
regAdd(r"HKCR\Applications\SumatraPDF.exe\DefaultIcon", '@', str(exePath) + ",1")
regAdd(r"HKCR\Applications\SumatraPDF.exe\Shell\Open\Command", '@', '"%s" "%%1" %%*' % str(exePath))
regAdd(r"HKCU\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\FileExts\.pdf\OpenWithList", "a", "SumatraPDF")
# Copy the config
srcDir = gs.GitDir/"backup/sumatra"
dstDir = gs.GitDir/"app"/folder
if srcDir.is_dir():
copyPath(srcDir, dstDir)
def main(argv):
# Parse the command line
parser = argparse.ArgumentParser()
parser.add_argument("-path", help="The path to add")
args = parser.parse_args(argv)
if not args.path:
args.path = os.getcwd()
userPath = regQuery(r"HKCU\Environment", "Path")[0]
pathList = [p.strip() for p in userPath.split(';')]
if args.path in pathList:
pathList.remove(args.path)
regAdd(r"HKCU\Environment", "Path", ";".join(pathList))
notifyEnvChange()
printf("Deleted user env path %s" % args.path, color=gs.GREEN)
else:
printf("%s is not in the user env path" % args.path, color=gs.YELLOW)
def setupListary():
folder = AppVersionDict["listary"]
printf("Setup %s..." % folder)
exePath = gs.GitDir/"app"/folder/"Listary.exe"
startupCmd = '"%s" -startup' % str(exePath)
regAdd(r"HKCU\SOFTWARE\Microsoft\Windows\CurrentVersion\Run",
"Listary", startupCmd)
if not getProcessCmd("Listary"):
call(startupCmd, DETACH=True)
if call("sc query ListaryService")[2] != 0:
cmd = ('sc create ListaryService binPath= "%s" start= auto group= "Extended Base" ' %
str(gs.GitDir/"app"/folder/"ListaryService.exe"))
call(cmd)
cmd = "sc start ListaryService"
call(cmd)
# Copy the config
srcPath = gs.GitDir/"backup/listary/UserData/Preferences.json"
dstPath = Path(os.environ["appdata"])/"Listary/UserData/Preferences.json"
if srcPath.is_file():
copyPath(srcPath, dstPath)
def setupScoop():
if not hasInternet():
printf("Have no internet connection. Will skip scoop setup")
return
printf("Setup scoop...")
ps1 = "Set-ExecutionPolicy RemoteSigned -scope CurrentUser"
calls('powershell -Command "%s"' % ps1)
ps1 = "iwr -useb get.scoop.sh | iex"
calls('powershell -Command "%s"' % ps1)
scoopDir = Path(os.environ["USERPROFILE"])/"scoop/shims"
calls("scoop bucket add extras", SHELL=True, DIR=str(scoopDir))
call("scoop update", SHELL=True, DIR=str(scoopDir))
printf("Install vscode...")
calls("scoop install vscode", SHELL=True, DIR=str(scoopDir))
vscodeRegPath = Path(os.environ["USERPROFILE"])/"scoop/apps/vscode/current/vscode-install-context.reg"
calls("reg import %s" % str(vscodeRegPath))
def main(argv):
if not argv:
printf("Please specifiy the setup target or use all instead")
return 1
if argv[0] == "backup":
backup()
return 0
if argv[0] == "all":
argv[0] = "7z,python,notepad++,git,font,optimize,conemu,ahk,pdf,listary,scoop"
setupList = argv[0].split(',')
printf("Running setup...")
if "python" in setupList:
setupPython()
if "7z" in setupList:
setup7z()
if "font" in setupList:
setupFont()
if "notepad++" in setupList:
setupNpp()
if "git" in setupList:
setupGit()
if "optimize" in setupList:
optimizeWin10()
if "conemu" in setupList:
setupConEmu()
if "ahk" in setupList:
setupAutoHotKey()
if "pdf" in setupList:
setupSumatraPdf()
if "listary" in setupList:
setupListary()
if "scoop" in setupList:
setupScoop()
printf("Setup done!")
return 0
def main(argv):
# Parse the command line
parser = argparse.ArgumentParser()
parser.add_argument("-linter",
help="Choose the linters (separated by ,)",
default="all")
args = parser.parse_args(argv)
supportedLinters = ["pylint", "flake8"]
printf("Supported linters: " + ','.join(supportedLinters))
if args.linter != "all":
linters = args.linter.split(',')
else:
linters = supportedLinters
os.chdir(str(gs.GitDir))
for linter in linters:
if linter in supportedLinters:
printf("Running linter %s..." % linter, color=gs.GREEN)
ret = call("python -m %s scripts" % linter)
print(ret[0])
else:
printf("Unsupported linter %s!" % linter, color=gs.RED)
def setupFont():
for p in (gs.GitDir/"res/fonts").iterdir():
if p.suffix.lower() == ".ttf":
printf("Installing font %s" % p.name)
installFont(p)
def optimizeWin10():
printf("Optimize win10 settings...")
# 隐藏任务栏搜索按钮/搜索框
restartExplorer = False
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Search",
"SearchboxTaskbarMode", 0):
restartExplorer = True
# <ExplorerNotify Type="Custom" msg="1A" lParam="TraySettings"/>
# 隐藏多任务按钮
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Advanced",
"ShowTaskViewButton", 0):
restartExplorer = True
# <ExplorerNotify Type="Custom" msg="1A" lParam="TraySettings"/>
# 从不合并任务栏
regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Explorer\Advanced", "TaskbarGlomLevel", 2)
# 隐藏任务栏上的人脉
if (regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Explorer\Advanced\People",
"PeopleBand", 0) or
regAdd(r"HKEY_CURRENT_USER\Software\Policies\Microsoft\Windows\Explorer",
"HidePeopleBar", 1)):
restartExplorer = True
# <Activate Restart="Explorer">
# 显示开始菜单、任务栏、操作中心和标题栏的颜色
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Themes\Personalize", "ColorPrevalence", 1):
restartExplorer = True
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\DWM", "ColorPrevalence", 1):
restartExplorer = True
# <ExplorerNotify Type="Custom" msg="111" wParam="#A220" Explorer="True">
# 使开始菜单、任务栏、操作中心透明
regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Themes\Personalize", "EnableTransparency", 1)
# 关闭商店应用推广
regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\ContentDeliveryManager",
"PreInstalledAppsEnabled", 0)
# 关闭锁屏时的Windows 聚焦推广
regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\ContentDeliveryManager",
"RotatingLockScreenEnable", 0)
# 关闭游戏录制工具
regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\GameDVR", "AppCaptureEnabled", 0)
# 关闭多嘴的小娜
regAdd(r"HKEY_LOCAL_MACHINE\SOFTWARE\Policies\Microsoft\Windows\Windows Search", "AllowCortana", 0)
# 打开资源管理器时显示此电脑
regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Advanced", "LaunchTo", 1)
# 显示所有文件扩展名
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Advanced", "HideFileExt", 0):
restartExplorer = True
# <ExplorerNotify Type="Custom" msg="111" wParam="#A220" Explorer="True">
# 隐藏快捷方式小箭头
if regAdd(r"HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Shell Icons",
"29", r"%systemroot%\Blank.ico", winreg.REG_EXPAND_SZ):
copyPath(gs.GitDir/"res/icon/blank.ico", Path(os.environ["systemroot"]))
restartExplorer = True
# 创建快捷方式时不添 快捷方式 文字
if regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Explorer",
"Link", bytes.fromhex("00000000")):
restartExplorer = True
# 收起资源管理器功能区 ribbon
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Ribbon",
"MinimizedStateTabletModeOff", 1):
restartExplorer = True
# <ExplorerNotify Type="Custom" msg="111" wParam="#A220" Explorer="True">
# 禁止自动播放
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\AutoplayHandlers",
"DisableAutoplay", 1):
restartExplorer = True
# <ExplorerNotify Type="Custom" msg="111" wParam="#A220" Explorer="True)
# 在桌面显示我的电脑
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\HideDesktopIcons\NewStartPanel",
"{20D04FE0-3AEA-1069-A2D8-08002B30309D}", 0):
restartExplorer = True
# <ExplorerNotify Type="AssocChanged>
# 在桌面显示回收站
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\HideDesktopIcons\NewStartPanel",
"{645FF040-5081-101B-9F08-00AA002F954E}", 0):
restartExplorer = True
# <ExplorerNotify Type="AssocChanged>
# 在桌面显示控制面板
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\HideDesktopIcons\NewStartPanel",
"{5399E694-6CE5-4D6C-8FCE-1D8870FDCBA0}", 0):
restartExplorer = True
# <ExplorerNotify Type="AssocChanged>
# 在桌面显示用户文件夹
# if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\HideDesktopIcons\NewStartPanel",
# "{59031a47-3f72-44a7-89c5-5595fe6b30ee}", 0):
# restartExplorer = True
# <ExplorerNotify Type="AssocChanged>
# 在桌面显示网络
# if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\HideDesktopIcons\NewStartPanel",
# "{F02C1A0D-BE21-4350-88B0-7367FC96EF3C}", 0):
# restartExplorer = True
# <ExplorerNotify Type="AssocChanged>
# 启用自动换行
regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Notepad", "fWrap", 1)
# 始终显示状态栏
regAdd(r"HKEY_CURRENT_USER\Software\Microsoft\Notepad", "StatusBar", 1)
# 禁用客户体验改善计划
regAdd(r"HKEY_LOCAL_MACHINE\SOFTWARE\Policies\Microsoft\SQMClient\Windows", "CEIPEnable", 0)
# 启用 Windows 照片查看器
regAddList(r"HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows Photo Viewer\Capabilities\FileAssociations", [
(".jpg", "PhotoViewer.FileAssoc.Tiff"),
(".png", "PhotoViewer.FileAssoc.Tiff"),
(".jpeg", "PhotoViewer.FileAssoc.Tiff"),
(".bmp", "PhotoViewer.FileAssoc.Tiff"),
(".jpe", "PhotoViewer.FileAssoc.Tiff"),
(".jfif", "PhotoViewer.FileAssoc.Tiff"),
(".dib", "PhotoViewer.FileAssoc.Tiff"),
(".ico", "PhotoViewer.FileAssoc.Tiff"),
(".gif", "PhotoViewer.FileAssoc.Tiff"),
(".tga", "PhotoViewer.FileAssoc.Tiff")])
# Set accent color
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Themes\Personalize",
"AppsUseLightTheme", 1):
restartExplorer = True
if regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Themes\Personalize",
"SystemUsesLightTheme", 0):
restartExplorer = True
regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\DWM", "AccentColor", 0x00748501)
regDelete(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\DWM", "AccentColorInactive")
# Clear the check box in explorer
regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Advanced", "AutoCheckSelect", 0)
# Change the wall paper
regAdd(r"HKEY_CURRENT_USER\control panel\desktop", "WallPaper", r"c:\windows\web\wallpaper\theme1\img13.jpg")
call("RunDll32.exe USER32.DLL UpdatePerUserSystemParameters")
# Close Feeds
regAdd(r"HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Feeds", "ShellFeedsTaskbarViewMode", 2)
# Disable one drive
regAdd(r"HKEY_LOCAL_MACHINE\SOFTWARE\Policies\Microsoft\Windows\OneDrive", "DisableFileSyncNGSC", 1)
if restartExplorer:
# Windows 10 will automatically start explorer after the process was killed
killProcess("Explorer")