Added convenience function to extract SPDX elements by their Id

HarshvMahawar · HarshvMahawar · commit 34b8bc842dfc · 2023-03-23T14:52:25.000+05:30
Signed-off-by: HarshvMahawar &lt;hv062727@gmail.com&gt;
diff --git a/src/spdx/document_utils.py b/src/spdx/document_utils.py
@@ -9,12 +9,25 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from typing import List
-
+from typing import Union
 from spdx.model.document import Document
+from spdx.model.snippet import Snippet
+from spdx.model.package import Package
+from spdx.model.file import File
 
 
 def get_contained_spdx_element_ids(document: Document) -> List[str]:
     element_ids = [file.spdx_id for file in document.files]
     element_ids.extend([package.spdx_id for package in document.packages])
     element_ids.extend([snippet.spdx_id for snippet in document.snippets])
     return element_ids
+
+
+def get_element_from_spdx_id(document: Document, spdx_id: str) -> Union[Package, File, Snippet, None]:
+    elements = [file_ for file_ in document.files]
+    elements.extend([package_ for package_ in document.packages])
+    elements.extend([snippet_ for snippet_ in document.snippets])
+    for element in elements:
+        if element.spdx_id == spdx_id:
+            return element
+
