Added optional variable to allow for custom root keys

Author: ddkasa

rss_parser/_parser.py

@@ -28,14 +28,17 @@ def to_xml(data: str, *args, **kwargs):
         return parse(str(data), *args, **kwargs)
 
     @classmethod
-    def parse(cls, data: str, *, schema: Optional[Type[XMLBaseModel]] = None) -> XMLBaseModel:
+    def parse(cls, data: str, *, schema: Optional[Type[XMLBaseModel]] = None, root_key: str = "") -> XMLBaseModel:
         """
         Parse XML data into schema (default: RSS 2.0).
 
         :param data: string of XML data that needs to be parsed
         :return: "schema" object
         """
         root = cls.to_xml(data)
-        root, schema = cls.check_schema(root)
+        if not isinstance(schema, XMLBaseModel):
+            root, schema = cls.check_schema(root)
+        else:
+            root = root.get(root_key, root)
 
         return schema.parse_obj(root)