diff -r 9c9db6355381 -r e37a29d23c86 sbin/sync/fabfile.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sbin/sync/fabfile.py Wed Jun 06 23:00:42 2012 +0200 @@ -0,0 +1,194 @@ +from fabric.api import task, run, local, env, cd, put, prefix, sudo +from fabric.colors import green +from fabric.contrib.project import rsync_project +from mercurial import commands, ui, hg, cmdutil +import imp +import os, os.path +import shutil +import sys + +import config + +def get_export_path(version): + base_path = os.path.join(env.base_export_path,env.export_prefix).lstrip("/") + return os.path.expanduser(base_path) + "_%s" % (str(version)) + +def clean_export_folder(path): + print("Removing %s" % path) + if os.path.isdir(path): + shutil.rmtree(path, ignore_errors=True) + +def do_export_version(path, version): + print("Export version %s"%str(version)) + + #hgui = ui.ui() + #repo = hg.repository(hgui, cmdutil.findrepo(os.getcwd())) + #commands.archive(hgui, repo, path, rev=str(version)) + + local("hg archive -r \'%s\' \"%s\"" % (str(version),path)) + print("Export version %s done"%str(version)) + +def rsync_export(path, remotepath, filters): + print("Rsync %s to %s",(path,remotepath)) + + if filters: + filter_option_str = " ".join(["--filter \"%s\"" % (f) for f in filters]) + else: + filter_option_str ="" + + run("mkdir -p \"%s\"" % remotepath) + rsync_project(remotepath, local_dir=path, extra_opts=filter_option_str, delete=True) + print("Rsync %s to %s done",(path,remotepath)) + +def clean_rsync_folder(remotepath): + print("clean rsync folder %s" % remotepath) + run("rm -fr \"%s\"" % remotepath) + +def build_src(path): + print("Build source dist at %s" % path) + f = None + try: + f, pathname, description = imp.find_module("setup", [path]) + print(" 2 Build source dist at %s" % path) + setup_mod = imp.load_module("setup", f, pathname, description) + print(" 3 Build source dist at %s" % path) + finally: + if f: + f.close() + + setup_mod.launch_setup("setup.py", ['sdist']) + + print("Build source dist at %s done" % path) + + +def get_src_version(path): + print("get src version at %s" % path) + f = None + try: + f, pathname, description = imp.find_module("ldt", [path]) + ldt_mod = imp.load_module("ldt", f, pathname, description) + finally: + if f: + f.close() + version = ldt_mod.VERSION + version_str = ldt_mod.get_version() + + return (version, version_str) + + +def sync_build(path): + print("Sync build %s" % path) + with cd(env.remote_ldt_base_path): + filename = os.path.basename(path) + res_trans = put(path, os.path.join(env.remote_ldt_base_path, filename)) + print("Sync build %s to %s" % (path,repr(res_trans))) + return res_trans + +def remove_build(path): + print("remove build build %s" % path) + run("rm \"%s\"" % path) + + +def install_build(remotepath, remotevirtualenvpath): + print("Install build %s in %s" % (remotepath, remotevirtualenvpath)) + activate_path = os.path.join(remotevirtualenvpath, "bin/activate") + + with prefix("source %s" % activate_path): + run("pip install \"%s\"" % remotepath) + +def collectstatic(remotepath, remotevirtualenvpath): + print("Collect static in %s with %s" % (remotepath, remotevirtualenvpath)) + activate_path = os.path.join(remotevirtualenvpath, "bin/activate") + with prefix("source \"%s\"" % activate_path), prefix("export PYTHONPATH=\"%s\"" % remotepath), cd(os.path.join(remotepath, env.platform_web_module)): + run("python manage.py collectstatic --noinput") + + + +def export_version(version): + print("export version %s" % str(version)) + + export_path = get_export_path(version) + + clean_export_folder(export_path) + do_export_version(export_path,version) + + return export_path + +def do_create_virtualenv(remote_venv_export_path, remotevirtualenvpath): + print("Create virtualenv export_path : %s - remote venvpath : %s" % (remote_venv_export_path, remotevirtualenvpath)) + activate_path = os.path.join(remotevirtualenvpath, "bin/activate") + if "remote_baseline_venv" in env and env.remote_baseline_venv: + prefix_str = "source \"%s\"" % os.path.join(env.remote_baseline_venv, "bin/activate") + else: + prefix_str = "echo" + run("mkdir -p \"%s\"" % remotevirtualenvpath) + with prefix(prefix_str), cd(os.path.join(remote_venv_export_path,"virtualenv","web")): + run("python create_python_env.py") + run("python project-boot.py --unzip-setuptools --no-site-packages --clear --type-install=local \"%s\"" % remotevirtualenvpath) + with prefix("source \"%s\"" % activate_path): + run("pip install -r \"%s\"" % os.path.join(remote_venv_export_path,"virtualenv","web","res","srvr_requirements.txt")) + +def do_sync_ldt(version, export_path): + print("do_sync_ldt with version %s and path %s" % (version,export_path)) + src_path = export_path + "/src/ldt" + build_src(src_path) + (_,version_str) = get_src_version(src_path) + build_path = os.path.join(src_path,"dist","ldt-%s.tar.gz" % version_str) + res_trans = None + try: + res_trans = sync_build(build_path) + install_build(res_trans[0], env.remote_virtualenv_path) + finally: + if res_trans: + remove_build(res_trans[0]) + +def do_sync_web(version, export_path): + print("do_sync_web with version %s and path %s" % (version,export_path)) + web_path = os.path.join(export_path,"web/") + rsync_export(web_path, env.remote_web_path, env.web_rsync_filters) + +def relaunch_server(): + print("Relaunch server") + collectstatic(env.remote_web_path, env.remote_virtualenv_path) + sudo(env.web_relaunch_cmd, shell=False) + +@task +def sync_web(version): + print(green("sync web with version %s" % version)) + export_path = export_version(version) + do_sync_web(version, export_path) + clean_export_folder(export_path) + relaunch_server() + +@task +def sync_ldt(version): + print(green("sync ldt with version %s" % version)) + export_path = export_version(version) + do_sync_ldt(version, export_path) + clean_export_folder(export_path) + relaunch_server() + +@task +def sync_platform(version): + print(green("sync platform with version %s" % version)) + export_path = export_version(version) + do_sync_ldt(version, export_path) + do_sync_web(version, export_path) + clean_export_folder(export_path) + relaunch_server() + +@task +def create_virtualenv(version): + print(green("create virtualenv with version %s" % version)) + export_path = export_version(version) + venv_remote_export_path = "" + try: + virtualenv_path = os.path.join(export_path, "virtualenv") + + venv_remote_export_path = os.path.join(env.remote_venv_export_path, env.export_prefix, version,"virtualenv") + rsync_export(virtualenv_path, venv_remote_export_path, env.venv_rsync_filters) + do_create_virtualenv(venv_remote_export_path, env.remote_virtualenv_path) + finally: + clean_export_folder(export_path) + if venv_remote_export_path: + clean_rsync_folder(venv_remote_export_path)