DB2Azure

DB2Azure

DB2Azure is a Python package designed to streamline the process of loading data from SQL Server (MSSQL), PostgreSQL, and MySQL databases to Azure Blob Storage in both JSON and CSV formats. This package simplifies the data extraction and upload processes with separate modules for SQL Server (MSSQLLoader), PostgreSQL (PostgreLoader), and MySQL (MySQLLoader), enabling efficient and seamless integration with Azure Blob Storage.

Table of Contents

• Description
• Installation
• Usage
  • SQL Server Loader
  • PostgreSQL Loader
  • MySQL Loader
• Methods
  • MSSQLLoader
  • PostgreLoader
  • MySQLLoader
• Configuration
• Error Handling
• License
• Contributing
• Acknowledgements

Description

DB2Azure helps automate the process of extracting data from SQL Server, PostgreSQL, and MySQL databases and uploading it directly to Azure Blob Storage in either JSON or CSV format. The package includes three main modules for SQL Server (MSSQLLoader), PostgreSQL (PostgreLoader), and MySQL (MySQLLoader), each providing methods for executing SQL queries and transferring the data to Azure Blob Storage.

Key Features:

• SQL Server Support: Extracts data from Microsoft SQL Server databases using pyodbc.
• PostgreSQL Support: Extracts data from PostgreSQL databases using psycopg.
• MySQL Support: Extracts data from MySQL databases using pymysql.
• Azure Blob Storage Upload: Uploads data as JSON or CSV to Azure Blob Storage using the azure-storage-blob.
• Flexibility: Allows customization of folder path, file name, and blob URL.
• Error Handling: Provides detailed error messages in case of failures.

Installation

To install the DB2Azure package, use the following pip command:

pip install DB2Azure

Alternatively, clone the repository and install it manually:

git clone https://github.com/mr-speedster/DB2Azure.git
cd DB2Azure
pip install .

Usage

SQL Server Loader

To use the SQL Server loader, you can use the MSSQLLoader class in the db2azure module. The MSSQLLoader class allows you to execute SQL queries and upload the resulting data to Azure Blob Storage in either JSON or CSV format.

Example:

from db2azure import MSSQLLoader

# SQL Query
query = "SELECT [UserID],[FirstName],[LastName],[Email],[Age] FROM [SampleDB].[dbo].[Users]"

# SQL Server connection string
sql_conn = r"Driver=<driver>;Server=<server_name>;Database=<database>;Trusted_Connection=yes;"

# Azure Blob Storage configurations
azure_config_json = {
    'container_name': "your_container",
    'folder_path': "your_folder",
    'file_name': "your_file.json",
    'azure_blob_url': "https://your_account_name.blob.core.windows.net",
    'sas_token': "your_sas_token"
}

azure_config_csv = {
    'container_name': "your_container",
    'folder_path': "your_folder",
    'file_name': "your_file.csv",
    'azure_blob_url': "https://your_account_name.blob.core.windows.net",
    'sas_token': "your_sas_token"
}

# Load to JSON
json_status = MSSQLLoader.load_to_json(query, sql_conn, azure_config_json)
print("JSON Upload Status:", json_status)

# Load to CSV
csv_status = MSSQLLoader.load_to_csv(query, sql_conn, azure_config_csv)
print("CSV Upload Status:", csv_status)

PostgreSQL Loader

To use the PostgreSQL loader, you can use the PostgreLoader class in the db2azure module. The PostgreLoader class operates similarly to MSSQLLoader, but it works with PostgreSQL databases.

Example:

from db2azure import PostgreLoader

# PostgreSQL Query
# PostgreSQL Query
query = "SELECT user_id, first_name, last_name, email, age FROM public.users;"

# PostgreSQL connection parameters
connection_params = {
    "host": "localhost",      # e.g., "localhost" or an IP address
    "port": "5432",           # default PostgreSQL port
    "dbname": "SampleDB",     # name of the database
    "user": "postgres",       # PostgreSQL username
    "password": "<your_password>"  # PostgreSQL password
}

# Azure Blob Storage configurations
azure_config_json = {
    'container_name': "your_container",
    'folder_path': "your_folder",
    'file_name': "your_file.json",
    'azure_blob_url': "https://your_account_name.blob.core.windows.net",
    'sas_token': "your_sas_token"
}

azure_config_csv = {
    'container_name': "your_container",
    'folder_path': "your_folder",
    'file_name': "your_file.csv",
    'azure_blob_url': "https://your_account_name.blob.core.windows.net",
    'sas_token': "your_sas_token"
}

# Load to JSON
json_status = PostgreLoader.load_to_json(query, connection_params, azure_config_json)
print("JSON Upload Status:", json_status)

# Load to CSV
csv_status = PostgreLoader.load_to_csv(query, connection_params, azure_config_csv)
print("CSV Upload Status:", csv_status)

MySQL Loader

To use the MySQL loader, you can use the MySQLLoader class in the db2azure module. The MySQLLoader class works similarly to MSSQLLoader and PostgreLoader, but it is designed to work with MySQL databases.

Example:

from db2azure import MySQLLoader

# SQL Query
query = "SELECT * FROM SampleDB.Users"

# MySQL connection parameters
mysql_conn = {
    "host": "localhost",      # e.g., "localhost" or an IP address
    "port": "3306",           # default MySQL port
    "database": "SampleDB",   # name of the database
    "user": "*****",          # MySQL username
    "password": "*****"       # MySQL password
}

# Azure Blob Storage configurations
azure_config_json = {
    'container_name': "your_container",
    'folder_path': "your_folder",
    'file_name': "your_file.json",
    'azure_blob_url': "https://your_account_name.blob.core.windows.net",
    'sas_token': "your_sas_token"
}

azure_config_csv = {
    'container_name': "your_container",
    'folder_path': "your_folder",
    'file_name': "your_file.csv",
    'azure_blob_url': "https://your_account_name.blob.core.windows.net",
    'sas_token': "your_sas_token"
}

# Load to JSON
json_status = MySQLLoader.load_to_json(query, mysql_conn, azure_config_json)
print("JSON Upload Status:", json_status)

# Load to CSV
csv_status = MySQLLoader.load_to_csv(query, mysql_conn, azure_config_csv)
print("CSV Upload Status:", csv_status)

Methods

MSSQLLoader

• load_to_json: Loads data from SQL Server to a JSON file in Azure Blob Storage.
  • Parameters: sql_query, connection_string, azure_config
• load_to_csv: Loads data from SQL Server to a CSV file in Azure Blob Storage.
  • Parameters: sql_query, connection_string, azure_config

PostgreLoader

• load_to_json: Loads data from PostgreSQL to a JSON file in Azure Blob Storage.
  • Parameters: sql_query, connection_params, azure_config
• load_to_csv: Loads data from PostgreSQL to a CSV file in Azure Blob Storage.
  • Parameters: sql_query, connection_params, azure_config

MySQLLoader

• load_to_json: Loads data from MySQL to a JSON file in Azure Blob Storage.
  • Parameters: sql_query, connection_params, azure_config
• load_to_csv: Loads data from MySQL to a CSV file in Azure Blob Storage.
  • Parameters: sql_query, connection_params, azure_config

Configuration

For each loader (SQL Server, PostgreSQL, MySQL), you will need to provide the following configuration:

• SQL Server: Use the connection_string parameter to configure the connection to your SQL Server.
• PostgreSQL: Use the connection_params dictionary to configure the connection to your PostgreSQL database.
• MySQL: Use the connection_params dictionary to configure the connection to your MySQL database.
• Azure Blob Storage: Provide the azure_config dictionary, containing container_name, folder_path, file_name, azure_blob_url, and sas_token, to specify where and how the data should be uploaded to Azure Blob Storage.

Error Handling

If any error occurs during the data extraction or upload process, the methods will return an error response containing:

• status: Always error.
• message: The error message describing the issue.

Example error response:

{
  "status": "error",
  "message": "Connection failed: 'Your error message here'"
}

License

DB2Azure is licensed under the MIT License. See the LICENSE file for more details.

Contributing

We welcome contributions! Feel free to open an issue or submit a pull request for any improvements or bug fixes.

Acknowledgements

• pyodbc: A Python DB API 2.0 interface for ODBC databases, used for connecting to SQL Server.
• psycopg: A PostgreSQL database adapter for Python, used for connecting to PostgreSQL.
• pymysql: MySQL database adapter for Python, used for connecting to MySQL.
• azure-storage-blob: Azure SDK for Python, used for uploading files to Azure Blob Storage.