From 3d3ab4ea235f5b9d1a3f0393dd57a615f2cae60a Mon Sep 17 00:00:00 2001 From: Faizan Azim Date: Wed, 19 Mar 2025 21:26:55 +0000 Subject: [PATCH 1/2] feat: :sparkles: Update ID function formatting and add CreateSchema functionality --- sql_db_utils/sql_extras.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/sql_db_utils/sql_extras.py b/sql_db_utils/sql_extras.py index 42f5465..e2fcf3f 100644 --- a/sql_db_utils/sql_extras.py +++ b/sql_db_utils/sql_extras.py @@ -127,7 +127,7 @@ def compile_create_prefixed_id_function(element: CreatePrefixedIdFunction, _comp next_val INTEGER; BEGIN next_val := nextval(seq_name); - RETURN prefix || next_val; + RETURN prefix || '_' || next_val; END; $$ LANGUAGE plpgsql; """ @@ -148,7 +148,17 @@ def compile_create_suffixed_id_function(element: CreateSuffixedIdFunction, _comp next_val INTEGER; BEGIN next_val := nextval(seq_name); - RETURN next_val || suffix; + RETURN next_val || '_' || suffix; END; $$ LANGUAGE plpgsql; """ + + +class CreateSchema(DDLElement): + def __init__(self, schema_name: str): + self.schema_name = schema_name + + +@compiler.compiles(CreateSchema) +def compile_create_schema(element: CreateSchema, compiler: Any, **kw: Any) -> str: + return f"CREATE SCHEMA IF NOT EXISTS {element.schema_name};" From 798a16e347e39608cd0e989d82c0f908dd504c3f Mon Sep 17 00:00:00 2001 From: Faizan Azim Date: Wed, 19 Mar 2025 21:27:33 +0000 Subject: [PATCH 2/2] chore: :bookmark: Update version to 1.1.1 --- sql_db_utils/__version__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sql_db_utils/__version__.py b/sql_db_utils/__version__.py index 6849410..a82b376 100644 --- a/sql_db_utils/__version__.py +++ b/sql_db_utils/__version__.py @@ -1 +1 @@ -__version__ = "1.1.0" +__version__ = "1.1.1"