stouputils 1.7.0__tar.gz → 1.7.1__tar.gz

{stouputils-1.7.0 → stouputils-1.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stouputils
-Version: 1.7.0
+Version: 1.7.1
 Summary: Stouputils is a collection of utility modules designed to simplify and enhance the development process. It includes a range of tools for tasks such as execution of doctests, display utilities, decorators, as well as context managers, and many more.
 Project-URL: Homepage, https://github.com/Stoupy51/stouputils
 Project-URL: Issues, https://github.com/Stoupy51/stouputils/issues

{stouputils-1.7.0 → stouputils-1.7.1}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "stouputils"
-version = "1.7.0"
+version = "1.7.1"
 description = "Stouputils is a collection of utility modules designed to simplify and enhance the development process. It includes a range of tools for tasks such as execution of doctests, display utilities, decorators, as well as context managers, and many more."
 readme = "README.md"
 requires-python = ">=3.10"

{stouputils-1.7.0 → stouputils-1.7.1}/stouputils/io.py

@@ -190,48 +190,58 @@ def super_csv_dump(
 	if isinstance(data, str | bytes | dict):
 		raise ValueError("Data must be a list of lists, list of dicts, pandas DataFrame, or Polars DataFrame")
 	output = StringIO()
+	done: bool = False
 
 	# Handle Polars DataFrame
 	try:
-		import polars as pl  # type: ignore  # noqa: F401
-		copy_kwargs = kwargs.copy()
-		copy_kwargs.setdefault("separator", delimiter)
-		copy_kwargs.setdefault("include_header", has_header)
-		copy_kwargs.setdefault("line_terminator", "\r\n")
-		data.write_csv(output, *args, **copy_kwargs)
+		import polars as pl  # type: ignore
+		if isinstance(data, pl.DataFrame):
+			copy_kwargs = kwargs.copy()
+			copy_kwargs.setdefault("separator", delimiter)
+			copy_kwargs.setdefault("include_header", has_header)
+			data.write_csv(output, *args, **copy_kwargs)
+			done = True
 	except Exception:
 		pass
 
 	# Handle pandas DataFrame
-	try:
-		import pandas as pd  # type: ignore  # noqa: F401
-		copy_kwargs = kwargs.copy()
-		copy_kwargs.setdefault("index", index)
-		copy_kwargs.setdefault("sep", delimiter)
-		copy_kwargs.setdefault("header", has_header)
-		copy_kwargs.setdefault("lineterminator", "\r\n")
-		data.to_csv(output, *args, **copy_kwargs)
-	except Exception:
-		pass
+	if not done:
+		try:
+			import pandas as pd  # type: ignore
+			if isinstance(data, pd.DataFrame):
+				copy_kwargs = kwargs.copy()
+				copy_kwargs.setdefault("index", index)
+				copy_kwargs.setdefault("sep", delimiter)
+				copy_kwargs.setdefault("header", has_header)
+				data.to_csv(output, *args, **copy_kwargs)
+		except Exception:
+			pass
+
+	if not done:
+		# Handle list of dicts
+		if isinstance(data[0], dict):
+			fieldnames = list(data[0].keys()) # type: ignore
+			kwargs.setdefault("fieldnames", fieldnames)
+			kwargs.setdefault("delimiter", delimiter)
+			dict_writer = csv.DictWriter(output, *args, **kwargs)
+			if has_header:
+				dict_writer.writeheader()
+			dict_writer.writerows(data)  # type: ignore
+			done = True
 
-	# Handle list of dicts
-	if isinstance(data[0], dict):
-		fieldnames = list(data[0].keys())
-		kwargs.setdefault("fieldnames", fieldnames)
-		kwargs.setdefault("delimiter", delimiter)
-		kwargs.setdefault("lineterminator", "\r\n")
-		dict_writer = csv.DictWriter(output, *args, **kwargs)
-		if has_header:
-			dict_writer.writeheader()
-		dict_writer.writerows(data)  # type: ignore
-
-	# Handle list of lists
-	else:
-		kwargs.setdefault("delimiter", delimiter)
-		kwargs.setdefault("lineterminator", "\r\n")
-		list_writer = csv.writer(output, *args, **kwargs)
-		list_writer.writerows(data)
+		# Handle list of lists
+		else:
+			kwargs.setdefault("delimiter", delimiter)
+			list_writer = csv.writer(output, *args, **kwargs)
+			list_writer.writerows(data) # type: ignore
+			done = True
 
+	# If still not done, raise error
+	if not done:
+		output.close()
+		raise ValueError(f"Data must be a list of lists, list of dicts, pandas DataFrame, or Polars DataFrame, got {type(data)} instead")
+
+	# Get content and write to file if needed
 	content: str = output.getvalue()
 	if file:
 		if isinstance(file, str):
@@ -274,6 +284,8 @@ def super_csv_load(file_path: str, delimiter: str = ',', has_header: bool = True
 			0  1  2  3
 			1  4  5  6
 
+		.. code-block:: console
+
 			> super_csv_load("test.csv", as_dataframe=True, use_polars=True)
 			shape: (2, 3)
 			┌─────┬─────┬─────┐

{stouputils-1.7.0 → stouputils-1.7.1}/stouputils/io.pyi

@@ -119,6 +119,8 @@ def super_csv_load(file_path: str, delimiter: str = ',', has_header: bool = True
 \t\t\t0  1  2  3
 \t\t\t1  4  5  6
 
+\t\t.. code-block:: console
+
 \t\t\t> super_csv_load("test.csv", as_dataframe=True, use_polars=True)
 \t\t\tshape: (2, 3)
 \t\t\t┌─────┬─────┬─────┐