summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNathan Binkert <nate@binkert.org>2012-01-06 18:19:13 -0500
committerNathan Binkert <nate@binkert.org>2012-01-06 18:19:13 -0500
commitabf26fd82850a7e2d123037ebf22dbe03d49fd5a (patch)
tree1f0f270db6fce8e1d28bde25ade86fbc1981e9bb
parentce941fd2ae2908dd0261132f35ab90e82c07b6b7 (diff)
downloadgem5-abf26fd82850a7e2d123037ebf22dbe03d49fd5a.tar.xz
hooks: Add a hook to limit the size of any individual file
-rw-r--r--util/hgfilesize.py32
1 files changed, 32 insertions, 0 deletions
diff --git a/util/hgfilesize.py b/util/hgfilesize.py
new file mode 100644
index 000000000..8b9ad67ea
--- /dev/null
+++ b/util/hgfilesize.py
@@ -0,0 +1,32 @@
+from mercurial import context
+from mercurial.i18n import _
+
+'''
+[extensions]
+hgfilesize=~/m5/incoming/util/hgfilesize.py
+
+[hooks]
+pretxncommit = python:hgfilesize.limit_file_size
+pretxnchangegroup = python:hgfilesize.limit_file_size
+
+[limit_file_size]
+maximum_file_size = 200000
+'''
+
+def limit_file_size(ui, repo, node=None, **kwargs):
+ '''forbid files over a given size'''
+
+ # default limit is 1 MB
+ limit = int(ui.config('limit_file_size', 'maximum_file_size', 1024*1024))
+ existing_tip = context.changectx(repo, node).rev()
+ new_tip = context.changectx(repo, 'tip').rev()
+ for rev in xrange(existing_tip, new_tip + 1):
+ ctx = context.changectx(repo, rev)
+ for f in ctx.files():
+ fctx = ctx.filectx(f)
+ if fctx.size() > limit:
+ ui.write(_('file %s of %s is too large: %d > %d\n') % \
+ (f, ctx, fctx.size(), limit))
+ return True # This is invalid
+
+ return False # Things are OK.