mirror of
https://github.com/ankitects/anki.git
synced 2025-09-19 14:32:22 -04:00
114 lines
3.7 KiB
Python
114 lines
3.7 KiB
Python
# Copyright: Damien Elmes <anki@ichi2.net>
|
|
# -*- coding: utf-8 -*-
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
import sys, os, re, traceback
|
|
from aqt.qt import *
|
|
from aqt.utils import showInfo, showWarning, openFolder, isWin
|
|
from anki.hooks import runHook
|
|
|
|
class AddonManager(object):
|
|
|
|
def __init__(self, mw):
|
|
self.mw = mw
|
|
f = self.mw.form; s = SIGNAL("triggered()")
|
|
self.mw.connect(f.actionOpenPluginFolder, s, self.onOpenAddonFolder)
|
|
self.mw.connect(f.actionEnableAllPlugins, s, self.onEnableAllAddons)
|
|
self.mw.connect(f.actionDisableAllPlugins, s, self.onDisableAllAddons)
|
|
if isWin:
|
|
self.clearAddonCache()
|
|
sys.path.insert(0, self.addonsFolder())
|
|
self.loadAddons()
|
|
|
|
def loadAddons(self):
|
|
on, off = self.files()
|
|
for file in on:
|
|
try:
|
|
__import__(file.replace(".py", ""))
|
|
except:
|
|
traceback.print_exc()
|
|
self.rebuildAddonsMenu()
|
|
|
|
# Menus
|
|
######################################################################
|
|
|
|
def rebuildAddonsMenu(self):
|
|
if getattr(self, "addonActions", None) is None:
|
|
self.addonActions = []
|
|
for action in self.addonActions:
|
|
self.mw.form.menuStartup.removeAction(action)
|
|
self.addonActions = []
|
|
on, off = self.files()
|
|
def addObjs(l, enabled):
|
|
l.sort()
|
|
for file in l:
|
|
p = re.sub("\.py(\.off)?", "", file)
|
|
a = QAction(p, self.mw)
|
|
a.setCheckable(True)
|
|
a.setChecked(enabled)
|
|
self.mw.connect(a, SIGNAL("triggered()"),
|
|
lambda f=file: self.toggleAddon(f))
|
|
self.mw.form.menuStartup.addAction(a)
|
|
self.addonActions.append(a)
|
|
addObjs(on, True)
|
|
addObjs(off, False)
|
|
|
|
def onOpenAddonFolder(self, path=None):
|
|
if path is None:
|
|
path = self.addonsFolder()
|
|
openFolder(path)
|
|
|
|
# Enabled/disabled list
|
|
######################################################################
|
|
|
|
def files(self):
|
|
on = []
|
|
off = []
|
|
for f in os.listdir(self.addonsFolder()):
|
|
if not f.endswith(".py"):
|
|
continue
|
|
if f in self.mw.pm.meta['disabledAddons']:
|
|
off.append(f)
|
|
else:
|
|
on.append(f)
|
|
return on, off
|
|
|
|
def onEnableAllAddons(self):
|
|
self.mw.pm.meta['disabledAddons'] = []
|
|
self.mw.pm.save()
|
|
self.rebuildAddonsMenu()
|
|
|
|
def onDisableAllAddons(self):
|
|
on, off = self.files()
|
|
self.mw.pm.meta['disabledAddons'] = on + off
|
|
self.rebuildAddonsMenu()
|
|
|
|
def toggleAddon(self, file):
|
|
if file in self.mw.pm.meta['disabledAddons']:
|
|
self.mw.pm.meta['disabledAddons'].remove(file)
|
|
else:
|
|
self.mw.pm.meta['disabledAddons'].append(file)
|
|
self.mw.pm.save()
|
|
self.rebuildAddonsMenu()
|
|
|
|
# Tools
|
|
######################################################################
|
|
|
|
def addonsFolder(self):
|
|
dir = self.mw.pm.addonFolder()
|
|
if isWin:
|
|
dir = dir.encode(sys.getfilesystemencoding())
|
|
return dir
|
|
|
|
def clearAddonCache(self):
|
|
"Clear .pyc files which may cause crashes if Python version updated."
|
|
dir = self.addonsFolder()
|
|
for curdir, dirs, files in os.walk(dir):
|
|
for f in files:
|
|
if not f.endswith(".pyc"):
|
|
continue
|
|
os.unlink(os.path.join(curdir, f))
|
|
|
|
def registerAddon(self, name, updateId):
|
|
# not currently used
|
|
return
|