/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to commit.py

  • Committer: Jelmer Vernooij
  • Date: 2011-11-02 15:29:17 UTC
  • Revision ID: jelmer@samba.org-20111102152917-7u4ac3okr363fped
Move commit message saving to separate file.

Show diffs side-by-side

added added

removed removed

Lines of Context:
commit.py
21
21
from gi.repository import GObject
22
22
from gi.repository import Pango
23
23
 
 
24
try:
 
25
    from bzrlib import bencode
 
26
except ImportError:
 
27
    from bzrlib.util import bencode
 
28
 
24
29
from bzrlib import (
25
30
    errors,
26
 
    osutils,
27
31
    trace,
28
32
    )
29
 
try:
30
 
    from bzrlib import bencode
31
 
except ImportError:
32
 
    from bzrlib.util import bencode
33
 
 
34
33
from bzrlib.plugins.gtk.dialog import question_dialog
35
34
from bzrlib.plugins.gtk.errors import show_bzr_error
36
35
from bzrlib.plugins.gtk.i18n import _i18n
 
36
from bzrlib.plugins.gtk.commitmsgs import SavedCommitMessagesManager
37
37
 
38
38
try:
39
39
    import dbus
793
793
        rev_dict['revision_id'] = rev.revision_id
794
794
        return rev_dict
795
795
 
796
 
 
797
 
class SavedCommitMessagesManager:
798
 
    """Save glogal and per-file commit messages.
799
 
 
800
 
    Saves global commit message and utf-8 file_id->message dictionary
801
 
    of per-file commit messages on disk. Re-reads them later for re-using.
802
 
    """
803
 
 
804
 
    def __init__(self, tree=None, branch=None):
805
 
        """If branch is None, builds empty messages, otherwise reads them
806
 
        from branch's disk storage. 'tree' argument is for the future."""
807
 
        if branch is None:
808
 
            self.global_message = u''
809
 
            self.file_messages = {}
810
 
        else:
811
 
            config = branch.get_config()
812
 
            self.global_message = config.get_user_option(
813
 
                'gtk_global_commit_message')
814
 
            if self.global_message is None:
815
 
                self.global_message = u''
816
 
            file_messages = config.get_user_option('gtk_file_commit_messages')
817
 
            if file_messages: # unicode and B-encoded:
818
 
                self.file_messages = bencode.bdecode(
819
 
                    file_messages.encode('UTF-8'))
820
 
            else:
821
 
                self.file_messages = {}
822
 
 
823
 
    def get(self):
824
 
        return self.global_message, self.file_messages
825
 
 
826
 
    def is_not_empty(self):
827
 
        return bool(self.global_message or self.file_messages)
828
 
 
829
 
    def insert(self, global_message, file_info):
830
 
        """Formats per-file commit messages (list of dictionaries, one per file)
831
 
        into one utf-8 file_id->message dictionary and merges this with
832
 
        previously existing dictionary. Merges global commit message too."""
833
 
        file_messages = {}
834
 
        for fi in file_info:
835
 
            file_message = fi['message']
836
 
            if file_message:
837
 
                file_messages[fi['file_id']] = file_message # utf-8 strings
838
 
        for k,v in file_messages.iteritems():
839
 
            try:
840
 
                self.file_messages[k] = v + '\n******\n' + self.file_messages[k]
841
 
            except KeyError:
842
 
                self.file_messages[k] = v
843
 
        if self.global_message:
844
 
            self.global_message = global_message + '\n******\n' \
845
 
                + self.global_message
846
 
        else:
847
 
            self.global_message = global_message
848
 
 
849
 
    def save(self, tree, branch):
850
 
        # We store in branch's config, which can be a problem if two gcommit
851
 
        # are done in two checkouts of one single branch (comments overwrite
852
 
        # each other). Ideally should be in working tree. But uncommit does
853
 
        # not always have a working tree, though it always has a branch.
854
 
        # 'tree' argument is for the future
855
 
        config = branch.get_config()
856
 
        # should it be named "gtk_" or some more neutral name ("gui_" ?) to
857
 
        # be compatible with qbzr in the future?
858
 
        config.set_user_option('gtk_global_commit_message', self.global_message)
859
 
        # bencode() does not know unicode objects but set_user_option()
860
 
        # requires one:
861
 
        config.set_user_option(
862
 
            'gtk_file_commit_messages',
863
 
            bencode.bencode(self.file_messages).decode('UTF-8'))
864
 
 
865
 
 
866
 
def save_commit_messages(local, master, old_revno, old_revid,
867
 
                         new_revno, new_revid):
868
 
    b = local
869
 
    if b is None:
870
 
        b = master
871
 
    mgr = SavedCommitMessagesManager(None, b)
872
 
    revid_iterator = b.repository.iter_reverse_revision_history(old_revid)
873
 
    cur_revno = old_revno
874
 
    new_revision_id = old_revid
875
 
    graph = b.repository.get_graph()
876
 
    for rev_id in revid_iterator:
877
 
        if cur_revno == new_revno:
878
 
            break
879
 
        cur_revno -= 1
880
 
        rev = b.repository.get_revision(rev_id)
881
 
        file_info = rev.properties.get('file-info', None)
882
 
        if file_info is None:
883
 
            file_info = {}
884
 
        else:
885
 
            file_info = bencode.bdecode(file_info.encode('UTF-8'))
886
 
        global_message = osutils.safe_unicode(rev.message)
887
 
        # Concatenate comment of the uncommitted revision
888
 
        mgr.insert(global_message, file_info)
889
 
 
890
 
        parents = graph.get_parent_map([rev_id]).get(rev_id, None)
891
 
        if not parents:
892
 
            continue
893
 
    mgr.save(None, b)