Mercurial > kallithea
view rhodecode/lib/hooks.py @ 850:b3ec9f4af18b beta
Added rhodecode to sys.path at Sphinx configuration
author | Lukasz Balcerzak <lukasz.balcerzak@python-center.org> |
---|---|
date | Thu, 16 Dec 2010 00:50:15 +0100 |
parents | 59670f091c76 |
children | 3d0661b8aaa4 07a6e8c65526 |
line wrap: on
line source
#!/usr/bin/env python # encoding: utf-8 # custom hooks for application # Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com> # # 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; version 2 # of the License or (at your opinion) any later version of the license. # # 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, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, # MA 02110-1301, USA. """ Created on Aug 6, 2010 @author: marcink """ from mercurial.cmdutil import revrange from mercurial.node import nullrev from rhodecode.lib import helpers as h from rhodecode.lib.utils import action_logger import os import sys def repo_size(ui, repo, hooktype=None, **kwargs): if hooktype != 'changegroup': return False size_hg, size_root = 0, 0 for path, dirs, files in os.walk(repo.root): if path.find('.hg') != -1: for f in files: try: size_hg += os.path.getsize(os.path.join(path, f)) except OSError: pass else: for f in files: try: size_root += os.path.getsize(os.path.join(path, f)) except OSError: pass size_hg_f = h.format_byte_size(size_hg) size_root_f = h.format_byte_size(size_root) size_total_f = h.format_byte_size(size_root + size_hg) sys.stdout.write('Repository size .hg:%s repo:%s total:%s\n' \ % (size_hg_f, size_root_f, size_total_f)) def log_pull_action(ui, repo, **kwargs): """ Logs user last pull action :param ui: :param repo: """ extra_params = dict(repo.ui.configitems('rhodecode_extras')) username = extra_params['username'] repository = extra_params['repository'] action = 'pull' action_logger(username, action, repository, extra_params['ip']) return 0 def log_push_action(ui, repo, **kwargs): """ Maps user last push action to new changeset id, from mercurial :param ui: :param repo: """ extra_params = dict(repo.ui.configitems('rhodecode_extras')) username = extra_params['username'] repository = extra_params['repository'] action = 'push:%s' node = kwargs['node'] def get_revs(repo, rev_opt): if rev_opt: revs = revrange(repo, rev_opt) if len(revs) == 0: return (nullrev, nullrev) return (max(revs), min(revs)) else: return (len(repo) - 1, 0) stop, start = get_revs(repo, [node + ':']) revs = (str(repo[r]) for r in xrange(start, stop + 1)) action = action % ','.join(revs) action_logger(username, action, repository, extra_params['ip']) return 0