Skip to content

Commit

Permalink
add more dbt commands
Browse files Browse the repository at this point in the history
  • Loading branch information
MiConnell committed Sep 3, 2024
1 parent 8bcdd32 commit 6570446
Show file tree
Hide file tree
Showing 2 changed files with 68 additions and 2 deletions.
15 changes: 14 additions & 1 deletion lua/dbtpal/init.lua
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,20 @@ M.test_model = main.test_model

M.compile = main.compile
M.build = main.build

M.list = main.list
M.docs_generate = main.docs_generate
M.docs_serve = main.docs_serve
M.run_operation = main.run_operation
M.seed = main.seed
M.show = main.show
M.snapshot = main.snapshot

-- no selectors needed
M.clean = main.clean
M.debug = main.debug
M.deps = main.deps
M.retry = main.retry
M.parse = main.parse
M.run_command = main.run_command
M.debug = main.debug

Expand Down
55 changes: 54 additions & 1 deletion lua/dbtpal/main.lua
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,33 @@ local _compile = function(selector, args) return _cmd_select_args("compile", sel

local _build = function(selector, args) return _cmd_select_args("build", selector, args) end

local _list = function(selector, args) return _cmd_select_args("list", selector, args) end

local _docs_generate = function(selector, args) return _cmd_select_args("docs generate", selector, args) end

local _docs_serve = function(selector, args) return _cmd_select_args("docs serve", selector, args) end

local _run_operation = function(selector, args) return _cmd_select_args("run operation", selector, args) end

local _seed = function(selector, args) return _cmd_select_args("seed", selector, args) end

local _show = function(selector, args) return _cmd_select_args("show", selector, args) end

local _snapshot = function(selector, args) return _cmd_select_args("snapshot", selector, args) end

local _source_freshness = function(args) return _cmd_select_args("source freshness", nil, args) end
-- dbt commands not requiring selectors

local _parse = function(args) return _cmd_select_args("parse", nil, args) end

local _clean = function(args) return _cmd_select_args("clean", nil, args) end

local _debug = function(args) return _cmd_select_args("debug", nil, args) end

local _deps = function(args) return _cmd_select_args("deps", nil, args) end

local _retry = function(args) return _cmd_select_args("retry", nil, args) end

M.run_all = function(args) return _run(nil, args) end

M.run_model = function(selector, args) return _run(selector, args) end
Expand All @@ -47,6 +74,32 @@ M.compile = function(selector, args) return _compile(selector, args) end

M.build = function(selector, args) return _build(selector, args) end

M.list = function(selector, args) return _list(selector, args) end

M.docs_generate = function(selector, args) return _docs_generate(selector, args) end

M.docs_serve = function(selector, args) return _docs_serve(selector, args) end

M.run_operation = function(selector, args) return _run_operation(selector, args) end

M.seed = function(selector, args) return _seed(selector, args) end

M.show = function(selector, args) return _show(selector, args) end

M.snapshot = function(selector, args) return _snapshot(selector, args) end

-- dbt commands not requiring selectors

M.clean = function(args) return _clean(args) end

M.debug = function(args) return _debug(args) end

M.deps = function(args) return _deps(args) end

M.retry = function(args) return _retry(args) end

M.parse = function(args) return _parse(args) end

M.run_command = function(cmd, args) return _cmd_select_args(cmd, args) end

M._create_job = function(cmd, args)
Expand All @@ -56,7 +109,7 @@ M._create_job = function(cmd, args)
if projects.detect_dbt_project_dir(bpath) == false then
log.warn(
"Could not detect dbt project dir, try setting it manually "
.. "or make sure this file is in a dbt project folder"
.. "or make sure this file is in a dbt project folder"
)
return
end
Expand Down

0 comments on commit 6570446

Please sign in to comment.