Git Updater

2011-08-17 17:37:25 +02:00
parent 6735cc9e5e
commit f3c142f531
15 changed files with 1033 additions and 0 deletions
@@ -0,0 +1,30 @@
+from .main import Updater
+
+def start():
+    return Updater()
+
+config = [{
+    'name': 'updater',
+    'groups': [
+        {
+            'tab': 'general',
+            'name': 'updater',
+            'label': 'Updater',
+            'git_only': True,
+            'options': [
+                {
+                    'name': 'enabled',
+                    'default': True,
+                    'type': 'enabler',
+                    'description': 'Enable periodic update checking',
+                },
+                {
+                    'name': 'automatic',
+                    'default': False,
+                    'type': 'enabler',
+                    'description': 'Automaticly update when update is available',
+                },
+            ],
+        },
+    ],
+}]
@@ -0,0 +1,79 @@
+from couchpotato.core.event import addEvent, fireEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+from couchpotato.environment import Env
+from git.repository import LocalRepository
+import time
+
+log = CPLog(__name__)
+
+
+class Updater(Plugin):
+
+    git = 'git://github.com/CouchPotato/CouchPotato.git'
+
+    running = False
+    version = None
+    updateFailed = False
+    updateAvailable = False
+    updateVersion = None
+    lastCheck = 0
+
+    def __init__(self):
+
+        self.repo = LocalRepository(Env.get('app_dir'))
+
+        fireEvent('schedule.interval', 'updater.check', self.check, hours = 6)
+
+        addEvent('app.load', self.check)
+
+    def getVersion(self):
+
+        if not self.version:
+            try:
+                output = self.repo.getHead() # Yes, please
+                log.debug('Git version output: %s' % output.hash)
+                self.version = output.hash
+            except Exception, e:
+                log.error('Failed using GIT updater, running from source, you need to have GIT installed. %s' % e)
+                return 'No GIT'
+
+        return self.version
+
+    def check(self):
+
+        if self.updateAvailable or self.isDisabled():
+            return
+
+        current_branch = self.repo.getCurrentBranch().name
+
+        for branch in self.repo.getRemoteByName('origin').getBranches():
+            if current_branch == branch.name:
+
+                local = self.repo.getHead()
+                remote = branch.getHead()
+
+                if local.getDate() <  remote.getDate():
+                    if self.conf('automatic') and not self.updateFailed:
+                        self.doUpdate()
+                    else:
+                        self.updateAvailable = True
+                        self.updateVersion = remote.hash
+
+        self.lastCheck = time.time()
+
+    def doUpdate(self):
+        try:
+            log.info('Updating to latest version');
+            self.repo.pull()
+            return True
+        except Exception, e:
+            log.error('Failed updating via GIT: %s' % e)
+
+        self.updateFailed = True
+
+        return False
+
+    def isEnabled(self):
+        return Plugin.isEnabled(self) and Env.get('uses_git')
+