Update drop_before_all to support fhir-schema and old approach. Closes #68

Author: ruscoder

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.1.12
+
+- Update drop_before_all to support fhir-schema and old approach #68
+
 ## 0.1.11
 
 - Add `headers` to SDKOperationRequest

aidbox_python_sdk/__init__.py

@@ -1,5 +1,5 @@
 __title__ = "aidbox-python-sdk"
-__version__ = "0.1.11"
+__version__ = "0.1.12"
 __author__ = "beda.software"
 __license__ = "None"
 __copyright__ = "Copyright 2024 beda.software"

aidbox_python_sdk/db_migrations.py

@@ -13,7 +13,26 @@
 END LOOP;
 END;
 
-$$ LANGUAGE plpgsql;
-        """,
-    }
+$$ LANGUAGE plpgsql;""",
+    },
+    {
+        "id": "20240913_change_drop_before_all",
+        "sql": """
+DROP FUNCTION IF EXISTS drop_before_all(integer);
+
+CREATE FUNCTION drop_before_all(integer) RETURNS VOID AS $$
+declare
+e record;
+BEGIN
+FOR e IN (
+  SELECT table_name
+  FROM information_schema.columns
+  WHERE column_name = 'txid' AND table_schema = 'public' AND table_name NOT LIKE '%_history'
+) LOOP
+    EXECUTE 'DELETE FROM "' || e.table_name || '" WHERE txid > ' || $1 ;
+END LOOP;
+END;
+
+$$ LANGUAGE plpgsql;""",
+    },
 ]