hgext/graphlog.py

--- a/hgext/graphlog.py	Sun Mar 13 14:36:47 2011 +0100
+++ b/hgext/graphlog.py	Sun Mar 13 15:04:13 2011 +0100
@@ -319,7 +319,11 @@
     '''wrap the command'''
     def graph(orig, *args, **kwargs):
         if kwargs['graph']:
-            return wrapfn(*args, **kwargs)
+            try:
+                return wrapfn(*args, **kwargs)
+            except TypeError, e:
+                if len(args) > wrapfn.func_code.co_argcount:
+                    raise util.Abort(_('--graph option allows at most one file'))
         return orig(*args, **kwargs)
     entry = extensions.wrapcommand(table, cmd, graph)
     entry[1].append(('G', 'graph', None, _("show the revision DAG")))