summaryrefslogtreecommitdiffstats
path: root/module/AccountManager.py
diff options
context:
space:
mode:
Diffstat (limited to 'module/AccountManager.py')
-rw-r--r--module/AccountManager.py296
1 files changed, 159 insertions, 137 deletions
diff --git a/module/AccountManager.py b/module/AccountManager.py
index c6c14daf7..b5c89dffc 100644
--- a/module/AccountManager.py
+++ b/module/AccountManager.py
@@ -20,144 +20,166 @@
from os.path import exists
from shutil import copy
+from module.PullEvents import AccountUpdateEvent
+
ACC_VERSION = 1
########################################################################
class AccountManager():
- """manages all accounts"""
-
- #----------------------------------------------------------------------
- def __init__(self, core):
- """Constructor"""
-
- self.core = core
-
- self.accounts = {} # key = ( plugin )
- self.plugins = {}
-
- self.initAccountPlugins()
- self.loadAccounts()
-
- self.saveAccounts() # save to add categories to conf
-
- #----------------------------------------------------------------------
- def getAccountPlugin(self, plugin):
- """get account instance for plugin or None if anonymous"""
- if self.accounts.has_key(plugin):
- if not self.plugins.has_key(plugin):
- self.plugins[plugin] = self.core.pluginManager.getAccountPlugin(plugin)(self, self.accounts[plugin])
-
- return self.plugins[plugin]
- else:
- return None
-
- def getAccountPlugins(self):
- """ get all account instances"""
-
- plugins = []
- for plugin in self.accounts.keys():
- plugins.append(self.getAccountPlugin(plugin))
-
- return plugins
- #----------------------------------------------------------------------
- def loadAccounts(self):
- """loads all accounts available"""
-
- if not exists("accounts.conf"):
- f = open("accounts.conf", "wb")
- f.write("version: " + str(ACC_VERSION))
- f.close()
-
- f = open("accounts.conf", "rb")
- content = f.readlines()
-
- version = content.pop(0)
-
- if int(version.split(":")[1]) < ACC_VERSION:
- copy("accounts.conf", "accounts.backup")
- f.close()
- f = open("accounts.conf", "wb")
- f.write("version: " + str(ACC_VERSION))
- f.close()
- self.core.log.warning(_("Account settings deleted, due to new config format."))
- return
-
-
-
- plugin = ""
- account = ""
-
- for line in content:
- line = line.strip()
-
- if not line: continue
- if line.startswith("#"): continue
- if line.startswith("version"): continue
-
- if line.endswith(":"):
- plugin = line[:-1]
- self.accounts[plugin] = {}
-
- elif line.startswith("@"):
- option = line[1:].split()
- self.accounts[plugin][name]["options"].append(tuple(option))
-
- elif ":" in line:
- name, pw = line.split(":")[:]
- self.accounts[plugin][name] = {"password": pw, "options": []}
-
-
-
- #----------------------------------------------------------------------
- def saveAccounts(self):
- """save all account information"""
-
- f = open("accounts.conf", "wb")
- f.write("version: " + str(ACC_VERSION) + "\n")
-
- for plugin, accounts in self.accounts.iteritems():
- f.write("\n")
- f.write(plugin+":\n")
-
- for name,data in accounts.iteritems():
- f.write("\n\t%s:%s\n" % (name,data["password"]) )
- for option in data["options"]:
- f.write("\t@%s\n" % " ".join(option) )
-
- f.close()
-
-
- #----------------------------------------------------------------------
- def initAccountPlugins(self):
- """init names"""
- for name in self.core.pluginManager.getAccountPlugins():
- self.accounts[name] = {}
-
- #----------------------------------------------------------------------
- def updateAccount(self, plugin , user, password, options):
- """add or update account"""
-
- if self.accounts.has_key(plugin):
- p = self.getAccountPlugin(plugin)
- p.updateAccounts(user, password, options)
-
- if self.accounts[plugin].has_key(user):
- self.accounts[plugin][user]["password"] = password
- self.accounts[plugin][user]["options"] = options
- else:
- self.accounts[plugin][user] = {"password": password, "options": options}
-
- self.saveAccounts()
-
- #----------------------------------------------------------------------
- def removeAccount(self, plugin, user):
- """remove account"""
-
- if self.accounts.has_key(plugin):
- p = self.getAccountPlugin(plugin)
- p.removeAccount(user)
-
- if self.accounts[plugin].has_key(user):
- del self.accounts[plugin][user]
-
- self.saveAccounts()
+ """manages all accounts"""
+
+ #----------------------------------------------------------------------
+ def __init__(self, core):
+ """Constructor"""
+
+ self.core = core
+
+ self.accounts = {} # key = ( plugin )
+ self.plugins = {}
+
+ self.initAccountPlugins()
+ self.loadAccounts()
+
+ self.saveAccounts() # save to add categories to conf
+
+ self.cachedAccountInfo = {}
+
+ #----------------------------------------------------------------------
+ def getAccountPlugin(self, plugin):
+ """get account instance for plugin or None if anonymous"""
+ if self.accounts.has_key(plugin):
+ if not self.plugins.has_key(plugin):
+ self.plugins[plugin] = self.core.pluginManager.getAccountPlugin(plugin)(self, self.accounts[plugin])
+
+ return self.plugins[plugin]
+ else:
+ return None
+
+ def getAccountPlugins(self):
+ """ get all account instances"""
+
+ plugins = []
+ for plugin in self.accounts.keys():
+ plugins.append(self.getAccountPlugin(plugin))
+
+ return plugins
+ #----------------------------------------------------------------------
+ def loadAccounts(self):
+ """loads all accounts available"""
+
+ if not exists("accounts.conf"):
+ f = open("accounts.conf", "wb")
+ f.write("version: " + str(ACC_VERSION))
+ f.close()
+
+ f = open("accounts.conf", "rb")
+ content = f.readlines()
+
+ version = content.pop(0)
+
+ if int(version.split(":")[1]) < ACC_VERSION:
+ copy("accounts.conf", "accounts.backup")
+ f.close()
+ f = open("accounts.conf", "wb")
+ f.write("version: " + str(ACC_VERSION))
+ f.close()
+ self.core.log.warning(_("Account settings deleted, due to new config format."))
+ return
+
+
+
+ plugin = ""
+ account = ""
+
+ for line in content:
+ line = line.strip()
+
+ if not line: continue
+ if line.startswith("#"): continue
+ if line.startswith("version"): continue
+
+ if line.endswith(":"):
+ plugin = line[:-1]
+ self.accounts[plugin] = {}
+
+ elif line.startswith("@"):
+ option = line[1:].split()
+ self.accounts[plugin][name]["options"].append(tuple(option))
+
+ elif ":" in line:
+ name, pw = line.split(":")[:]
+ self.accounts[plugin][name] = {"password": pw, "options": []}
+
+
+
+ #----------------------------------------------------------------------
+ def saveAccounts(self):
+ """save all account information"""
+
+ f = open("accounts.conf", "wb")
+ f.write("version: " + str(ACC_VERSION) + "\n")
+
+ for plugin, accounts in self.accounts.iteritems():
+ f.write("\n")
+ f.write(plugin+":\n")
+
+ for name,data in accounts.iteritems():
+ f.write("\n\t%s:%s\n" % (name,data["password"]) )
+ for option in data["options"]:
+ f.write("\t@%s\n" % " ".join(option) )
+
+ f.close()
+
+
+ #----------------------------------------------------------------------
+ def initAccountPlugins(self):
+ """init names"""
+ for name in self.core.pluginManager.getAccountPlugins():
+ self.accounts[name] = {}
+
+ #----------------------------------------------------------------------
+ def updateAccount(self, plugin , user, password, options):
+ """add or update account"""
+
+ if self.accounts.has_key(plugin):
+ p = self.getAccountPlugin(plugin)
+ p.updateAccounts(user, password, options)
+
+ if self.accounts[plugin].has_key(user):
+ self.accounts[plugin][user]["password"] = password
+ self.accounts[plugin][user]["options"] = options
+ else:
+ self.accounts[plugin][user] = {"password": password, "options": options}
+
+ self.saveAccounts()
+
+ #----------------------------------------------------------------------
+ def removeAccount(self, plugin, user):
+ """remove account"""
+
+ if self.accounts.has_key(plugin):
+ p = self.getAccountPlugin(plugin)
+ p.removeAccount(user)
+
+ if self.accounts[plugin].has_key(user):
+ del self.accounts[plugin][user]
+
+ self.saveAccounts()
+
+ def getAccountInfos(self, force=False):
+ if not self.cachedAccountInfo:
+ self.cacheAccountInfos()
+ elif force:
+ self.core.scheduler.addJob(0, self.cacheAccountInfos, done=self.sendChange)
+ return self.cachedAccountInfo
+
+ def cacheAccountInfos(self):
+ plugins = self.getAccountPlugins()
+ data = {}
+ for p in plugins:
+ data[p.__name__] = p.getAllAccounts()
+ self.cachedAccountInfo = data
+
+ def sendChange(self):
+ e = AccountUpdateEvent()
+ self.core.pullManager.addEvent(e)
generated by cgit v1.2.3 (git 2.25.1) at 2025-01-23 23:36:51 +0100