Add itertools to modules

.changes/unreleased/Features-20220424-132655.yaml

@@ -0,0 +1,7 @@
+kind: Features
+body: Adds itertools to modules Jinja namespace
+time: 2022-04-24T13:26:55.008246+01:00
+custom:
+ Author: bd3dowling
+ Issue: "5130"
+ PR: "5140"

core/dbt/context/base.py

@@ -23,6 +23,7 @@
 import pytz
 import datetime
 import re
+import itertools
 
 # Contexts in dbt Core
 # Contexts are used for Jinja rendering. They include context methods,
@@ -77,11 +78,35 @@ def get_re_module_context() -> Dict[str, Any]:
  return {name: getattr(re, name) for name in context_exports}
 
 
+def get_itertools_module_context() -> Dict[str, Any]:
+ # Excluded dropwhile, filterfalse, takewhile and groupby;
+ # first 3 illogical for Jinja and last redundant.
+ context_exports = [
+ "count",
+ "cycle",
+ "repeat",
+ "accumulate",
+ "chain",
+ "compress",
+ "islice",
+ "starmap",
+ "tee",
+ "zip_longest",
+ "product",
+ "permutations",
+ "combinations",
+ "combinations_with_replacement",
+ ]
+
+ return {name: getattr(itertools, name) for name in context_exports}
+
+
 def get_context_modules() -> Dict[str, Dict[str, Any]]:
  return {
  "pytz": get_pytz_module_context(),
  "datetime": get_datetime_module_context(),
  "re": get_re_module_context(),
+ "itertools": get_itertools_module_context(),
  }