Merge pull request #18 from ntoll/master

Ensure source_uuid is added to Submission objects.
freedomofpress · Sep 26, 2018 · bb9ddc9 · bb9ddc9
2 parents f18264d + e2060fb
commit bb9ddc9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sdclientapi/__init__.py b/sdclientapi/__init__.py
@@ -92,6 +92,7 @@ def __init__(self, **kwargs) -> None:
         self.is_read = False  # type: bool
         self.size = 0  # type: int
         self.source_url = ""  # type: str
+        self.source_uuid = ""  # type: str
         self.submission_url = ""  # type: str
         self.uuid = ""  # type: str
 
@@ -112,6 +113,7 @@ def __init__(self, **kwargs) -> None:
             if not key in kwargs:
                 AttributeError("Missing key {}".format(key))
             setattr(self, key, kwargs[key])
+        _, self.source_uuid = self.source_url.rsplit('/', 1)
 
 
 class Source: