# HG changeset patch
# User Marcin Kuzminski <marcin@python-works.com>
# Date 2012-06-04 19:37:41
# Node ID ed8c2fc8dd3bbf76190499b5ad3396fcd27c2b69
# Parent  a455b2c79c85bb9125dbf00326e1dfa8762a0d08

Add support for vcs Changeset objects in get_diff method

diff --git a/rhodecode/lib/vcs/backends/hg/repository.py b/rhodecode/lib/vcs/backends/hg/repository.py
--- a/rhodecode/lib/vcs/backends/hg/repository.py
+++ b/rhodecode/lib/vcs/backends/hg/repository.py
@@ -231,6 +231,12 @@ class MercurialRepository(BaseRepository
         :param context: How many lines before/after changed lines should be
           shown. Defaults to ``3``.
         """
+        if hasattr(rev1, 'raw_id'):
+            rev1 = getattr(rev1, 'raw_id')
+
+        if hasattr(rev2, 'raw_id'):
+            rev2 = getattr(rev2, 'raw_id')
+
         # Check if given revisions are present at repository (may raise
         # ChangesetDoesNotExistError)
         if rev1 != self.EMPTY_CHANGESET: