Fix reposit

Author: xtao

lain_cli/reposit.py

@@ -3,7 +3,7 @@
 
 from lain_sdk.util import info
 from lain_cli.auth import SSOAccess, get_auth_header, authorize_and_check
-from lain_cli.utils import check_phase, lain_yaml, reposit_app, get_domain
+from lain_cli.utils import check_phase, lain_yaml, reposit_app, get_domain, get_phase_stage
 from lain_cli.validate import validate_only_warning
 
 
@@ -17,7 +17,8 @@ def reposit(phase):
     validate_only_warning()
     info("Repositing ...")
 
-    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)
 
     access_token = SSOAccess.get_token(phase)

lain_cli/rmi.py

@@ -2,12 +2,13 @@
 from argh.decorators import arg
 
 import lain_sdk.mydocker as docker
-from lain_cli.utils import check_phase
+from lain_cli.utils import check_phase, get_phase_stage
 from lain_cli.utils import lain_yaml, get_meta_versions_from_tags, get_domain
 
 
 def get_repo_tags_to_remove(phase):
-    yml = lain_yaml(ignore_prepare=True)
+    stage = get_phase_stage(phase)
+    yml = lain_yaml(ignore_prepare=True, stage=stage)
     domain = get_domain(phase)
     registry = "registry.%s" % domain
     all_tags = docker.get_tag_list_in_docker_daemon(registry, yml.appname)