79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
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')
|