Support stage yaml

Author: xtao

lain_cli/appversion.py

@@ -3,7 +3,7 @@
 
 from lain_sdk.util import warn, info
 from lain_cli.auth import authorize_and_check
-from lain_cli.utils import get_version_lists, lain_yaml, check_phase
+from lain_cli.utils import get_version_lists, lain_yaml, check_phase, get_phase_stage
 
 
 @arg('phase', help="lain cluster phase id, can be added by lain config save")
@@ -13,7 +13,8 @@ def appversion(phase):
     """
 
     check_phase(phase)
-    yml = lain_yaml(ignore_prepare=True)
+    stage = get_phase_stage(phase)
+    yml = lain_yaml(ignore_prepare=True, stage=stage)
     authorize_and_check(phase, yml.appname)
 
     version_list = get_version_lists(phase, yml.appname)

lain_cli/attach.py

@@ -3,7 +3,7 @@
 from entryclient import EntryClient
 from lain_sdk.util import error, info
 from lain_cli.auth import SSOAccess, authorize_and_check
-from lain_cli.utils import check_phase, lain_yaml, get_domain
+from lain_cli.utils import check_phase, lain_yaml, get_domain, get_phase_stage
 
 
 @arg('phase', help="lain cluster phase id, can be added by lain config save")
@@ -14,7 +14,8 @@ def attach(phase, proc_name, instance_no, target=None):
     """
 
     check_phase(phase)
-    yml = lain_yaml(ignore_prepare=True)
+    stage = get_phase_stage(phase)
+    yml = lain_yaml(ignore_prepare=True, stage=stage)
     appname = target if target else yml.appname
     authorize_and_check(phase, appname)
     domain = get_domain(phase)

lain_cli/backup.py

@@ -5,7 +5,7 @@
 from argh.decorators import arg, aliases
 
 from lain_sdk.util import warn
-from lain_cli.utils import lain_yaml_data, check_phase
+from lain_cli.utils import lain_yaml_data, check_phase, get_phase_stage
 from lain_cli.utils import TwoLevelCommandBase, get_domain
 
 
@@ -30,7 +30,8 @@ def jobs(cls, phase):
         list backup jobs for this lain app
         """
         check_phase(phase)
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase)
+        appname = lain_yaml_data(stage=stage)['appname']
         route = "api/v2/app/%s/cron/jobs" % appname
         data = cls._request('GET', phase, route, None)
         if data:
@@ -46,7 +47,8 @@ def list(cls, phase, proc, path):
         list files in the incremental backup direcotry
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         route = "api/v2/app/%s/proc/%s/backups/%s?open=true" % (appname, proc[0], path[0])
         data = cls._request('GET', phase[0], route, None)
         if data:
@@ -63,7 +65,8 @@ def get(cls, phase, proc, volume):
         get all the backups of the given proc's volume
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         route = "api/v2/app/%s/proc/%s/backups?volume=%s" % (appname, proc[0], volume[0])
         data = cls._request('GET', phase[0], route, None)
         if data:
@@ -78,7 +81,8 @@ def delete(cls, phase, proc, files):
         delete a backup of a proc
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         route = "api/v2/app/%s/proc/%s/backups/actions/delete" % (appname, proc[0])
         data = cls._request('POST', phase[0], route, {'files': files})
         if data:
@@ -94,7 +98,8 @@ def recover(cls, phase, proc, backup, files):
         recover the volume from given backup
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         if not files:
             route = "api/v2/app/%s/proc/%s/backups/%s/actions/recover" % (appname, proc[0], backup[0])
             data = cls._request('POST', phase[0], route, None)
@@ -116,7 +121,8 @@ def migrate(cls, phase, proc, backup, files, volume="", to=0):
         recover a instance's volume from other instance's backup
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         if not files:
             route = "api/v2/app/%s/proc/%s/backups/%s/actions/migrate" % (appname, proc[0], backup[0])
             data = cls._request('POST', phase[0], route, {"volume": volume, "to": to})
@@ -135,7 +141,8 @@ def records(cls, phase, rid, num=10):
         list job records of this lain app
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         route = "api/v2/app/%s/cron/records" % appname
         if rid:
             route = "%s/%s" % (route, rid)
@@ -153,7 +160,8 @@ def run(cls, phase, id):
         run a job right now
         """
         check_phase(phase[0])
-        appname = lain_yaml_data()['appname']
+        stage = get_phase_stage(phase[0])
+        appname = lain_yaml_data(stage=stage)['appname']
         route = "api/v2/app/%s/cron/jobs/%s/actions/run" % (appname, id[0])
         data = cls._request('POST', phase[0], route, None)
         if data:

lain_cli/build.py

@@ -4,20 +4,22 @@
 
 import lain_sdk.mydocker as docker
 from lain_sdk.util import error, warn, info
-from lain_cli.utils import lain_yaml
+from lain_cli.utils import lain_yaml, get_phase_stage
 from lain_cli.validate import validate_only_warning
 
 
 @arg('--release', help="build from build image if it exists")
 @arg('--push',    help="tag release and meta image with version and push to registry")
-def build(push=False, release=False):
+@arg('--phase', help="lain cluster phase id, can be added by lain config save")
+def build(phase=None, push=False, release=False):
     """
     Build release and meta images
     """
 
     info("Building meta and release images ...")
     validate_only_warning()
-    yml = lain_yaml()
+    stage = get_phase_stage(phase) if phase else None
+    yml = lain_yaml(stage=stage)
     meta_version = yml.repo_meta_version()
     use_prepare = docker.exist(yml.img_names['prepare'])
     use_build = release and docker.exist(yml.img_names['build'])
@@ -38,4 +40,4 @@ def build(push=False, release=False):
         tag_meta_name = yml.tag_meta_version(meta_name)
         docker.tag(meta_name, tag_meta_name)
         docker.push(tag_meta_name)
-    info("Done lain build.")
+    info("Done lain build.")

lain_cli/deploy.py

@@ -8,7 +8,7 @@
 
 from lain_sdk.util import error, info
 from lain_cli.auth import SSOAccess, authorize_and_check, get_auth_header
-from lain_cli.utils import check_phase, get_domain, lain_yaml, is_resource_instance
+from lain_cli.utils import check_phase, get_domain, lain_yaml, is_resource_instance, get_phase_stage
 from lain_cli.utils import reposit_app, get_version_lists, get_app_state
 from lain_cli.utils import render_app_status, render_proc_status
 
@@ -24,7 +24,8 @@ def deploy(phase, version=None, target=None, proc=None, output='pretty'):
     """
 
     check_phase(phase)
-    yml = lain_yaml(ignore_prepare=True)
+    stage = get_phase_stage(phase)
+    yml = lain_yaml(ignore_prepare=True, stage=stage)
     appname = target if target else yml.appname
     authorize_and_check(phase, appname)
 
@@ -62,7 +63,7 @@ def deploy_app(phase, appname, console, auth_header, version, output):
                 deploy_params = {"meta_version": deploy_version}
             else:
                 deploy_version = valid_version
-        
+
         deploy_r = requests.put(app_url, headers=auth_header, json=deploy_params)
     elif app_r.status_code == 404:
         operation = "deploying"

lain_cli/enter.py

@@ -3,7 +3,7 @@
 from entryclient import EntryClient
 from lain_sdk.util import error
 from lain_cli.auth import SSOAccess, authorize_and_check
-from lain_cli.utils import check_phase, lain_yaml, get_domain
+from lain_cli.utils import check_phase, lain_yaml, get_domain, get_phase_stage
 
 import os
 
@@ -16,7 +16,8 @@ def enter(phase, proc_name, instance_no, target=None):
     """
 
     check_phase(phase)
-    yml = lain_yaml(ignore_prepare=True)
+    stage = get_phase_stage(phase)
+    yml = lain_yaml(ignore_prepare=True, stage=stage)
     appname = target if target else yml.appname
     authorize_and_check(phase, appname)
     domain = get_domain(phase)

lain_cli/imagecheck.py

@@ -32,7 +32,7 @@ def _check_phase_tag(phase):
 
 
 @arg('phase', help="lain phase, can be added by lain config save")
-def check(phase):    
+def check(phase):
     """
     Check current version of release and meta images in the remote registry
     """

lain_cli/logout.py

@@ -17,7 +17,7 @@ def logout(phase):
     domain = get_domain(phase)
     logout_success = SSOAccess.clear_token(phase)
     if logout_success:
-    	docker.logout('registry.%s'%domain)
+        docker.logout('registry.%s'%domain)
         info("Logout successfully!")
     else:
         warn('Logout failed!')

lain_cli/maintainer.py

@@ -33,7 +33,7 @@ def show(cls, phase, username=None):
 
         username: sso username
         """
-        
+
         check_phase(phase)
         yml = lain_yaml(ignore_prepare=True)
         authorize_and_check(phase, yml.appname)
@@ -44,7 +44,7 @@ def show(cls, phase, username=None):
             console, yml.appname)
         if username:
             maintainer_url += '%s/' % username
-        
+
         show_response = requests.get(maintainer_url, headers=auth_header)
         if show_response.status_code < 300:
             info("maintainer detail:")
@@ -60,7 +60,7 @@ def add(cls, phase, username, role):
         """
         add maintianer for different phase
         """
-        
+
         check_phase(phase)
         yml = lain_yaml(ignore_prepare=True)
         authorize_and_check(phase, yml.appname)
@@ -91,7 +91,7 @@ def delete(cls, phase, username):
         authorize_and_check(phase, yml.appname)
         auth_header = get_auth_header(SSOAccess.get_token(phase))
         console = "console.%s" % get_domain(phase)
-        
+
         maintainer_url = "http://%s/api/v1/repos/%s/maintainers/%s/" % (
             console, yml.appname, username)
 

lain_cli/ps.py

@@ -5,7 +5,7 @@
 
 from lain_sdk.util import error
 from lain_cli.auth import SSOAccess, get_auth_header, authorize_and_check
-from lain_cli.utils import check_phase, get_domain, lain_yaml
+from lain_cli.utils import check_phase, get_domain, lain_yaml, get_phase_stage
 from lain_cli.utils import render_app_status
 
 
@@ -17,7 +17,8 @@ def ps(phase, output='pretty'):
     """
 
     check_phase(phase)
-    yml = lain_yaml(ignore_prepare=True)
+    stage = get_phase_stage(phase)
+    yml = lain_yaml(ignore_prepare=True, stage=stage)
     authorize_and_check(phase, yml.appname)
     console = "console.%s" % get_domain(phase)