sqlServerConnector 0.1.4__tar.gz → 0.1.5__tar.gz

{sqlserverconnector-0.1.4 → sqlserverconnector-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlServerConnector
-Version: 0.1.4
+Version: 0.1.5
 Summary: A custom SQL Server Connector for ETL processes with Pandas
 Author-email: Nguyen Minh Son <nguyen.minhson1511@gmail.com>
 Project-URL: Homepage, https://github.com/johnnyb1509/sqlServerConnector

{sqlserverconnector-0.1.4 → sqlserverconnector-0.1.5}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sqlServerConnector"
-version = "0.1.4"
+version = "0.1.5"
 description = "A custom SQL Server Connector for ETL processes with Pandas"
 readme = "README.md"
 requires-python = ">=3.8"

{sqlserverconnector-0.1.4 → sqlserverconnector-0.1.5}/src/connector.py

@@ -177,4 +177,46 @@ class SQLServerConnector:
                 for col in new_cols:
                     sql_type = "NVARCHAR(MAX)" if df[col].dtype == 'object' else "FLOAT"
                     conn.execute(text(f"ALTER TABLE [{table_name}] ADD [{col}] {sql_type} NULL"))
-                conn.commit()
+                conn.commit()
+
+    # ========================================================
+    # DATA RETRIEVAL METHODS
+    # ========================================================
+    def get_data(self,  query: str, params: Optional[Dict[str, Any]] = None, chunksize: Optional[int] = None) -> Union[pd.DataFrame, Any]:
+        """
+        Executes a SQL query and returns a Pandas DataFrame.
+        
+        Args:
+            query (str): The SQL query string. Use :param_name for parameters.
+            params (dict, optional): Dictionary of parameters to bind to the query.
+            chunksize (int, optional): If specified, returns an iterator where each chunk is the given size.
+        
+        Returns:
+            pd.DataFrame or Iterator[pd.DataFrame]
+        """
+        try:
+            with self.engine.connect() as conn:
+                # Use text() explicitly for SQLAlchemy 2.0 compatibility
+                sql_query = text(query)
+                
+                # If chunksize is provided, read_sql returns a generator
+                result = pd.read_sql(
+                    sql_query, 
+                    conn, 
+                    params=params, 
+                    chunksize=chunksize
+                )
+                
+                if chunksize is None:
+                    logger.info(f"Successfully retrieved {len(result)} rows.")
+                else:
+                    logger.info(f"Retrieving data in chunks of {chunksize} rows.")
+                    
+                return result
+                
+        except SQLAlchemyError as e:
+            logger.error(f"Failed to retrieve data: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"An unexpected error occurred during get_data: {e}")
+            raise

{sqlserverconnector-0.1.4 → sqlserverconnector-0.1.5}/src/sqlServerConnector.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlServerConnector
-Version: 0.1.4
+Version: 0.1.5
 Summary: A custom SQL Server Connector for ETL processes with Pandas
 Author-email: Nguyen Minh Son <nguyen.minhson1511@gmail.com>
 Project-URL: Homepage, https://github.com/johnnyb1509/sqlServerConnector