commit 4f5d4af7a6bc1211e7c376f18e0d0ff2bac88bb2
author Gregory Nisbet <gregorynisbet@google.com> Mon Dec 09 15:21:47 2019 -0800
committer Commit Bot <commit-bot@chromium.org> Tue Dec 10 06:46:43 2019 +0000
tree 863e65473a069f64d06249e05364a0470f3f7985
parent fdc484e03812463af9e6962df1938f473e8c4c6f

[stableversion] [autotest] provide workaround for manually updating stable version from cli

BUG=chromium:1030463
TEST=None
Change-Id: I3ac405d429e14fd291959194304ffdec3fdbe29d
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/third_party/autotest/+/1957840
Tested-by: Gregory Nisbet <gregorynisbet@google.com>
Auto-Submit: Gregory Nisbet <gregorynisbet@google.com>
Reviewed-by: Xixuan Wu <xixuan@chromium.org>
Commit-Queue: Gregory Nisbet <gregorynisbet@google.com>

frontend/afe/models.py

diff --git a/frontend/afe/models.py b/frontend/afe/models.py
index 963558a..dc6ed98 100644
--- a/frontend/afe/models.py
+++ b/frontend/afe/models.py
@@ -1,9 +1,11 @@
 # pylint: disable=missing-docstring
 
 import contextlib
-import logging
 from datetime import datetime
 from datetime import timedelta
+import logging
+import os
+
 import django.core
 try:
     from django.db import models as dbmodels, connection
@@ -2298,7 +2300,13 @@
         db_table = 'afe_stable_versions'
 
     def save(self, *args, **kwargs):
-        raise RuntimeError("the ability to save StableVersions has been intentionally removed")
+        if os.getenv("OVERRIDE_STABLE_VERSION_BAN"):
+            super(StableVersion, self).save(*args, **kwargs)
+        else:
+            raise RuntimeError("the ability to save StableVersions has been intentionally removed")
 
     def delete(self):
-        raise RuntimeError("the ability to delete StableVersions has been intentionally removed")
+        if os.getenv("OVERRIDE_STABLE_VERSION_BAN"):
+            super(StableVersion, self).delete(*args, **kwargs)
+        else:
+            raise RuntimeError("the ability to delete StableVersions has been intentionally removed")