From d8e96d95e2f1c2d82fd43c04cbc31a8b558a01c6 Mon Sep 17 00:00:00 2001
From: Aurélien Bompard
Date: Mon, 14 Sep 2015 09:07:26 +0200
Subject: Fixes for the review comments

---
 src/mailman/database/helpers.py | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 src/mailman/database/helpers.py

(limited to 'src/mailman/database/helpers.py')

diff --git a/src/mailman/database/helpers.py b/src/mailman/database/helpers.py
new file mode 100644
index 000000000..dafd3b512
--- /dev/null
+++ b/src/mailman/database/helpers.py
@@ -0,0 +1,41 @@
+# Copyright (C) 2015 by the Free Software Foundation, Inc.
+#
+# This file is part of GNU Mailman.
+#
+# GNU Mailman is free software: you can redistribute it and/or modify it under
+# the terms of the GNU General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option)
+# any later version.
+#
+# GNU Mailman is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+# more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# GNU Mailman.  If not, see <http://www.gnu.org/licenses/>.
+
+"""Common database helpers."""
+
+__all__ = [
+    'exists_in_db',
+    'is_sqlite',
+    ]
+
+import sqlalchemy as sa
+
+
+def is_sqlite(bind):
+    return bind.dialect.name == 'sqlite'
+
+
+def exists_in_db(bind, tablename, columnname=None):
+    md = sa.MetaData()
+    md.reflect(bind=bind)
+    if columnname is None:
+        return tablename in md.tables
+    else:
+        return (
+            tablename in md.tables and
+            columnname in [c.name for c in md.tables[tablename].columns]
+            )
-- 
cgit v1.2.3-70-g09d2