From abf26fd82850a7e2d123037ebf22dbe03d49fd5a Mon Sep 17 00:00:00 2001 From: Nathan Binkert Date: Fri, 6 Jan 2012 18:19:13 -0500 Subject: [PATCH] hooks: Add a hook to limit the size of any individual file --- util/hgfilesize.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 util/hgfilesize.py 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. -- 2.30.2