yet (another spark) etl framework
Project description
YETL
Website & Docs: Yet (another Apache Spark) ETL Framework
Example:
Define a dataflow
from yetl_flow import (
yetl_flow,
IDataflow,
Context,
Timeslice,
TimesliceUtcNow,
OverwriteSave,
Save
)
from pyspark.sql.functions import *
from typing import Type
@yetl_flow(log_level="ERROR")
def customer_landing_to_rawdb_csv(
context: Context,
dataflow: IDataflow,
timeslice: Timeslice = TimesliceUtcNow(),
save_type: Type[Save] = None
) -> dict:
"""Load the demo customer data as is into a raw delta hive registered table."""
# the config for this dataflow has 2 landing sources that are joined
# and written to delta table
# delta tables are automatically created and if configured schema exceptions
# are loaded syphened into a schema exception table
df_cust = dataflow.source_df("landing.customer")
df_prefs = dataflow.source_df("landing.customer_preferences")
context.log.info("Joining customers with customer_preferences")
df = df_cust.join(df_prefs, "id", "inner")
df = df_cust
dataflow.destination_df("raw.customer", df)
Run an incremental load:
timeslice = Timeslice(2022, 7, 12)
results = customer_landing_to_rawdb_csv(
timeslice = Timeslice(2022, 7, 12)
)
Run a full load:
results = customer_landing_to_rawdb_csv(
timeslice = Timeslice(2022, '*', '*'),
save_type = OverwriteSave
)
Dependencies & Setup
This is a spark application with DeltaLake it requires following dependencies installed in order to run locally:
Ensure that the spark home path and is added to youy path is set Eg:
export SPARK_HOME="$HOME/opt/spark-3.2.2-bin-hadoop3.3"
Enable DeltaLake by:
cp $SPARK_HOME/conf/spark-defaults.conf.template $SPARK_HOME/conf/spark-defaults.conf
Add the following to spark-defaults.conf
:
spark.jars.packages io.delta:delta-core_2.12:2.0.0
spark.sql.extensions io.delta.sql.DeltaSparkSessionExtension
spark.sql.catalog.spark_catalog org.apache.spark.sql.delta.catalog.DeltaCatalog
spark.sql.catalogImplementation hive
Python Project Setup
Create virual environment and install dependencies for local development:
python -m venv venv
source venv/bin/activate
pip install --upgrade pip
pip install -r requirements.txt
pip install --editable .
Build
Build python wheel:
python setup.py sdist bdist_wheel
There is a CI build configured for this repo that builds on main origin and publishes to PyPi.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
yetl-framework-0.0.5.tar.gz
(11.0 kB
view hashes)
Built Distribution
Close
Hashes for yetl_framework-0.0.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1cb1850003469d200a8d876946a52c0e3667ea417316db6c57c8a2c3916f0880 |
|
MD5 | 1bb954ceecc1ddf824d11d9181435212 |
|
BLAKE2b-256 | e9ebc9f94a2244c44aac15b4ee8f63e5b2a58c028977906fb201e8450c3b42d0 |