diff hgext3rd/fastimport/vendor/python_fastimport/processors/query_processor.py @ 86:28704a2a7461 vendor/python-fastimport

Import python-fastimport-0.9.8
author Roy Marples <roy@marples.name>
date Tue, 19 Jan 2021 22:56:34 +0000
parents
children 2fc99e3479d9
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hgext3rd/fastimport/vendor/python_fastimport/processors/query_processor.py	Tue Jan 19 22:56:34 2021 +0000
@@ -0,0 +1,98 @@
+# Copyright (C) 2008 Canonical Ltd
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+"""Import processor that queries the input (and doesn't import)."""
+from __future__ import print_function
+
+
+from fastimport import (
+    commands,
+    processor,
+    )
+
+
+class QueryProcessor(processor.ImportProcessor):
+    """An import processor that queries the input.
+
+    No changes to the current repository are made.
+    """
+
+    known_params = (
+        commands.COMMAND_NAMES +
+        commands.FILE_COMMAND_NAMES +
+        [b'commit-mark']
+    )
+
+    def __init__(self, params=None, verbose=False):
+        processor.ImportProcessor.__init__(self, params, verbose)
+        self.parsed_params = {}
+        self.interesting_commit = None
+        self._finished = False
+        if params:
+            if 'commit-mark' in params:
+                self.interesting_commit = params['commit-mark']
+                del params['commit-mark']
+            for name, value in params.items():
+                if value == 1:
+                    # All fields
+                    fields = None
+                else:
+                    fields = value.split(',')
+                self.parsed_params[name] = fields
+
+    def pre_handler(self, cmd):
+        """Hook for logic before each handler starts."""
+        if self._finished:
+            return
+        if self.interesting_commit and cmd.name == 'commit':
+            if cmd.mark == self.interesting_commit:
+                print(cmd.to_string())
+                self._finished = True
+            return
+        if cmd.name in self.parsed_params:
+            fields = self.parsed_params[cmd.name]
+            str = cmd.dump_str(fields, self.parsed_params, self.verbose)
+            print("%s" % (str,))
+
+    def progress_handler(self, cmd):
+        """Process a ProgressCommand."""
+        pass
+
+    def blob_handler(self, cmd):
+        """Process a BlobCommand."""
+        pass
+
+    def checkpoint_handler(self, cmd):
+        """Process a CheckpointCommand."""
+        pass
+
+    def commit_handler(self, cmd):
+        """Process a CommitCommand."""
+        pass
+
+    def reset_handler(self, cmd):
+        """Process a ResetCommand."""
+        pass
+
+    def tag_handler(self, cmd):
+        """Process a TagCommand."""
+        pass
+
+    def feature_handler(self, cmd):
+        """Process a FeatureCommand."""
+        feature = cmd.feature_name
+        if feature not in commands.FEATURE_NAMES:
+            self.warning("feature %s is not supported - parsing may fail"
+                % (feature,))