rrs: default python2/3 environments for new maintenance plan layer branches

It's a bit of a pain to have to set the two python environment fields on
every record in order to have things set correctly, and it can easily
get forgotten, so try to set them automatically by default (assuming
reasonable naming).

Note that this does introduce an annoying behaviour whereby if you click
"Add another Maintenance plan layer branch" and then decide you don't
want it, the admin form will insist you fill in the fields unless you
clear out the python2/3 environment fields. I'm not sure how to fix
that, so I'm leaving it as-is for now.

Signed-off-by: Paul Eggleton <paul.eggleton@linux.intel.com>

1 files changed, 14 insertions, 0 deletions

diff --git a/layerindex/models.py b/layerindex/models.py
index e9751f3b6a..ff164baaea 100644
--- a/layerindex/models.py
+++ b/layerindex/models.py
@@ -47,6 +47,20 @@ class PythonEnvironment(models.Model):
             cmd = self.python_command
         return cmd
 
+    @staticmethod
+    def get_default_python2_environment():
+        for env in PythonEnvironment.objects.all().order_by('id'):
+            if env.name.replace(' ', '').lower().startswith(('python2', 'py2')):
+                return env
+        return None
+
+    @staticmethod
+    def get_default_python3_environment():
+        for env in PythonEnvironment.objects.all().order_by('id'):
+            if env.name.replace(' ', '').lower().startswith(('python3', 'py3')):
+                return env
+        return None
+
     def __str__(self):
         return self.name