Added convenience function to extract SPDX elements by their Id

Signed-off-by: HarshvMahawar <[email protected]>

Author: HarshvMahawar

src/spdx/document_utils.py

@@ -8,13 +8,27 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import List
 
+
+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