Produits Entr'ouvert » w.c.s.

                cur.execute("SELECT table_name FROM information_schema.tables \

                            WHERE table_schema='public' AND \

                            table_type='BASE TABLE'")

                tables_names = [x[0] for x in cur.fetchall()]

               if 'wcs_meta' in tables_names :

                    new_database = False

#w.c.s. -  web application for online forms 

# Copyright (C) 2005-2014  Entr'ouvert 

# 

# This program 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 2 of the License, or 

# (at your option) any later version. 

# 

# This program 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 this program; if not, see <http://www.gnu.org/licenses/>.

import os

import sys

import psycopg2

import psycopg2.errorcodes

from datetime import datetime

from shutil import rmtree

from qommon.ctl import Command, make_option

class CmdDeleteTenant(Command):

    name = 'delete_tenant'

    def __init__(self):

        Command.__init__(self, [

                make_option('--force-drop', action='store_true', default=False),

                ])

    def execute(self, base_options, sub_options, args):

        import publisher

        publisher.WcsPublisher.configure(self.config)

        pub = publisher.WcsPublisher.create_publisher(

                register_cron=False, register_tld_names=False)

        hostname = args[0]

        pub.app_dir = os.path.join(pub.app_dir, hostname)

        pub.set_config()

        postgresql_cfg = {}

        for k, v in pub.cfg['postgresql'].items():

            if v and isinstance(v, basestring):

                postgresql_cfg[k] = v

        createdb_cfg = pub.cfg['postgresql'].get('createdb-connection-params')

        if not createdb_cfg:

            createdb_cfg = postgresql_cfg

        try:

            pgconn = psycopg2.connect(**createdb_cfg)

        except psycopg2.Error as e:

            print >> sys.stderr, 'failed to connect to postgresql (%s)' % \

                    psycopg2.errorcodes.lookup(e.pgcode)

            return

        pgconn.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)

        cur = pgconn.cursor()

        try:

            cur.execute("SELECT table_name FROM information_schema.tables \

                        WHERE table_schema='public' AND \

                        table_type='BASE TABLE'")

            tables_names = [x[0] for x in cur.fetchall()]

            if sub_options.force_drop:

                for table_name in tables_names:

                    cur.execute('DROP TABLE %s CASCADE' % table_name)

                rmtree(pub.app_dir)

            else:

                deletion_date = datetime.now().strftime('%Y%m%d_%H%M%S_%f')

                for table_name in tables_names:

                    cur.execute('ALTER TABLE %s RENAME TO removed_%s_%s' % \

                               (table_name, deletion_date, table_name))

                os.rename(pub.app_dir, pub.app_dir + '_removed_%s.invalid' % \

                          deletion_date)

        except psycopg2.Error as e:

            if e.pgcode == psycopg2.errorcodes.DUPLICATE_TABLE:

                # when tables names are too long, postgresql truncates, which

                # can create some duplicates

                temp_date = datetime.now().strftime('%Y%m%d_%H%M%S_%f')

                cur.execute('ALTER TABLE %s RENAME TO removed_%s_%s' % \

                            (table_name, temp_date, table_name))

            else:

                print >> sys.stderr, 'failed to alter database %s : (%s)' % \

                        (createdb_cfg['database'], psycopg2.errorcodes.lookup(e.pgcode))

                return

        cur.close()

CmdDeleteTenant.register()