Set the stream when submitting a document

 - Legacy-Id: 3512

Author: OleLaursen

ietf/submit/tests.py

@@ -134,6 +134,7 @@ def test_submit_new(self):
         self.assertEquals(new_revision.by.name, "Test Name")
         self.assertTrue(not os.path.exists(os.path.join(self.staging_dir, u"%s-%s.txt" % (name, rev))))
         self.assertTrue(os.path.exists(os.path.join(self.repository_dir, u"%s-%s.txt" % (name, rev))))
+        self.assertEquals(draft.stream_id, "ietf")
         self.assertEquals(draft.authors.count(), 1)
         self.assertEquals(draft.authors.all()[0].get_name(), "Test Name")
         self.assertEquals(draft.authors.all()[0].address, "testname@example.com")
@@ -200,6 +201,7 @@ def test_submit_existing(self):
         self.assertEquals(new_revision.by.name, "Test Name")
         self.assertTrue(not os.path.exists(os.path.join(self.staging_dir, u"%s-%s.txt" % (name, rev))))
         self.assertTrue(os.path.exists(os.path.join(self.repository_dir, u"%s-%s.txt" % (name, rev))))
+        self.assertEquals(draft.stream_id, "ietf")
         self.assertEquals(draft.authors.count(), 1)
         self.assertEquals(draft.authors.all()[0].get_name(), "Test Name")
         self.assertEquals(draft.authors.all()[0].address, "testname@example.com")

ietf/submit/utils.py

@@ -109,6 +109,17 @@ def perform_postREDESIGN(submission):
     draft.abstract = submission.abstract
     was_rfc = draft.state_id == "rfc"
     draft.state_id = "active"
+
+    if draft.name.startswith("draft-iab-"):
+        stream_slug = "iab"
+    elif draft.name.startswith("draft-irtf-"):
+        stream_slug = "irtf"
+    elif not draft.group_id or draft.group.type_id == "individ":
+        stream_slug = "ise"
+    else:
+        stream_slug = "ietf"
+
+    draft.stream = DocStreamName.objects.get(slug=stream_slug)
     draft.save()
 
     DocAlias.objects.get_or_create(name=submission.filename, document=draft)