chore(controller): modify query latest release

Author: jianxiaoguo

rootfs/api/models/app.py

@@ -269,7 +269,7 @@ def restart(self, **kwargs):  # noqa
 
     def scale(self, user, structure):
         err_msg = None
-        release = self.release_set.filter(failed=False).latest()
+        release = Release.latest(self)
         if (PTYPE_RUN in structure or release.build is None):
             if PTYPE_RUN in structure:
                 err_msg = 'Cannot set scale for reserved types, procfile type is: run'
@@ -348,9 +348,9 @@ def deploy(self, release, ptypes=None, force_deploy=False, rollback_on_failure=T
         release.clean(ptypes)
 
     def mount(self, user, volume, structure=None):
-        if self.release_set.filter(failed=False).latest().build is None:
+        release = Release.latest(self)
+        if release is None or release.build is None:
             raise DryccException('No build associated with this release')
-        release = self.release_set.filter(failed=False).latest()
         app_settings = self.appsettings_set.latest()
         self._mount(user, volume, release, app_settings, structure=structure)
 
@@ -387,8 +387,8 @@ def pod_name(size=5, chars=string.ascii_lowercase + string.digits):
             return ''.join(random.choice(chars) for _ in range(size))
 
         """Run a one-off command in an ephemeral app container."""
-        release = self.release_set.filter(failed=False).latest()
-        if release.build is None:
+        release = Release.latest(self)
+        if release is None or release.build is None:
             raise DryccException('No build associated with this release to run this command')
 
         app_settings = self.appsettings_set.latest()
@@ -656,8 +656,8 @@ def image_pull_secret(self, namespace, registry, image):
         return name
 
     def state_to_k8s(self):
-        release = self.release_set.filter(failed=False).latest()
-        if release.build is None:
+        release = Release.latest(self)
+        if release is None or release.build is None:
             self.log('the last release does not have a build, skipping deployment...')
             return
         ptypes = set()

rootfs/api/models/build.py

@@ -4,6 +4,7 @@
 from api.exceptions import DryccException
 from .base import UuidAuditedModel, PTYPE_WEB
 from .config import Config
+from .release import Release
 
 User = get_user_model()
 logger = logging.getLogger(__name__)
@@ -72,7 +73,7 @@ def get_image(self, ptype, default_image=None):
         return default_image if default_image else self.image
 
     def create_release(self, user, *args, **kwargs):
-        latest_release = self.app.release_set.filter(failed=False).latest()
+        latest_release = Release.latest(self.app)
         try:
             new_release = latest_release.new(
                 user,
@@ -109,7 +110,7 @@ def _get_or_create_config(self):
         """
         dryccfile to config
         """
-        latest_release = self.app.release_set.filter(failed=False).latest()
+        latest_release = Release.latest(self.app)
         config_values, config_values_ref, config_healthcheck = [], {}, {}
         for group, values in self.dryccfile.get('config', {}).items():
             for value in values:

rootfs/api/models/config.py

@@ -125,9 +125,10 @@ def save(self, ignore_update_fields=None, *args, **kwargs):
         """merge the old config with the new"""
         try:
             # Get config from the latest available release
-            try:
-                previous_config = self.app.release_set.filter(failed=False).latest().config
-            except Release.DoesNotExist:
+            latest_releases = Release.latest(self.app)
+            if latest_releases:
+                previous_config = latest_releases.config
+            else:
                 # If that doesn't exist then fallback on app config
                 # usually means a totally new app
                 previous_config = self.app.config_set.latest()

rootfs/api/models/release.py

@@ -211,28 +211,31 @@ def deploy(self, ptypes=None, force_deploy=False):
             raise DryccException(f"{msg}: {lock.locked(ptypes)}")
         run_pipeline.delay(self, ptypes, force_deploy)
 
-    def previous(self, state="succeed"):
-        """
-        Return the previous Release to this one.
-
-        :return: the previous :class:`Release`, or None
-        """
-        releases = self.app.release_set
-        if self.pk:
-            releases = releases.exclude(pk=self.pk)
+    @classmethod
+    def latest(cls, app, state=None, exclude_pk=None):
+        releases = app.release_set
+        if exclude_pk:
+            releases = releases.exclude(pk=exclude_pk)
         q = Q(failed=False, state="succeed") if state else Q(failed=False)
         try:
-            app_settings = self.app.appsettings_set.latest()
+            app_settings = app.appsettings_set.latest()
             if app_settings.autorollback is False:
-                q = Q(state__in=["crashed", "succeed"])
+                q = Q()
         except AppSettings.DoesNotExist:
             pass
         try:
             # Get the Release previous to this one
-            prev_release = releases.filter(q).latest()
+            release = releases.filter(q).latest()
         except Release.DoesNotExist:
-            prev_release = None
-        return prev_release
+            release = None
+        return release
+
+    def previous(self, state="succeed"):
+        """
+        Return the previous Release to this one.
+        :return: the previous :class:`Release`, or None
+        """
+        return self.latest(self.app, state, self.pk)
 
     def rollback(self, user, ptypes=None, version=None):
         try:

rootfs/api/views.py

@@ -279,7 +279,7 @@ def get_object(self):
             release = get_object_or_404(
                 models.release.Release, app=self.get_app(), version=int(version))
         else:
-            release = self.get_app().release_set.filter(failed=False).latest()
+            release = models.release.Release.latest(self.get_app())
         return getattr(release, self.model.__name__.lower())
 
 
@@ -321,7 +321,7 @@ def run(self, request, **kwargs):
             expires = settings.KUBERNETES_JOB_MAX_TTL_SECONDS_AFTER_FINISHED
         if not command:
             raise DryccException('command is a required field, or it can be defined in Procfile')
-        release = app.release_set.filter(failed=False).latest()
+        release = models.release.Release.latest(app)
         if release.build is None:
             raise DryccException('no build available, please deploy a release')
         volumes = request.data.get('volumes', None)
@@ -417,7 +417,7 @@ def delete(self, request, **kwargs):
         return Response(status=status.HTTP_200_OK)
 
     def post_save(self, config):
-        latest_release = config.app.release_set.filter(failed=False).latest()
+        latest_release = models.release.Release.latest(self.get_app())
         try:
             release = latest_release.new(
                 self.request.user, config=config, build=latest_release.build)
@@ -673,7 +673,7 @@ def rollback(self, request, **kwargs):
         Create a new release as a copy of the state of the compiled slug and config vars of a
         previous release.
         """
-        latest_release = self.get_app().release_set.filter(failed=False).latest()
+        latest_release = models.release.Release.latest(self.get_app())
         ptypes = set(
             [ptype for ptype in request.data.get("ptypes", "").split(",") if ptype])
         ptypes = latest_release.app.check_ptypes(ptypes)
@@ -786,7 +786,7 @@ def create(self, request, *args, **kwargs):
         # return the application databag
         response = {
             'release': {
-                'version': app.release_set.filter(failed=False).latest().version
+                'version': models.release.Release.latest(app).version
             }
         }
         return Response(response, status=status.HTTP_200_OK)
@@ -807,7 +807,7 @@ def create(self, request, *args, **kwargs):
         has_permission, message = permissions.has_app_permission(request.user, app, request.method)
         if not has_permission:
             return Response(message, status=status.HTTP_403_FORBIDDEN)
-        config = app.release_set.filter(failed=False).latest().config
+        config = models.release.Release.latest(app).config
         serializer = self.get_serializer(config)
         return Response(serializer.data, status=status.HTTP_200_OK)