annotate pylons_app/lib/hooks.py @ 492:a5a17000e45b celery

timeproxy logging memory leak fix.
author Marcin Kuzminski <marcin@python-works.com>
date Tue, 21 Sep 2010 15:36:46 +0200
parents b27d32cb3157
children 3d6d548ad3cc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
392
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
1 #!/usr/bin/env python
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
2 # encoding: utf-8
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
3 # custom hooks for application
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
4 # Copyright (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com>
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
5 #
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
6 # This program is free software; you can redistribute it and/or
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
7 # modify it under the terms of the GNU General Public License
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
8 # as published by the Free Software Foundation; version 2
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
9 # of the License or (at your opinion) any later version of the license.
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
10 #
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
11 # This program is distributed in the hope that it will be useful,
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
14 # GNU General Public License for more details.
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
15 #
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
16 # You should have received a copy of the GNU General Public License
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
17 # along with this program; if not, write to the Free Software
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
19 # MA 02110-1301, USA.
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
20 """
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
21 Created on Aug 6, 2010
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
22
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
23 @author: marcink
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
24 """
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
25
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
26 import sys
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
27 import os
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
28 from pylons_app.lib import helpers as h
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
29
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
30 def repo_size(ui, repo, hooktype=None, **kwargs):
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
31
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
32 if hooktype != 'changegroup':
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
33 return False
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
34
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
35 size_hg, size_root = 0, 0
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
36 for path, dirs, files in os.walk(repo.root):
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
37 if path.find('.hg') != -1:
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
38 for f in files:
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
39 size_hg += os.path.getsize(os.path.join(path, f))
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
40 else:
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
41 for f in files:
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
42 size_root += os.path.getsize(os.path.join(path, f))
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
43
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
44 size_hg_f = h.format_byte_size(size_hg)
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
45 size_root_f = h.format_byte_size(size_root)
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
46 size_total_f = h.format_byte_size(size_root + size_hg)
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
47 sys.stdout.write('Repository size .hg:%s repo:%s total:%s\n' \
b27d32cb3157 Implemented hooks system,
Marcin Kuzminski <marcin@python-works.com>
parents:
diff changeset
48 % (size_hg_f, size_root_f, size_total_f))