/brz/remove-bazaar

# 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, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""Import command classes."""

# Lists of command names

COMMAND_NAMES = ['blob', 'checkpoint', 'commit', 'progress', 'reset', 'tag']

FILE_COMMAND_NAMES = ['filemodify', 'filedelete', 'filecopy', 'filerename',

    'filedeleteall']

# Bazaar file kinds

FILE_KIND = 'file'

SYMLINK_KIND = 'symlink'

class ImportCommand(object):

    """Base class for import commands."""

    def __init__(self, name):

        self.name = name

        # List of field names not to display

        self._binary = []

    def dump_str(self, names=None, child_lists=None, verbose=False):

        """Dump fields as a string.

        :param names: the list of fields to include or

            None for all public fields

        :param child_lists: dictionary of child command names to

            fields for that child command to include

        :param verbose: if True, prefix each line with the command class and

            display fields as a dictionary; if False, dump just the field

            values with tabs between them

        """

        interesting = {}

        if names is None:

            fields = [k for k in self.__dict__.keys() if not k.startswith('_')]

        else:

            fields = names

        for field in fields:

            value = self.__dict__.get(field)

            if field in self._binary and value is not None:

                value = '(...)'

            interesting[field] = value

        if verbose:

            return "%s: %s" % (self.__class__.__name__, interesting)

        else:

            return "\t".join([str(interesting[k]) for k in fields])

class BlobCommand(ImportCommand):

    def __init__(self, mark, data):

        ImportCommand.__init__(self, 'blob')

        self.mark = mark

        self.data = data

        self._binary = ['data']

class CheckpointCommand(ImportCommand):

    def __init__(self):

        ImportCommand.__init__(self, 'checkpoint')

class CommitCommand(ImportCommand):

    def __init__(self, ref, mark, author, committer, message, parents,

        file_iter):

        ImportCommand.__init__(self, 'commit')

        self.ref = ref

        self.mark = mark

        self.author = author

        self.committer = committer

        self.message = message

        self.parents = parents

        self.file_iter = file_iter

        self._binary = ['file_iter']

    def dump_str(self, names=None, child_lists=None, verbose=False):

        result = [ImportCommand.dump_str(self, names, verbose=verbose)]

        for f in self.file_iter():

            if child_lists is None:

                continue

            try:

                child_names = child_lists[f.name]

            except KeyError:

                continue

            result.append("\t%s" % f.dump_str(child_names, verbose=verbose))

        return '\n'.join(result)

class ProgressCommand(ImportCommand):

    def __init__(self, message):

        ImportCommand.__init__(self, 'progress')

        self.message = message

class ResetCommand(ImportCommand):

    def __init__(self, ref, from_):

        ImportCommand.__init__(self, 'reset')

        self.ref = ref

        self.from_ = from_

class TagCommand(ImportCommand):

    def __init__(self, id, from_, tagger, message):

        ImportCommand.__init__(self, 'tag')

        self.id = id

        self.from_ = from_

        self.tagger = tagger

        self.message = message

class FileCommand(ImportCommand):

    """Base class for file commands."""

    pass

class FileModifyCommand(FileCommand):

    def __init__(self, path, kind, is_executable, dataref, data):

        # Either dataref or data should be null

        FileCommand.__init__(self, 'filemodify')

        self.path = path

        self.kind = kind

        self.is_executable = is_executable

        self.dataref = dataref

        self.data = data

        self._binary = ['data']

class FileDeleteCommand(FileCommand):

    def __init__(self, path):

        FileCommand.__init__(self, 'filedelete')

        self.path = path

class FileCopyCommand(FileCommand):

    def __init__(self, src_path, dest_path):

        FileCommand.__init__(self, 'filecopy')

        self.src_path = src_path

        self.dest_path = dest_path

class FileRenameCommand(FileCommand):

    def __init__(self, old_path, new_path):

        FileCommand.__init__(self, 'filerename')

        self.old_path = old_path

        self.new_path = new_path

class FileDeleteAllCommand(FileCommand):

    def __init__(self):

        FileCommand.__init__(self, 'filedeleteall')