rand-engine 0.3.7__tar.gz → 0.3.8__tar.gz

{rand_engine-0.3.7 → rand_engine-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: rand-engine
-Version: 0.3.7
+Version: 0.3.8
 Summary: Rand Engine v2. Package with some methods to generate random data in different formats. Great to mock data while testing or developing.
 Author: marcoaureliomenezes
 Author-email: marcoaurelioreislima@gmail.com
@@ -10,6 +10,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: fastavro (>=1.10.0,<2.0.0)
 Requires-Dist: fastparquet (>=2024.11.0,<2025.0.0)
 Requires-Dist: numpy (>=2.1.1,<3.0.0)
@@ -43,6 +44,6 @@ CoreDistinct().randint(0, 100, 10)
 
 ```
 
-git tag 0.0.1
+git tag  0.3.8
 
 git push origin --tags

{rand_engine-0.3.7 → rand_engine-0.3.8}/README.md

@@ -24,6 +24,6 @@ CoreDistinct().randint(0, 100, 10)
 
 ```
 
-git tag 0.0.1
+git tag  0.3.8
 
 git push origin --tags

{rand_engine-0.3.7 → rand_engine-0.3.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rand-engine"
-version = "0.3.7"
+version = "0.3.8"
 description = "Rand Engine v2. Package with some methods to generate random data in different formats. Great to mock data while testing or developing."
 authors = ["marcoaureliomenezes <marcoaurelioreislima@gmail.com>"]
 repository = "https://github.com/marcoaureliomenezes/rand_engine"

{rand_engine-0.3.7 → rand_engine-0.3.8}/rand_engine/main/file_writer.py

@@ -12,7 +12,8 @@ class FileWriter:
     self.write_options = {}
     self.dict_format = {
         "csv": self.to_csv,
-        "parquet": self.to_parquet
+        "parquet": self.to_parquet,
+        "json": self.to_json
     }
 
   def __handle_fs(self, path, flag=True) -> None:
@@ -89,6 +90,17 @@ class FileWriter:
     writer = lambda: dataframe().to_csv(full_path, index=False, **self.write_options)
     return writer
   
+  def to_json(self, dataframe, full_path) -> Callable:
+    """
+    This method writes a pandas DataFrame to a json file.
+    :param dataframe: pd.DataFrame: DataFrame to be written.
+    :param full_path: str: Full path of the file to be written.
+    :return: Callable: Function to write the Pandas DataFrame to a json file.
+    """
+    if self.write_options.get("compression"):
+      full_path= full_path.replace("json", f"json.{self.write_options['compression']}")
+    writer = lambda: dataframe().to_json(full_path, index=False, **self.write_options)
+    return writer
 
   def to_parquet(self, dataframe, full_path):
     """
@@ -110,6 +122,7 @@ class FileWriter:
     """
     self.__handle_fs(path)
     dataframe = self.microbatch_def()
+    print(dataframe)
     self.dict_format[self.write_format](dataframe, path)()