2018-04-03 02:58:14 +00:00
|
|
|
#!/usr/bin/env python
|
|
|
|
|
#coding: UTF-8
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
This script is used to run proper upgrade scripts automatically.
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
import json
|
|
|
|
|
import re
|
|
|
|
|
import glob
|
|
|
|
|
import os
|
|
|
|
|
from os.path import abspath, basename, exists, dirname, join, isdir
|
|
|
|
|
import shutil
|
|
|
|
|
import sys
|
|
|
|
|
import time
|
|
|
|
|
|
|
|
|
|
from utils import (
|
|
|
|
|
call, get_install_dir, get_script, get_command_output, replace_file_pattern,
|
|
|
|
|
read_version_stamp, wait_for_mysql, update_version_stamp, loginfo
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
installdir = get_install_dir()
|
|
|
|
|
topdir = dirname(installdir)
|
|
|
|
|
|
|
|
|
|
def collect_upgrade_scripts(from_version, to_version):
|
|
|
|
|
"""
|
|
|
|
|
Give the current installed version, calculate which upgrade scripts we need
|
|
|
|
|
to run to upgrade it to the latest verison.
|
|
|
|
|
|
|
|
|
|
For example, given current version 5.0.1 and target version 6.1.0, and these
|
|
|
|
|
upgrade scripts:
|
|
|
|
|
|
|
|
|
|
upgrade_4.4_5.0.sh
|
|
|
|
|
upgrade_5.0_5.1.sh
|
|
|
|
|
upgrade_5.1_6.0.sh
|
|
|
|
|
upgrade_6.0_6.1.sh
|
|
|
|
|
|
|
|
|
|
We need to run upgrade_5.0_5.1.sh, upgrade_5.1_6.0.sh, and upgrade_6.0_6.1.sh.
|
|
|
|
|
"""
|
|
|
|
|
from_major_ver = '.'.join(from_version.split('.')[:2])
|
|
|
|
|
to_major_ver = '.'.join(to_version.split('.')[:2])
|
|
|
|
|
|
|
|
|
|
scripts = []
|
2018-06-05 02:54:47 +00:00
|
|
|
for fn in sorted(glob.glob(join(installdir, 'upgrade', 'upgrade_*_*.sh'))):
|
2018-04-03 02:58:14 +00:00
|
|
|
va, vb = parse_upgrade_script_version(fn)
|
|
|
|
|
if va >= from_major_ver and vb <= to_major_ver:
|
|
|
|
|
scripts.append(fn)
|
|
|
|
|
return scripts
|
|
|
|
|
|
|
|
|
|
def parse_upgrade_script_version(script):
|
|
|
|
|
script = basename(script)
|
|
|
|
|
m = re.match(r'upgrade_([0-9+.]+)_([0-9+.]+).sh', basename(script))
|
|
|
|
|
return m.groups()
|
|
|
|
|
|
2018-10-12 23:41:48 +00:00
|
|
|
def run_script_and_update_version_stamp(script, new_version):
|
|
|
|
|
replace_file_pattern(script, 'read dummy', '')
|
|
|
|
|
call(script)
|
|
|
|
|
update_version_stamp(new_version)
|
|
|
|
|
|
2018-04-03 02:58:14 +00:00
|
|
|
def check_upgrade():
|
|
|
|
|
last_version = read_version_stamp()
|
|
|
|
|
current_version = os.environ['SEAFILE_VERSION']
|
|
|
|
|
if last_version == current_version:
|
|
|
|
|
return
|
2018-10-12 23:41:48 +00:00
|
|
|
elif is_minor_upgrade(last_version, current_version):
|
|
|
|
|
minor_upgrade_script = join(installdir, 'upgrade', 'minor-upgrade.sh')
|
|
|
|
|
run_script_and_update_version_stamp(minor_upgrade_script, current_version)
|
|
|
|
|
return
|
|
|
|
|
|
2018-04-03 02:58:14 +00:00
|
|
|
|
|
|
|
|
scripts_to_run = collect_upgrade_scripts(from_version=last_version, to_version=current_version)
|
|
|
|
|
for script in scripts_to_run:
|
|
|
|
|
loginfo('Running scripts {}'.format(script))
|
|
|
|
|
# Here we use a trick: use a version stamp like 6.1.0 to prevent running
|
|
|
|
|
# all upgrade scripts before 6.1 again (because 6.1 < 6.1.0 in python)
|
|
|
|
|
new_version = parse_upgrade_script_version(script)[1] + '.0'
|
2018-10-12 23:41:48 +00:00
|
|
|
run_script_and_update_version_stamp(script, new_version)
|
2018-04-03 02:58:14 +00:00
|
|
|
|
|
|
|
|
update_version_stamp(current_version)
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
|
wait_for_mysql()
|
|
|
|
|
|
|
|
|
|
os.chdir(installdir)
|
|
|
|
|
check_upgrade()
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
main()
|
