/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>
#
# 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 os

try:
    import pygtk
    pygtk.require("2.0")
except:
    pass

import gtk

import bzrlib.add
import bzrlib.errors as errors

from bzrlib.plugins.gtk import _i18n
from bzrlib.plugins.gtk.dialog import error_dialog
from bzrlib.plugins.gtk.errors import show_bzr_error


class AddDialog(gtk.Dialog):
    """ Dialog for adding selected file or recursively all unknown files/folders in branch """
    
    def __init__(self, wt, wtpath, selected, parent=None):
        """ Initialize the Add dialog. """
        gtk.Dialog.__init__(self, title="Olive - Add file(s)",
                                  parent=parent,
                                  flags=0,
                                  buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
        
        # Get arguments
        self.wt = wt
        self.wtpath = wtpath
        self.selected = selected
        
        # Create widgets
        self._label_add_question = gtk.Label(_i18n("Which file(s) do you want to add?"))
        self._radiobutton_add_selected = gtk.RadioButton(None,_i18n("Selected: %s"%self.selected))
        self._radiobutton_add_unknown = gtk.RadioButton(self._radiobutton_add_selected, 
                                                        _i18n("All unknowns recursively"))
        self._button_add = gtk.Button(stock=gtk.STOCK_ADD)        
        
        self._button_add.connect('clicked', self._on_add_clicked)
        
        # Add widgets to dialog window and decorate
        self.vbox.add(self._label_add_question)
        self.vbox.add(self._radiobutton_add_selected)
        self.vbox.add(self._radiobutton_add_unknown)
        self.vbox.set_spacing(3)
        self.action_area.pack_end(self._button_add)
        
        self.vbox.show_all()
        
    @show_bzr_error
    def _on_add_clicked(self, button):
        """ """
        if self._radiobutton_add_selected.get_active():
            # Add only the selected file
            filename = self.selected
            
            if filename is None:
                error_dialog(_i18n('No file was selected'),
                             _i18n('Please select a file from the list,\nor choose the other option.'))
                return
            
            try:
                self.wt.add([filename])
            except errors.NotBranchError:
                error_dialog(_i18n('Directory is not a branch'),
                             _i18n('You can perform this action only in a branch.'))
                return
        elif self._radiobutton_add_unknown.get_active():
            # Add unknown files recursively
            try:
                self.wt.add(self.wt.unknowns())
            except errors.NotBranchError:
                error_dialog(_i18n('Directory is not a branch'),
                             _i18n('You can perform this action only in a branch.'))
                return
        
        self.response(gtk.RESPONSE_OK)