diff options
| author | Aurélien Bompard | 2015-09-14 09:07:26 +0200 |
|---|---|---|
| committer | Barry Warsaw | 2015-09-15 01:33:37 +0000 |
| commit | d8e96d95e2f1c2d82fd43c04cbc31a8b558a01c6 (patch) | |
| tree | 19050489ccc4c5a82960d31dc2a82a91c968e13a /src/mailman/database/helpers.py | |
| parent | 16903aae68c1fa5967595aa94dfc0880214f61f4 (diff) | |
| download | mailman-d8e96d95e2f1c2d82fd43c04cbc31a8b558a01c6.tar.gz mailman-d8e96d95e2f1c2d82fd43c04cbc31a8b558a01c6.tar.zst mailman-d8e96d95e2f1c2d82fd43c04cbc31a8b558a01c6.zip | |
Diffstat (limited to 'src/mailman/database/helpers.py')
| -rw-r--r-- | src/mailman/database/helpers.py | 41 |
1 files changed, 41 insertions, 0 deletions
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] + ) |
