Mercurial > hg > hg-fastimport
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,))
