Merge pull request #229 from azwanzuharimi/feature/aws_athena

Feat: Added AWS Athena data loader

Author: Chenglong-MS

py-src/data_formulator/data_loader/README.md

@@ -22,8 +22,15 @@ The UI automatically provide the query completion option to help user generate q
 
 ### Example Implementations
 
+- `AthenaDataLoader`: AWS Athena integration (SQL queries on S3 data lakes)
+- `BigQueryDataLoader`: Google BigQuery integration
 - `KustoDataLoader`: Azure Data Explorer (Kusto) integration
 - `MySQLDataLoader`: MySQL database integration
+- `PostgreSQLDataLoader`: PostgreSQL database integration
+- `MSSQLDataLoader`: Microsoft SQL Server integration
+- `S3DataLoader`: Amazon S3 file integration (CSV, Parquet, JSON)
+- `AzureBlobDataLoader`: Azure Blob Storage integration
+- `MongoDBDataLoader`: MongoDB integration
 
 ### Testing
 

py-src/data_formulator/data_loader/__init__.py

@@ -7,6 +7,7 @@
 from data_formulator.data_loader.postgresql_data_loader import PostgreSQLDataLoader
 from data_formulator.data_loader.mongodb_data_loader import MongoDBDataLoader
 from data_formulator.data_loader.bigquery_data_loader import BigQueryDataLoader
+from data_formulator.data_loader.athena_data_loader import AthenaDataLoader
 
 DATA_LOADERS = {
     "mysql": MySQLDataLoader,
@@ -16,7 +17,8 @@
     "azure_blob": AzureBlobDataLoader,
     "postgresql": PostgreSQLDataLoader,
     "mongodb": MongoDBDataLoader,
-    "bigquery": BigQueryDataLoader
+    "bigquery": BigQueryDataLoader,
+    "athena": AthenaDataLoader
 }
 
-__all__ = ["ExternalDataLoader", "MySQLDataLoader", "MSSQLDataLoader", "KustoDataLoader", "S3DataLoader", "AzureBlobDataLoader","PostgreSQLDataLoader", "MongoDBDataLoader", "BigQueryDataLoader", "DATA_LOADERS"]
+__all__ = ["ExternalDataLoader", "MySQLDataLoader", "MSSQLDataLoader", "KustoDataLoader", "S3DataLoader", "AzureBlobDataLoader", "PostgreSQLDataLoader", "MongoDBDataLoader", "BigQueryDataLoader", "AthenaDataLoader", "DATA_LOADERS"]