summaryrefslogtreecommitdiffstats
path: root/deluge/plugins/Label/deluge/plugins/label/gtkui/__init__.py
blob: bf69e289b6ac432a444d8e011cac8af07a0a8ef0 (plain)
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
#
# __init__.py
#
# Copyright (C) 2008 Martijn Voncken <mvoncken@gmail.com>
#
# Deluge is free software.
#
# You may redistribute it and/or modify it under the terms of the
# GNU General Public License, as published by the Free Software
# Foundation; either version 3 of the License, or (at your option)
# any later version.
#
# deluge 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 deluge.    If not, write to:
# 	The Free Software Foundation, Inc.,
# 	51 Franklin Street, Fifth Floor
# 	Boston, MA  02110-1301, USA.
#
#    In addition, as a special exception, the copyright holders give
#    permission to link the code of portions of this program with the OpenSSL
#    library.
#    You must obey the GNU General Public License in all respects for all of
#    the code used other than OpenSSL. If you modify file(s) with this
#    exception, you may extend this exception to your version of the file(s),
#    but you are not obligated to do so. If you do not wish to do so, delete
#    this exception statement from your version. If you delete this exception
#    statement from all source files in the program, then also delete it here.
#
#


import os
import logging
import pkg_resources    # access plugin egg
from deluge import component    # for systray
from deluge.plugins.pluginbase import GtkPluginBase
import gtk, gobject
from deluge.ui.client import client

import sidebar_menu
import label_config
import submenu

log = logging.getLogger(__name__)

NO_LABEL = "No Label"

def cell_data_label(column, cell, model, row, data):
    cell.set_property('text', str(model.get_value(row, data)))

class GtkUI(GtkPluginBase):
    def start(self):
        if self.label_menu:
            self.label_menu.on_show()

    def enable(self):
        self.plugin = component.get("PluginManager")
        self.label_menu = None
        self.labelcfg = None
        self.sidebar_menu = None
        self.load_interface()

    def disable(self):
        try:
            torrentmenu = component.get("MenuBar").torrentmenu
            torrentmenu.remove(self.label_menu) # ok

            self.labelcfg.unload() # ok
            self.sidebar_menu.unload()
            del self.sidebar_menu

            component.get("TorrentView").remove_column(_("Label"))
            log.debug(1.1)

        except Exception, e:
            log.debug(e)

    def load_interface(self):
        #sidebar
        #disabled
        if not self.sidebar_menu:
            self.sidebar_menu  = sidebar_menu.LabelSidebarMenu()
        #self.sidebar.load()

        #menu:
        log.debug("add items to torrentview-popup menu.")
        torrentmenu = component.get("MenuBar").torrentmenu
        self.label_menu = submenu.LabelMenu()
        torrentmenu.append(self.label_menu)
        self.label_menu.show_all()

        #columns:
        self.load_columns()

        #config:
        if not self.labelcfg:
            self.labelcfg  = label_config.LabelConfig(self.plugin)
        self.labelcfg.load()

        log.debug('Finished loading Label plugin')

    def load_columns(self):
        log.debug("add columns")

        component.get("TorrentView").add_text_column(_("Label"), status_field=["label"])