Skip to content

Commit

Permalink
Add dagster and dbt folders
Browse files Browse the repository at this point in the history
  • Loading branch information
maximearmstrong committed Jul 23, 2024
0 parents commit cb374ba
Show file tree
Hide file tree
Showing 26 changed files with 946 additions and 0 deletions.
188 changes: 188 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,188 @@
.venv/
.vscode/
bin/
__pycache__
pyvenv.cfg
*.egg-info
.ipynb_checkpoints/
.bashrc
.DS_STORE

# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

# C extensions
*.so

# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST

# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec

# Installer logs
pip-log.txt
pip-delete-this-directory.txt

# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py,cover
.hypothesis/
.pytest_cache/
cover/

# Translations
*.mo
*.pot

# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal

# Flask stuff:
instance/
.webassets-cache

# Scrapy stuff:
.scrapy

# Sphinx documentation
docs/_build/

# PyBuilder
.pybuilder/
target/

# Jupyter Notebook
.ipynb_checkpoints

# IPython
profile_default/
ipython_config.py

# pyenv
# For a library or package, you might want to ignore these files since the code is
# intended to run in multiple environments; otherwise, check them in:
# .python-version

# pipenv
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
# However, in case of collaboration, if having platform-specific dependencies or dependencies
# having no cross-platform support, pipenv may install dependencies that don't work, or not
# install all needed dependencies.
#Pipfile.lock

# poetry
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
#poetry.lock

# pdm
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
#pdm.lock
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
# in version control.
# https://pdm.fming.dev/#use-with-ide
.pdm.toml

# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
__pypackages__/

# Celery stuff
celerybeat-schedule
celerybeat.pid

# SageMath parsed files
*.sage.py

# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/

# Spyder project settings
.spyderproject
.spyproject

# Rope project settings
.ropeproject

# mkdocs documentation
/site

# mypy
.mypy_cache/
.dmypy.json
dmypy.json

# Pyre type checker
.pyre/

# pytype static type analyzer
.pytype/

# Cython debug symbols
cython_debug/

# PyCharm
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
# and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/

tmp*/

*.duckdb
*.geojson
*.csv
*.parquet

data/outputs/*
data/raw/*
data/requests/*.json
data/staging/*

*.user.yml

*/package-lock.yml

.idea
4 changes: 4 additions & 0 deletions dagster_cloud.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
locations:
- code_source:
module_name: jaffle_dagster.jaffle_dagster.definitions
location_name: jaffle_dagster
Empty file.
10 changes: 10 additions & 0 deletions jaffle_dagster/jaffle_dagster/assets.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
from dagster import AssetExecutionContext
from dagster_dbt import DbtCliResource, dbt_assets

from .project import jaffle_shop_project


@dbt_assets(manifest=jaffle_shop_project.manifest_path)
def jaffle_shop_dbt_assets(context: AssetExecutionContext, dbt: DbtCliResource):
yield from dbt.cli(["build"], context=context).stream()

14 changes: 14 additions & 0 deletions jaffle_dagster/jaffle_dagster/definitions.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
from dagster import Definitions
from dagster_dbt import DbtCliResource

from .assets import jaffle_shop_dbt_assets
from .project import jaffle_shop_project
from .schedules import schedules

defs = Definitions(
assets=[jaffle_shop_dbt_assets],
schedules=schedules,
resources={
"dbt": DbtCliResource(project_dir=jaffle_shop_project),
},
)
9 changes: 9 additions & 0 deletions jaffle_dagster/jaffle_dagster/project.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
from pathlib import Path

from dagster_dbt import DbtProject

jaffle_shop_project = DbtProject(
project_dir=Path(__file__).joinpath("..", "..", "..", "jaffle_shop").resolve(),
packaged_project_dir=Path(__file__).joinpath("..", "..", "dbt-project").resolve(),
)
jaffle_shop_project.prepare_if_dev()
15 changes: 15 additions & 0 deletions jaffle_dagster/jaffle_dagster/schedules.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
"""
To add a daily schedule that materializes your dbt assets, uncomment the following lines.
"""
from dagster_dbt import build_schedule_from_dbt_selection

from .assets import jaffle_shop_dbt_assets

schedules = [
# build_schedule_from_dbt_selection(
# [jaffle_shop_dbt_assets],
# job_name="materialize_dbt_models",
# cron_schedule="0 0 * * *",
# dbt_select="fqn:*",
# ),
]
7 changes: 7 additions & 0 deletions jaffle_dagster/pyproject.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[tool.dagster]
module_name = "jaffle_dagster.definitions"
code_location_name = "jaffle_dagster"
23 changes: 23 additions & 0 deletions jaffle_dagster/setup.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
from setuptools import find_packages, setup

setup(
name="jaffle_dagster",
version="0.0.1",
packages=find_packages(),
package_data={
"jaffle_dagster": [
"dbt-project/**/*",
],
},
install_requires=[
"dagster",
"dagster-cloud",
"dagster-dbt",
"dbt-duckdb<1.9",
],
extras_require={
"dev": [
"dagster-webserver",
]
},
)
5 changes: 5 additions & 0 deletions jaffle_shop/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@

target/
dbt_modules/
logs/
**/.DS_Store
Loading

0 comments on commit cb374ba

Please sign in to comment.