bb156a0923a0aadf3d7b5595468163e74bc1091a
[stack/code/dboxswitch.git] / gui.py
1 # -*- coding: utf-8 -*-
2
3 """
4 Dboxswitch dropbox profile switcher
5
6 license: Modified BSD License
7
8 Copyright (c) 2012,  <stack@inventati.org>
9 All rights reserved.
10
11 Redistribution and use in source and binary forms, with or without
12 modification, are permitted provided that the following conditions are met:
13     * Redistributions of source code must retain the above copyright
14       notice, this list of conditions and the following disclaimer.
15     * Redistributions in binary form must reproduce the above copyright
16       notice, this list of conditions and the following disclaimer in the
17       documentation and/or other materials provided with the distribution.
18     * Neither the name of the <organization> nor the
19       names of its contributors may be used to endorse or promote products
20       derived from this software without specific prior written permission.
21
22 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
23 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
24 WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
25 DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
26 DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
27 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
29 ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
31 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32
33 """
34 import sys
35 import os
36 from time import sleep
37 from PyQt4 import QtGui, QtCore
38 from PyQt4.QtCore import SIGNAL
39 from qmenutooltip import QMenuToolTip
40
41 from apperror import AppError
42 from settings import appconf
43
44 class Gui(QtGui.QDialog):
45     def __init__(self, prManager):
46         """ Gui init, an QApplication is created and stored here, also a main window and a trayIcon
47         are created.
48         Default actions are builded and profile manager stored from argument (instance of ProfHandler """
49
50         self.app = QtGui.QApplication(sys.argv)
51         for i in range(3):
52                 if not QtGui.QSystemTrayIcon.isSystemTrayAvailable():
53                         sleep(4)
54                         else:
55                                 break
56         #check if system tray is avaiable on the system
57         if not QtGui.QSystemTrayIcon.isSystemTrayAvailable():
58             QtGui.QMessageBox.critical(None, "Systray",
59                     "I couldn't detect any system tray on this system.")
60             sys.exit(1)
61         
62         QtGui.QApplication.setQuitOnLastWindowClosed(False)
63     
64         super(Gui, self).__init__()
65
66         self.profiles = []
67
68         self.createActions()
69         self.createTrayIcon()
70         self.createMainDialog()
71
72         self.trayIcon.show()
73  
74         self.profileManager = prManager
75
76     def main(self): 
77         """ Like Gtk application this main executes the app, thi is somewhat writed for
78         compatibility in porting the app """
79
80         sys.exit(self.app.exec_())
81
82     def closeEvent(self, event):
83         """ Handle application closing """
84
85         if self.trayIcon.isVisible():
86             self.hide()
87             event.ignore()
88
89     def createActions(self):
90         """ Create actions for the various components """
91
92         self.manageprofiles = QtGui.QAction("Manage &Profiles", self,
93                 triggered=self.show)
94         self.quitAction = QtGui.QAction("&Quit", self,
95                 triggered=QtGui.qApp.quit)
96
97         #profile manager component
98         self.addProfileAction = QtGui.QAction("  Add &Profile  ", self,
99                 triggered=self.addProfile)
100
101     def createMainDialog(self):    
102         """ Build the main dialog layout """ 
103         
104         self.setWindowTitle("Profile manager / Dboxswitch - dropbox profile switcher")
105         #add a default action to create a profile
106         self.addButton = QtGui.QPushButton(QtGui.QIcon(appconf.addpicon), 
107                     "Add Profile")
108         self.addButton.clicked.connect(self.addProfile)
109         self.addButton.setVisible(False)
110
111         self.la            = QtGui.QFormLayout()
112
113         self.la.addRow(self.addButton)
114         self.la.sizePolicy = QtGui.QSizePolicy.Minimum
115         self.la.setVerticalSpacing(15)
116         self.la.setHorizontalSpacing(15)
117         self.formGroupBox = QtGui.QGroupBox("Manage profiles:")
118         self.formGroupBox.setLayout(self.la)
119         mainLayout = QtGui.QVBoxLayout()
120         mainLayout.addWidget(self.formGroupBox)
121         self.setLayout(mainLayout)
122
123
124     def show(self):
125         """ Show the main dialog for dealing with profiles """
126
127         profiles = self.profileManager.getProfilesList()
128
129         for pr in profiles:
130             prname = self.profileManager.getBaseProfileName(pr)
131             delButton = QtGui.QToolButton()
132             label     = QtGui.QLabel(prname)
133             self.la.addRow(label, delButton)
134             delAction = QtGui.QAction(QtGui.QIcon(appconf.delpicon),"Delete", self,
135                 triggered=self.deleteProfileAction(prname, (label, delButton)))
136             delButton.setDefaultAction(delAction)
137             self.profiles.append((label, delButton))
138
139         #if profiles was empty
140         if len(profiles) == 0:
141             self.addButton.setVisible(True)
142         else:
143             self.addButton.setVisible(False)
144
145         super(Gui, self).show()
146
147     def hide(self):
148         """ Hides the main dialog """
149         
150         #Hides all the profiles in the Dialog
151         #User can deal with profiles like removing the dir manually
152
153         for ws in self.profiles:
154             for w in ws:
155                 w.setVisible(False)
156                 w.setParent(None)
157         super(Gui, self).hide()
158
159     def createTrayIcon(self):
160          """ Builds a new tray icon with a context menu and an action for the profile manager menu """
161
162          #context menu build, right click
163          self.trayIconMenu = QtGui.QMenu(self)
164          self.trayIconMenu.addAction(self.manageprofiles)
165          self.trayIconMenu.addSeparator()
166          self.trayIconMenu.addAction(self.quitAction)
167
168          #create the tray with an incon
169          self.trayIcon = QtGui.QSystemTrayIcon(QtGui.QIcon(appconf.icon), self.app)
170          #attach a context menu for the right click to the tray
171          self.trayIcon.setContextMenu(self.trayIconMenu)
172          #baloon on hover for the tray
173          self.trayIcon.setToolTip(appconf.appname+" "+appconf.appversion+"\nRight Click to manage profiles.")
174          #left click profiles show for the tray
175          self.trayIcon.activated.connect(self.showTrayProfiles)
176
177
178     def showTrayProfiles(self,reason):
179         """ Pops up a system tray profile Manager with a list of activable profiles and an 
180         action to add a new One """
181
182         if reason in (QtGui.QSystemTrayIcon.Trigger, QtGui.QSystemTrayIcon.DoubleClick):
183             self.menuProfiles = QMenuToolTip()
184             self.menuProfiles.setTitle("Profiles")
185             QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_DontShowIconsInMenus, False)
186
187             #Get profiles from the ProfHandler embedded in the gui
188             profiles = self.profileManager.getProfilesList()
189
190             for prpath in profiles:
191                 pr = self.profileManager.getBaseProfileName(prpath)
192                 receiver = self.activateProfileAction(prpath)
193                 menuItem_Profile = self.menuProfiles.addAction(pr)
194                 if self.profileManager.isCurrentProfile(prpath):
195                     menuItem_Profile.setIcon(QtGui.QIcon(appconf.cpicon))
196                  
197                 #Using lambda function to pass additional arguments to the function, in this case the path of the profile
198                 self.connect(menuItem_Profile, SIGNAL('triggered()'), receiver)
199                 #set menu item ToolTip
200                 menuItem_Profile.setToolTip("Activate profile: "+pr)
201
202                 self.menuProfiles.addAction(menuItem_Profile)
203
204             self.menuProfiles.addSeparator()
205             #action and menu item for adding a New Profile
206             self.menuProfiles.addAction(self.addProfileAction)
207
208             self.menuProfiles.activateWindow()
209             self.menuProfiles.popup(QtGui.QCursor.pos())
210
211     def activateProfileAction(self, pr):
212         """ Returns a callable to be passed as an action for the switching profile mechanism 
213         to self.connect(menuItem_Profile... 
214         It compiles a function with a profile name as argument and also handle the displaying of errors."""
215
216         def f():
217             try:
218                 self.profileManager.activateProfile(pr)
219             except AppError, e:
220                 self.showError(str(e))
221         return f
222
223     def deleteProfileAction(self, pr, prWidgets):
224         """ Returns a callable to be passed as an action for the profile manager mechanism 
225         to self.connect(menuItem_Profile... 
226         It compiles a function with a profile name as argument and also handle the displaying of errors."""
227
228         def f():
229             try:
230                 self.profileManager.delProfile(pr)
231                 for w in prWidgets:
232                     w.setParent(None)
233                     w.setVisible(False)
234             except AppError, e:
235                 self.showError(str(e))
236         return f
237
238
239     def addProfile(self):
240         """ Gui frontend to add a new Profile, it requests the user a profile name 
241         through a QInputDialog and creates a new profile with the help of the ProfHandler embedded in the Gui """
242
243         self.setWindowTitle("Add New Profile - Dboxswitch - dropbox profile switcher")
244         self.resize(300, 100)
245         text, ok = QtGui.QInputDialog.getText(self, 'Input Dialog', 
246                             'Enter profile name:')
247         if ok:
248             try:
249                 self.profileManager.addProfile(unicode(text))
250             except AppError, e:
251                 self.showError(str(e))
252
253     def showError(self, err):
254         """ Display an error message through a QErrorMessage """
255
256         self.setWindowTitle("Error - Dboxswitch - dropbox profile switcher")
257         self.resize(200, 100)
258         err = QtGui.QErrorMessage.showMessage(QtGui.QErrorMessage.qtHandler(), "Error: "+err)