wcs/wcs/ctl/rebuild_indexes.py

61 lines
1.9 KiB
Python

# w.c.s. - web application for online forms
# Copyright (C) 2005-2010 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/>.
from ..qommon.ctl import Command, make_option
def rebuild_vhost_indexes(pub, destroy=False):
from wcs.formdef import FormDef
if destroy:
FormDef.destroy_indexes()
FormDef.rebuild_indexes()
from wcs.roles import Role
if destroy:
Role.destroy_indexes()
Role.rebuild_indexes()
class CmdRebuildIndexes(Command):
name = 'rebuild_indexes'
def __init__(self):
Command.__init__(
self,
[
make_option('--all', action='store_true', dest='all', default=False),
make_option('--destroy', action='store_true', dest='destroy', default=False),
],
)
def execute(self, base_options, sub_options, args):
from .. import publisher
publisher.WcsPublisher.configure(self.config)
pub = publisher.WcsPublisher.create_publisher(register_tld_names=False)
if sub_options.all:
hostnames = [x.domain for x in publisher.WcsPublisher.get_tenants()]
else:
hostnames = args
for hostname in hostnames:
pub.set_tenant(hostname)
rebuild_vhost_indexes(pub, destroy=sub_options.destroy)
CmdRebuildIndexes.register()