(fix) Allow read partitions with extra = in the value (#1779)

* Allow read partitions with extra = in the value

* fix num of ='s and apply same fix for metadata calss

Author: kukushking

awswrangler/s3/_read.py

@@ -44,10 +44,10 @@ def _extract_partitions_metadata_from_paths(
             raise exceptions.InvalidArgumentValue(f"Object {p} is not under the root path ({path}).")
         path_wo_filename: str = p.rpartition("/")[0] + "/"
         if path_wo_filename not in partitions_values:
-            path_wo_prefix: str = path_wo_filename.replace(f"{path}/", "")
-            dirs: Tuple[str, ...] = tuple(x for x in path_wo_prefix.split("/") if (x != "") and (x.count("=") == 1))
+            path_wo_prefix: str = path_wo_filename.replace(f"{path}", "")
+            dirs: Tuple[str, ...] = tuple(x for x in path_wo_prefix.split("/") if (x != "") and (x.count("=") > 0))
             if dirs:
-                values_tups = cast(Tuple[Tuple[str, str]], tuple(tuple(x.split("=")[:2]) for x in dirs))
+                values_tups = cast(Tuple[Tuple[str, str]], tuple(tuple(x.split("=", maxsplit=1)[:2]) for x in dirs))
                 values_dics: Dict[str, str] = dict(values_tups)
                 p_values: List[str] = list(values_dics.values())
                 p_types: Dict[str, str] = {x: "string" for x in values_dics.keys()}
@@ -71,11 +71,11 @@ def _extract_partitions_from_path(path_root: str, path: str) -> Dict[str, str]:
     if path_root not in path:
         raise exceptions.InvalidArgumentValue(f"Object {path} is not under the root path ({path_root}).")
     path_wo_filename: str = path.rpartition("/")[0] + "/"
-    path_wo_prefix: str = path_wo_filename.replace(f"{path_root}/", "")
-    dirs: Tuple[str, ...] = tuple(x for x in path_wo_prefix.split("/") if (x != "") and (x.count("=") == 1))
+    path_wo_prefix: str = path_wo_filename.replace(f"{path_root}", "")
+    dirs: Tuple[str, ...] = tuple(x for x in path_wo_prefix.split("/") if (x != "") and (x.count("=") > 0))
     if not dirs:
         return {}
-    values_tups = cast(Tuple[Tuple[str, str]], tuple(tuple(x.split("=")[:2]) for x in dirs))
+    values_tups = cast(Tuple[Tuple[str, str]], tuple(tuple(x.split("=", maxsplit=1)[:2]) for x in dirs))
     values_dics: Dict[str, str] = dict(values_tups)
     return values_dics