♻️ clean: move multiple strip funcs to trim.

Author: korawica

src/sqlplate/sqlplate.py

@@ -6,7 +6,7 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import Any, Iterator, Optional
+from typing import Any, Iterator, Optional, Callable
 
 from jinja2 import Template
 
@@ -15,6 +15,9 @@
 from .utils import get_env, remove_sql_comment
 
 
+trim: Callable[[str], str] = lambda x: x.strip().strip('\n')
+
+
 class SQLPlate:
     """A SQLPlate object for render any SQL template that prepare by Jinja
     template.
@@ -100,17 +103,15 @@ def load(self, remove_comment: bool = False, **kwargs) -> str:
                 "Template object does not create before load, you should use "
                 "`.template(name=?)`."
             )
-        render: str = (
+        render: str = trim(
             self._template.render(
                 **(
                     {"_system": self.name, "_template": self._template_name}
                     | config().export(self._template_type)
                     | self._option
                     | kwargs
-                )
+                ),
             )
-            .strip()
-            .strip('\n')
         )
         if remove_comment:
             return remove_sql_comment(render)
@@ -130,10 +131,10 @@ def stream(
                 statement. Default is ';'.
         """
         yield from (
-            s.strip().strip('\n')
+            trim(s)
             for s in (
                 self.load(remove_comment=remove_comment, **kwargs)
                 .split(split_char)
             )
-            if s.strip().strip('\n') != ''
+            if trim(s) != ''
         )

tests/test_databricks.py

@@ -259,7 +259,6 @@ def test_sql_full_dump(template_path):
         .option('query', 'SELECT * FROM catalog-name.schema-name.source-name')
         .load()
     )
-    print(prepare(statement))
     assert prepare(statement) == dedent("""
         DELETE FROM catalog-name.schema-name.table-name
         WHERE load_src = 'SOURCE_FOO'