Mercurial > kallithea
view rhodecode/lib/paster_commands/cleanup.py @ 3774:60335b702a00 beta
invalidation: don't create CacheInvalidation records on startup
Creating the records early gave an advantage before lightweight was introduced.
With lightweight it is no longer necessary.
The records will be created on demand anyway and there is no reason to create and
maintain them before they are used.
author | Mads Kiilerich <madski@unity3d.com> |
---|---|
date | Wed, 03 Apr 2013 15:56:12 +0200 |
parents | 4839bc1d89fa |
children | a42bfe8a9335 |
line wrap: on
line source
# -*- coding: utf-8 -*- """ rhodecode.lib.paster_commands.cleanup ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ cleanup-repos paster command for RhodeCode :created_on: Jul 14, 2012 :author: marcink :copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com> :license: GPLv3, see COPYING for more details. """ # 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 3 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 __future__ import with_statement import os import sys import re import shutil import logging import datetime from os.path import dirname as dn, join as jn #to get the rhodecode import rc_path = dn(dn(dn(os.path.realpath(__file__)))) sys.path.append(rc_path) from rhodecode.lib.utils import BasePasterCommand, ask_ok, REMOVED_REPO_PAT from rhodecode.lib.utils2 import safe_str from rhodecode.model.db import RhodeCodeUi log = logging.getLogger(__name__) class Command(BasePasterCommand): max_args = 1 min_args = 1 usage = "CONFIG_FILE" group_name = "RhodeCode" takes_config_file = -1 parser = BasePasterCommand.standard_parser(verbose=True) summary = "Cleanup deleted repos" def _parse_older_than(self, val): regex = re.compile(r'((?P<days>\d+?)d)?((?P<hours>\d+?)h)?((?P<minutes>\d+?)m)?((?P<seconds>\d+?)s)?') parts = regex.match(val) if not parts: return parts = parts.groupdict() time_params = {} for (name, param) in parts.iteritems(): if param: time_params[name] = int(param) return datetime.timedelta(**time_params) def _extract_date(self, name): """ Extract the date part from rm__<date> pattern of removed repos, and convert it to datetime object :param name: """ date_part = name[4:19] # 4:19 since we don't parse milisecods return datetime.datetime.strptime(date_part, '%Y%m%d_%H%M%S') def command(self): #get SqlAlchemy session self._init_session() repos_location = RhodeCodeUi.get_repos_location() to_remove = [] for dn, dirs, f in os.walk(safe_str(repos_location)): alldirs = list(dirs) del dirs[:] if ('.hg' in alldirs or 'objects' in alldirs and ('refs' in alldirs or 'packed-refs' in f)): continue for loc in alldirs: if REMOVED_REPO_PAT.match(loc): to_remove.append([os.path.join(dn, loc), self._extract_date(loc)]) else: dirs.append(loc) #filter older than (if present)! now = datetime.datetime.now() older_than = self.options.older_than if older_than: to_remove_filtered = [] older_than_date = self._parse_older_than(older_than) for name, date_ in to_remove: repo_age = now - date_ if repo_age > older_than_date: to_remove_filtered.append([name, date_]) to_remove = to_remove_filtered print >> sys.stdout, 'removing %s deleted repos older than %s (%s)' \ % (len(to_remove), older_than, older_than_date) else: print >> sys.stdout, 'removing all [%s] deleted repos' \ % len(to_remove) if self.options.dont_ask or not to_remove: # don't ask just remove ! remove = True else: remove = ask_ok('the following repositories will be deleted completely:\n%s\n' 'are you sure you want to remove them [y/n]?' % ', \n'.join(['%s removed on %s' % (safe_str(x[0]), safe_str(x[1])) for x in to_remove])) if remove: for path, date_ in to_remove: print >> sys.stdout, 'removing repository %s' % path shutil.rmtree(path) else: print 'nothing done exiting...' sys.exit(0) def update_parser(self): self.parser.add_option( '--older-than', action='store', dest='older_than', help=("only remove repos that have been removed " "at least given time ago. " "The default is to remove all removed repositories. " "Possible suffixes: " "d (days), h (hours), m (minutes), s (seconds). " "For example --older-than=30d deletes repositories " "removed more than 30 days ago.") ) self.parser.add_option( '--dont-ask', action="store_true", dest="dont_ask", help="remove repositories without asking for confirmation." )