def gitUpdate():
"""Update radio using git"""
output, err = runGit('pull origin %s' % branch)
if not output:
logger.log('Couldn\'t download latest version', 'ERROR')
radio.USE_GIT = False
return 'failed'
for line in output.split('\n'):
if 'Already up-to-date.' in line:
logger.log('UPDATER :: Already up to date', 'INFO')
logger.log('UPDATER :: Git output: ' + str(output), 'DEBUG')
return 'complete'
elif 'Aborting' in line:
logger.log('UPDATER :: Unable to update from git: ' + line, 'ERROR')
logger.log('UPDATER :: Output: ' + str(output), 'DEBUG')
radio.USE_GIT = False
return 'failed'
radio.CURRENT_COMMIT = radio.LATEST_COMMIT
writeVersion(radio.LATEST_COMMIT)
radio.COMMITS_BEHIND = 0
return 'complete'
def runGit(args):
"""Run git command with args as arguments"""
git_locations = ['git']
if platform.system().lower() == 'darwin':
git_locations.append('/usr/local/git/bin/git')
output = err = None
for cur_git in git_locations:
cmd = cur_git + ' ' + args
try:
logger.log('UPDATER :: Trying to execute: "' + cmd + '" with shell in ' + RUNDIR, 'DEBUG')
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=True, cwd=RUNDIR)
output, err = p.communicate()
logger.log('UPDATER :: Git output: ' + output, 'DEBUG')
except OSError:
logger.log('UPDATER :: Command ' + cmd + ' didn\'t work, couldn\'t find git', 'WARNING')
continue
if 'not found' in output or "not recognized as an internal or external command" in output:
logger.log('UPDATER :: Unable to find git with command ' + cmd, 'WARNING')
output = None
elif 'fatal:' in output or err:
logger.log('UPDATER :: Git returned bad info. Are you sure this is a git installation?', 'WARNING')
output = None
elif output:
break
return (output, err)
def gitCurrentVersion():
"""Get version hash for local installation"""
output, err = runGit('rev-parse HEAD')
if not output:
logger.log('UPDATER :: Couldn\'t find latest installed version with git', 'WARNING')
radio.USE_GIT = False
return None
current_commit = output.strip()
if not re.match('^[a-z0-9]+$', current_commit):
logger.log('UPDATER :: Git output doesn\'t look like a hash, not using it', 'WARNING')
return None
writeVersion(current_commit)
return
def RemoveUpdateFiles():
"""Remove the downloaded new version"""
logger.log('UPDATER :: Removing update files', 'INFO')
tar_file = joinRundir('radio.tar.gz')
update_folder = joinRundir('radio-update')
try:
if os.path.exists(tar_file):
logger.log('UPDATER :: Removing %s' % tar_file, 'DEBUG')
os.remove(tar_file)
except:
logger.log('UPDATER :: Could not remove %s' % tar_file, 'WARNING')
try:
if os.path.exists(update_folder):
logger.log('UPDATER :: Removing %s' % update_folder, 'DEBUG')
shutil.rmtree(update_folder)
except:
logger.log('UPDATER :: Could not remove %s' % update_folder, 'WARNING')
return
def checkGithub():
"""Check github repo for updates"""
logger.log('UPDATER :: Checking for updates', 'INFO')
try:
radio.LATEST_COMMIT = latestCommit()
if radio.FIRST_RUN:
radio.CURRENT_COMMIT = radio.LATEST_COMMIT
writeVersion(radio.CURRENT_COMMIT)
except:
logger.log('UPDATER :: Could not get latest commit from github', 'WARNING')
if radio.CURRENT_COMMIT:
try:
radio.COMMITS_BEHIND = commitsBehind()
except:
logger.log('UPDATER :: Could not get commits behind from github', 'WARNING')
if radio.COMMITS_BEHIND >= 1:
logger.log('UPDATER :: Update available, you are %i commits behind' % radio.COMMITS_BEHIND, 'INFO')
radio.COMMITS_COMPARE_URL = 'https://github.com/%s/radio/compare/%s...%s' % (user, radio.CURRENT_COMMIT, radio.LATEST_COMMIT)
elif radio.COMMITS_BEHIND == 0:
logger.log('UPDATER :: Up to date', 'INFO')
elif radio.COMMITS_BEHIND == -1:
logger.log('UPDATER :: Unknown version. Please run the updater', 'INFO')
else:
logger.log('UPDATER :: Unknown version. Please run the updater', 'INFO')
return radio.COMMITS_BEHIND
def Update():
"""Update radio installation"""
if radio.USE_GIT:
update = gitUpdate()
if update == 'complete':
return True
else:
logger.log('Git update failed, attempting tarball update', 'INFO')
tar_file = joinRundir('radio.tar.gz')
update_folder = joinRundir('radio-update')
# Download repo
try:
logger.log('UPDATER :: Downloading update file to %s' % tar_file, 'DEBUG')
url = urllib2.urlopen('https://github.com/%s/radio/tarball/%s' % (user, branch))
f = open(tar_file, 'wb')
f.write(url.read())
f.close()
except:
logger.log('UPDATER :: Failed to download update file', 'WARNING')
RemoveUpdateFiles()
return False
# Write new hash to file
try:
logger.log('UPDATER :: Writing new hash to %s' % version_file, 'DEBUG')
writeVersion(radio.LATEST_COMMIT)
except:
logger.log('UPDATER :: Faied to write new hash to version file', 'WARNING')
RemoveUpdateFiles()
return False
# Extract to temp folder
try:
logger.log('UPDATER :: Extracting %s' % tar_file, 'DEBUG')
tar = tarfile.open(tar_file)
tar.extractall(update_folder)
tar.close()
except:
logger.log('Failed to extract update file', 'WARNING')
RemoveUpdateFiles()
return False
# Overwrite old files with new ones
root_src_dir = os.path.join(update_folder, '%s-radio-%s' % (user, radio.LATEST_COMMIT[:7]))
try:
logger.log('UPDATER :: Overwriting old files', 'DEBUG')
for src_dir, dirs, files in os.walk(root_src_dir):
dst_dir = src_dir.replace(root_src_dir, RUNDIR)
if not os.path.exists(dst_dir):
os.mkdir(dst_dir)
for file_ in files:
src_file = os.path.join(src_dir, file_)
dst_file = os.path.join(dst_dir, file_)
if os.path.exists(dst_file):
os.remove(dst_file)
shutil.move(src_file, dst_dir)
except:
logger.log('UPDATER :: Failed to overwrite old files', 'WARNING')
RemoveUpdateFiles()
return False
# Clean up
RemoveUpdateFiles()
radio.CURRENT_COMMIT = radio.LATEST_COMMIT
radio.COMMITS_BEHIND = 0
return True
