hooks.py
174 lines
| 4.6 KiB
| text/x-python
|
PythonLexer
r913 | # -*- coding: utf-8 -*- | |||
""" | ||||
rhodecode.lib.hooks | ||||
~~~~~~~~~~~~~~~~~~~ | ||||
Hooks runned by rhodecode | ||||
r1203 | ||||
r913 | :created_on: Aug 6, 2010 | |||
:author: marcink | ||||
r1824 | :copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com> | |||
r913 | :license: GPLv3, see COPYING for more details. | |||
""" | ||||
r1206 | # 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. | ||||
r1203 | # | |||
r547 | # 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. | ||||
r1203 | # | |||
r547 | # You should have received a copy of the GNU General Public License | |||
r1206 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |||
r913 | import os | |||
import sys | ||||
r547 | ||||
r1415 | from mercurial.scmutil import revrange | |||
r654 | from mercurial.node import nullrev | |||
r2105 | from rhodecode import EXTENSIONS | |||
r654 | from rhodecode.lib import helpers as h | |||
from rhodecode.lib.utils import action_logger | ||||
r2105 | from inspect import isfunction | |||
r547 | ||||
r1307 | ||||
r547 | def repo_size(ui, repo, hooktype=None, **kwargs): | |||
r1722 | """ | |||
Presents size of repository after push | ||||
r1203 | ||||
r913 | :param ui: | |||
:param repo: | ||||
:param hooktype: | ||||
""" | ||||
r547 | ||||
size_hg, size_root = 0, 0 | ||||
for path, dirs, files in os.walk(repo.root): | ||||
if path.find('.hg') != -1: | ||||
for f in files: | ||||
r675 | try: | |||
size_hg += os.path.getsize(os.path.join(path, f)) | ||||
except OSError: | ||||
pass | ||||
r547 | else: | |||
for f in files: | ||||
r675 | try: | |||
size_root += os.path.getsize(os.path.join(path, f)) | ||||
except OSError: | ||||
pass | ||||
r547 | 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) | ||||
r1814 | ||||
last_cs = repo[len(repo) - 1] | ||||
msg = ('Repository size .hg:%s repo:%s total:%s\n' | ||||
'Last revision is now r%s:%s\n') % ( | ||||
size_hg_f, size_root_f, size_total_f, last_cs.rev(), last_cs.hex()[:12] | ||||
) | ||||
sys.stdout.write(msg) | ||||
r675 | ||||
r1307 | ||||
r654 | def log_pull_action(ui, repo, **kwargs): | |||
r1722 | """ | |||
Logs user last pull action | ||||
r1203 | ||||
r654 | :param ui: | |||
:param repo: | ||||
""" | ||||
r675 | ||||
r2105 | extras = dict(repo.ui.configitems('rhodecode_extras')) | |||
username = extras['username'] | ||||
repository = extras['repository'] | ||||
r654 | action = 'pull' | |||
r675 | ||||
r2105 | action_logger(username, action, repository, extras['ip'], commit=True) | |||
# extension hook call | ||||
callback = getattr(EXTENSIONS, 'PULL_HOOK', None) | ||||
r675 | ||||
r2105 | if isfunction(callback): | |||
kw = {} | ||||
kw.update(extras) | ||||
callback(**kw) | ||||
r654 | return 0 | |||
r547 | ||||
r1307 | ||||
r654 | def log_push_action(ui, repo, **kwargs): | |||
r1722 | """ | |||
Maps user last push action to new changeset id, from mercurial | ||||
r1203 | ||||
r604 | :param ui: | |||
:param repo: | ||||
r547 | """ | |||
r675 | ||||
r2105 | extras = dict(repo.ui.configitems('rhodecode_extras')) | |||
username = extras['username'] | ||||
repository = extras['repository'] | ||||
action = extras['action'] + ':%s' | ||||
r654 | node = kwargs['node'] | |||
r675 | ||||
r654 | def get_revs(repo, rev_opt): | |||
if rev_opt: | ||||
revs = revrange(repo, rev_opt) | ||||
r675 | ||||
r654 | if len(revs) == 0: | |||
return (nullrev, nullrev) | ||||
return (max(revs), min(revs)) | ||||
else: | ||||
return (len(repo) - 1, 0) | ||||
r675 | ||||
r654 | stop, start = get_revs(repo, [node + ':']) | |||
r675 | ||||
r654 | revs = (str(repo[r]) for r in xrange(start, stop + 1)) | |||
r675 | ||||
r654 | action = action % ','.join(revs) | |||
r675 | ||||
r2105 | action_logger(username, action, repository, extras['ip'], commit=True) | |||
r675 | ||||
r2105 | # extension hook call | |||
callback = getattr(EXTENSIONS, 'PUSH_HOOK', None) | ||||
if isfunction(callback): | ||||
kw = {'pushed_revs': revs} | ||||
kw.update(extras) | ||||
callback(**kw) | ||||
r654 | return 0 | |||
r1972 | ||||
def log_create_repository(repository_dict, created_by, **kwargs): | ||||
""" | ||||
Post create repository Hook. This is a dummy function for admins to re-use | ||||
r2105 | if needed. It's taken from rhodecode-extensions module and executed | |||
if present | ||||
r1972 | ||||
r1982 | :param repository: dict dump of repository object | |||
r1972 | :param created_by: username who created repository | |||
:param created_date: date of creation | ||||
available keys of repository_dict: | ||||
'repo_type', | ||||
'description', | ||||
'private', | ||||
'created_on', | ||||
'enable_downloads', | ||||
'repo_id', | ||||
'user_id', | ||||
'enable_statistics', | ||||
'clone_uri', | ||||
'fork_id', | ||||
'group_id', | ||||
'repo_name' | ||||
""" | ||||
r2105 | callback = getattr(EXTENSIONS, 'CREATE_REPO_HOOK', None) | |||
if isfunction(callback): | ||||
kw = {} | ||||
kw.update(repository_dict) | ||||
kw.update({'created_by': created_by}) | ||||
kw.update(kwargs) | ||||
return callback(**kw) | ||||
r1972 | ||||
r1982 | return 0 | |||