Handle script parsers whose script versions have been deleted

anukaal · Apr 12, 2020 · 2866280 · 2866280
1 parent a295461
commit 2866280
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/wooey/models/core.py b/wooey/models/core.py
@@ -300,7 +300,8 @@ class ScriptParser(WooeyPy2Mixin, models.Model):
     script_version = models.ManyToManyField('ScriptVersion')
 
     def __str__(self):
-        return '{}: {}'.format(self.script_version.first().script.script_name, self.name)
+        script_version = self.script_version.first()
+        return '{}: {}'.format(script_version.script.script_name if script_version else 'No Script Assigned', self.name)
 
 
 class ScriptParameter(UpdateScriptsMixin, WooeyPy2Mixin, models.Model):

diff --git a/wooey/tests/test_models.py b/wooey/tests/test_models.py
@@ -45,6 +45,11 @@ class ScriptGroupTestCase(TestCase):
     def test_script_group_creation(self):
         group = factories.ScriptGroupFactory()
 
+class TestScriptParsers(mixins.ScriptFactoryMixin, TestCase):
+    def test_renders_if_script_version_deleted(self):
+        parser = self.choice_script.scriptparser_set.first()
+        self.choice_script.delete()
+        self.assertIn(parser.name, str(parser))
 
 class ScriptParameterTestCase(TestCase):
     def test_script_parameter_default(self):