--- a/sbin/sync/fabfile.py Fri Nov 30 14:02:30 2012 +0100
+++ b/sbin/sync/fabfile.py Tue Dec 04 00:17:10 2012 +0100
@@ -1,236 +1,13 @@
-from fabric.api import task, run, local, env, cd, put, prefix, sudo, lcd
+from fabric.api import task, env, sudo
from fabric.colors import green
-from fabric.contrib.project import rsync_project
-from fabric.contrib.files import exists, upload_template
-from fabric.context_managers import settings
-from mercurial import commands, ui, hg, cmdutil
+from fabric.tasks import Task
+from ldt_fablib import (check_folder_access, syncdb, collectstatic,
+ export_version, do_sync_web, create_config, clean_export_folder, do_sync_comp,
+ sync_install_build, do_create_virtualenv, clean_rsync_folder, rsync_export,
+ get_comp_versions_dict)
import imp
-import os, os.path
-import shutil
-import sys
-import urlparse
-
-import config
-
-def get_export_path(version):
- base_path = os.path.join(env.base_export_path,env.export_prefix).rstrip("/")
- 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, **export_keys):
- print("Export version %s " % (repr(export_keys)))
-
- for export_key, version in export_keys.items():
- export_path = os.path.join(path,export_key)
-
- repo_url = env.repos[export_key]
- url_part = urlparse.urlparse(repo_url)
- if url_part.scheme or url_part.netloc:
- # this is a remote repo. Let's clone first
- clone_path = os.path.join(path,'clone',export_keys)
- os.makedirs(clone_path)
- local("hg clone \"%s\" \"%s\"" % (repo_url,clone_path))
- else:
- clone_path = repo_url
-
- with lcd(clone_path):
- local("hg archive -r \'%s\' \"%s\"" % (str(version),export_path))
-
- print("Export version %s done"%repr(export_keys))
-
-
-def get_remote_env(remotepath, remotevirtualenvpath, platform_web_module, settings_key):
- activate_path = os.path.join(remotevirtualenvpath, "bin/activate")
- res = ""
- with prefix("source \"%s\"" % activate_path), prefix("export PYTHONPATH=\"%s\"" % remotepath), cd(remotepath):
- tempfilepath = run("mktemp -t ldtplatform.XXXXXX")
- with settings(warn_only=True):
- run("echo \"import os\" > %s" % (tempfilepath))
- map(lambda str: run("echo \"%s\" >> %s" % (str, tempfilepath)),
- ["os.environ.setdefault('DJANGO_SETTINGS_MODULE', '%s.settings')" % (platform_web_module),
- "from django.conf import settings",
- "print settings.%s" % (settings_key)])
- res = run("python < %s" % (tempfilepath))
- run("rm -f \"%s\"" % (tempfilepath))
- return res
-
-
-
-def rsync_export(path, remotepath, filters):
- print("Rsync %s to %s",(path,remotepath))
-
- filter_option_str = "--progress --stats"
- if filters:
- filter_option_str += " " + " ".join(["--filter \"%s\"" % (f) for f in filters])
-
- 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_path['ldt_base']):
- filename = os.path.basename(path)
- res_trans = put(path, os.path.join(env.remote_path['ldt_base'], filename))
- print("Sync build %s to %s" % (path,repr(res_trans)))
- return res_trans
+import os.path
-def remove_build(path):
- print("remove build build %s" % path)
- run("rm \"%s\"" % path)
-
-
-def install_build(remotepath, remotevirtualenvpath, module_to_uninstall= None):
- print("Install build %s in %s" % (remotepath, remotevirtualenvpath))
- activate_path = os.path.join(remotevirtualenvpath, "bin/activate")
-
- with prefix("source %s" % activate_path):
- if module_to_uninstall:
- with settings(warn_only=True):
- run("pip uninstall -y %s" % module_to_uninstall)
- run("pip install \"%s\"" % remotepath)
-
-def collectstatic(remotepath, remotevirtualenvpath, platform_web_module):
- print("Collect static in %s with %s" % (remotepath, remotevirtualenvpath))
- remotestaticsitepath = get_remote_env(remotepath, remotevirtualenvpath, platform_web_module, "STATIC_ROOT")
- activate_path = os.path.join(remotevirtualenvpath, "bin/activate")
- with prefix("source \"%s\"" % activate_path), prefix("export PYTHONPATH=\"%s\"" % remotepath), cd(remotepath):
- #remocve old files optio -c of collect static fail !
- run("rm -fr \"%s\"" % (remotestaticsitepath))
- run("python manage.py collectstatic --noinput")
-
-def syncdb(remotepath, remotevirtualenvpath):
- activate_path = os.path.join(remotevirtualenvpath, "bin/activate")
- with prefix("source \"%s\"" % activate_path), prefix("export PYTHONPATH=\"%s\"" % remotepath), cd(remotepath):
- run("python manage.py syncdb --migrate --noinput")
-
-def create_config(export_path):
- print("Create config from %s" % (export_path,))
- remotepath = env.remote_path['src']
- remote_config_path = os.path.join(remotepath, env.platform_web_module, "config.py")
- template_path = os.path.join(export_path, "src", env.platform_web_module, "config.py.tmpl")
-
- context = {
- 'base_dir': os.path.join(remotepath, env.platform_web_module).rstrip("/")+"/",
- 'asctime': '%(asctime)s',
- 'levelname': '%(levelname)s',
- 'message': '%(message)s',
- 'module': '%(module)s',
- }
- context.update(env.config['web'])
-
- if not exists(remote_config_path, verbose=True):
- upload_template(template_path, remote_config_path, context=context)
-
-def export_version(**kwargs):
- print("export version %s" % (repr(kwargs)))
-
- export_path = get_export_path("_".join(["%s_%s" % (k,v) for k,v in kwargs.items()]))
-
- clean_export_folder(export_path)
-
- do_export_version(export_path,**kwargs)
-
- 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"
- with settings(warn_only=True):
- run("rm -fr \"%s\"" % remotevirtualenvpath)
- 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 \"%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)
- sync_install_build(build_path, 'ldt')
-
-
-def sync_install_build(build_path, module_to_uninstall=None):
- res_trans = None
- try:
- res_trans = sync_build(build_path)
- install_build(res_trans[0], env.remote_path['virtualenv'], module_to_uninstall)
- 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))
- #sync web
- web_path = os.path.join(export_path,"web/")
- rsync_export(web_path, env.remote_path['web'], env.rsync_filters['web'])
- #sync src
- src_path = os.path.join(export_path,"src/")
- rsync_export(src_path, env.remote_path['src'], env.rsync_filters['src'])
-
-
-def check_folder_access():
- print("Check folder access")
- # get remote user
- for folder_path in env.folders:
- if not os.path.isabs(folder_path):
- folder_path = env.remote_path['web'].rstrip("/")+ "/" + folder_path
- with settings(warn_only=True):
- if not exists(folder_path):
- run("mkdir -p \"%s\"" % folder_path)
- run("chown -R -c :%s \"%s\"" % (env.web_group, folder_path))
- run("chmod -R -c g+w \"%s\"" % folder_path)
@task
def relaunch_server(do_collectstatic=True, do_syncdb=True):
print("Relaunch server")
@@ -250,15 +27,6 @@
create_config(export_path_full)
clean_export_folder(export_path)
relaunch_server()
-
-@task
-def sync_ldt(version):
- print(green("sync ldt with version %s" % version))
- export_path = export_version(ldt=version)
- export_path_full = os.path.join(export_path,'ldt')
- do_sync_ldt(version, export_path_full)
- clean_export_folder(export_path)
- relaunch_server()
@task
def update_lib(version, package):
@@ -268,24 +36,32 @@
lib_path = os.path.join(export_path_full, "virtualenv", "res", "lib")
f, pathname, description = imp.find_module("patch", [lib_path])
- patch = imp.load_module("patch", f, pathname, description)
+ imp.load_module("patch", f, pathname, description)
f, pathname, description = imp.find_module("lib_create_env", [lib_path])
lib_create_env = imp.load_module("lib_create_env", f, pathname, description)
- package_path = os.path.join(export_path_full, "virtualenv", "res", "src", lib_create_env.URLS[package]['local'])
+ package_path_full = os.path.join(export_path_full, "virtualenv", "res", "src", lib_create_env.URLS[package]['local'])
sync_install_build(package_path_full)
clean_export_folder(export_path)
relaunch_server()
@task
-def sync_platform(version_web, version_ldt):
- print(green("sync platform with version web %s, ldt %s" % (version_web, version_ldt)))
- export_path = export_version(ldt=version_ldt, web=version_web)
- export_path_ldt = os.path.join(export_path,'ldt')
+def sync_platform(version):
+ print(green("sync platform with version web %s" % (version)))
+ export_path = export_version(web=version)
+
export_path_web = os.path.join(export_path,'web')
- do_sync_ldt(version_ldt, export_path_ldt)
- do_sync_web(version_web, export_path_web)
+ do_sync_web(version, export_path_web)
+
+ comp_versions = get_comp_versions_dict(export_path_web)
+
+ for key in [k for k in env.repos if key != 'web']:
+ export_path_key = export_version(**{key: comp_versions[key]})
+ export_path_comp = os.path.join(export_path_key, key)
+ do_sync_comp(key, export_path_comp)
+ clean_export_folder(export_path_key)
+
create_config(export_path_web)
clean_export_folder(export_path)
relaunch_server()
@@ -307,4 +83,29 @@
if venv_remote_export_path:
clean_rsync_folder(venv_remote_export_path)
+class SyncComp(Task):
+
+ def __init__(self, key):
+ self.key = key
+ def __get_name(self):
+ return "sync_" + self.key
+
+ name = property(__get_name)
+
+ def run(self, version):
+ print(green("sync ldt with version %s" % version))
+ export_path_web = export_version(web=version)
+ export_path_web_full = os.path.join(export_path_web,'web')
+ comp_versions = get_comp_versions_dict(export_path_web_full)
+
+ export_path = export_version(**{self.key:comp_versions[self.key]})
+ export_path_full = os.path.join(export_path,self.key)
+ do_sync_comp(self.key, export_path_full)
+ clean_export_folder(export_path)
+ clean_export_folder(export_path_web)
+
+ relaunch_server()
+
+for sync_name in [key for key in env.repos if key != 'web']:
+ globals()[sync_name] = SyncComp(sync_name)