AWS Redshift ETL using Pandas API
Author(s): Vivek Chaudhary
The Objective of this blog is to perform a simple ETL exercise with AWS Redshift Database. Oracle Database tables are used as the source dataset, perform simple transformations using Pandas methods on the dataset and write the dataset into AWS Redshift table.
- Import prerequisites and connection with source Oracle:
import pandas as pd
from sqlalchemy import create_engine
engine = create_engine(‘oracle://scott:[email protected]’, echo=False)
2. Extract Datasets from Oracle Database:
emp_df=pd.read_sql_query(‘select * from emp’,engine)
dept_df=pd.read_sql_query(‘select * from dept’,engine)
3. Transform Dataset
Create AWS Redshift Target Table using the below script:
create table emp (
Join the EMP and DEPT datasets:
Drop the the columns that are not present in target:
4. Create Redshift connection and insert data
#create connection object
joined_df.to_sql(‘emp’, conn, index=False, if_exists=’append’)
Verify the data in the Redshift table.
Querying the “emp” table from AWS console, we can also set up SQLWorkbench on local system to query Redshift tables.
DML operation is successful.
5. Connectivity issue I faced
OperationalError: (psycopg2.OperationalError) could not connect to server: Connection timed out (0x0000274C/10060) Is the server running on host “redshift_cluster_name.unique_here.region.redshift.amazonaws.com” (<IP address>) and accepting TCP/IP connections on port 5439?
The issue was that the inbound rule in the Security Group specified a security group as the source. Changing it to a CIDR that included my IP address fixed the issue.
How to Fix?
Go to Cluster Properties → Network Security
GO to VPC Security Group → Inbound rules →Edit inbound rules and Add both below rules → Click Save Rules.
And we are ready to go. In absence of the second rule, there might be a situation where one may face connectivity issues with AWS Redshift DB. So follow the above steps to avoid/resolve the issue.
Thanks to all for reading my blog. Do share your views or feedback.
Published via Towards AI