source: lliurex-gdrive/trunk/fuentes/lliurex-gdrive-gui.install/usr/share/lliurex-gdrive/ProfileBox.py @ 3981

Last change on this file since 3981 was 3981, checked in by jrpelegrina, 2 years ago

Add disable indicator option

File size: 17.9 KB
Line 
1#!/usr/bin/env python
2
3
4import gi
5gi.require_version('Gtk', '3.0')
6from gi.repository import Gtk, Pango, GdkPixbuf, Gdk, Gio, GObject,GLib
7
8import copy
9import gettext
10import Core
11
12import Dialog
13import time
14import threading
15import multiprocessing
16import sys
17import os
18import psutil
19
20import gettext
21gettext.textdomain('lliurex-gdrive')
22_ = gettext.gettext
23
24
25
26RSRC="/usr/share/lliurex-gdrive/"
27CSS_FILE="/usr/share/lliurex-gdrive/lliurex-gdrive.css"
28PROFILE_IMAGE=RSRC+"rsrc/profile.svg"
29FOLDER_IMAGE=RSRC+"rsrc/folder.svg"
30MOUNT_ON_IMAGE=RSRC+"rsrc/mount_on.svg"
31MOUNT_OFF_IMAGE=RSRC+"rsrc/mount_off.svg"
32EDIT_IMAGE=RSRC+"rsrc/edit.svg"
33DELETE_IMAGE=RSRC+"rsrc/trash.svg"
34MAX_RETRY_INTENTS=900
35MSG_CHANGE_SUCCESS=_("Changes applied successfully")
36MSG_CHANGE_ERROR=_("An error ocurred applying changes. See log file for more information")
37#DISABLE_INDICATOR="~/.config/lliurex-google-drive-profiles/disableIndicator"
38
39
40class ProfileBox(Gtk.VBox):
41       
42        def __init__(self):
43               
44                Gtk.VBox.__init__(self)
45               
46                self.core=Core.Core.get_core()
47               
48                builder=Gtk.Builder()
49                ui_path=RSRC + "/rsrc/lliurex-gdrive.ui"
50                builder.add_from_file(ui_path)
51
52                self.main_box=builder.get_object("profile_data_box")
53                self.profiles_list_label=builder.get_object("profiles_list_label")
54                self.profile_list_box=builder.get_object("profile_list_box")
55                self.profile_list_vp=builder.get_object("profile_list_viewport")
56                self.msg_label=builder.get_object("msg_label")
57
58                image = Gtk.Image()
59                image.set_from_stock(Gtk.STOCK_ADD,Gtk.IconSize.MENU)
60                #self.indicator_label=builder.get_object("indicator_label")
61                #self.indicator_switch=builder.get_object("indicator_switch")
62
63                #self.disable_indicator=os.path.expanduser(DISABLE_INDICATOR)
64                #if os.path.exists(self.disable_indicator):
65                        #self.indicator_switch.set_active(False)
66
67
68                self.add_new_profile_button=builder.get_object("add_new_profile_button")
69                self.add_new_profile_button.set_image(image)
70                self.new_profile_window=builder.get_object("new_profile_window")
71                self.profile_label=builder.get_object("profile_label")
72                self.profile_entry=builder.get_object("profile_entry")
73                self.email_label=builder.get_object("email_label")
74                self.email_entry=builder.get_object("email_entry")
75                self.mountpoint_label=builder.get_object("mountpoint_label")
76                self.mountpoint_entry=builder.get_object("mountpoint_entry")
77                self.automount_label=builder.get_object("automount_label")
78                self.automount_entry=builder.get_object("automount_entry")
79                self.profile_msg=builder.get_object("profile_msg")
80                self.profile_pbar=builder.get_object("profile_pbar")
81                self.accept_add_profile_button=builder.get_object("accept_add_profile_button")
82                self.cancel_add_profile_button=builder.get_object("cancel_add_profile_button")
83               
84               
85               
86                self.pack_start(self.main_box,True,True,0)
87                self.connect_signals()
88                self.set_css_info()
89
90                self.current_status={}
91                self.edition=False
92                self.profile_pbar.hide()
93
94                self.init_threads()
95
96                #self.current_var=None
97                #self.current_id=None
98               
99                #self.thread=threading.Thread()
100                #self.thread_ret=None
101               
102        #def __init__
103       
104        def init_threads(self):
105
106                self.create_profile_t=multiprocessing.Process(target=self.create_profile)
107                self.create_mountpoint_t=threading.Thread(target=self.create_mountpoint)
108                self.edit_profile_t=threading.Thread(target=self.edit_profile)
109
110               
111                self.create_profile_t.daemon=True
112                self.create_mountpoint_t.daemon=True
113                self.edit_profile_t.daemon=True
114
115                self.create_mountpoint_t.done=False
116                self.edit_profile_t.done=False
117
118                self.create_profile_t.launched=False
119                self.create_mountpoint_t.launched=False
120                self.edit_profile_t.launched=False
121               
122                GObject.threads_init()
123
124
125        def set_css_info(self):
126               
127                self.style_provider=Gtk.CssProvider()
128
129                f=Gio.File.new_for_path(CSS_FILE)
130                self.style_provider.load_from_file(f)
131
132                Gtk.StyleContext.add_provider_for_screen(Gdk.Screen.get_default(),self.style_provider,Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION)
133                self.profiles_list_label.set_name("OPTION_LABEL")
134                self.profile_label.set_name("OPTION_LABEL")
135                self.email_label.set_name("OPTION_LABEL")
136                self.mountpoint_label.set_name("OPTION_LABEL")
137                self.automount_label.set_name("OPTION_LABEL")
138                self.msg_label.set_name("MSG_LABEL")
139                self.profile_msg.set_name("MSG_LABEL")
140               
141                       
142        #def set-css_info
143       
144               
145        def connect_signals(self):
146
147                self.add_new_profile_button.connect("clicked",self.add_new_profile_button_clicked)
148                self.accept_add_profile_button.connect("clicked",self.accept_add_profile_clicked)
149                self.cancel_add_profile_button.connect("clicked",self.cancel_add_profile_clicked)
150                self.new_profile_window.connect("delete_event",self.hide_window)
151
152
153        #def connect_signals
154       
155        def load_info(self,info):
156               
157                self.profiles_info=info
158
159                for item in self.profiles_info:
160                        profile=item
161                        email=self.profiles_info[item]["email"]
162                        mountpoint=self.profiles_info[item]["mountpoint"]
163                        self.new_profile_button(profile,email,mountpoint)
164                        print self.profiles_info[item]["automount"]
165
166
167                print self.profiles_info       
168       
169        #def load_info()               
170
171        def hide_window(self,widget,event):
172               
173                widget.hide()
174                return True
175
176        #def hide_window       
177
178        def init_profile_dialog_button(self):
179
180                self.accept_add_profile_button.show()
181                image = Gtk.Image()
182                image.set_from_stock(Gtk.STOCK_CANCEL,Gtk.IconSize.MENU)
183                self.cancel_add_profile_button.set_image(image)
184                self.cancel_add_profile_button.set_label(_("Cancel"))
185                self.cancel_add_profile_button.show()
186
187        #def init_profile_dialog_button
188
189        def disable_entry_profile_dialog(self):
190
191                self.profile_entry.set_editable(False)
192                self.email_entry.set_editable(False)
193                self.mountpoint_entry.set_sensitive(False)
194                self.automount_entry.set_sensitive(False)
195
196        #def disable_entry_profile_dialog
197       
198        def enable_entry_profile_dialog(self):
199
200                if not self.edition:
201                        self.profile_entry.set_editable(True)
202                        self.email_entry.set_editable(True)
203
204                else:
205                        self.profile_entry.set_editable(False)
206                        self.email_entry.set_editable(False)
207
208                self.mountpoint_entry.set_sensitive(True)
209                self.automount_entry.set_sensitive(True)       
210
211        #def enable_entry_profile_dialog       
212
213        def change_cancel_button(self):
214               
215                image = Gtk.Image()
216                image.set_from_stock(Gtk.STOCK_CLOSE,Gtk.IconSize.MENU)
217                self.cancel_add_profile_button.set_image(image)
218                self.cancel_add_profile_button.set_label(_("Close"))   
219                self.cancel_add_profile_button.show()   
220
221        #def change_cancel_button       
222
223        def add_new_profile_button_clicked(self,widget):
224
225                self.edition=False
226
227                self.enable_entry_profile_dialog()
228                self.profile_entry.set_text("")
229                self.email_entry.set_text("")
230                self.mountpoint_entry.set_filename(os.environ["HOME"])
231                self.automount_entry.set_active(False)
232                self.profile_msg.set_text("")
233               
234
235                self.init_threads()
236               
237                self.msg_label.set_text("")
238                self.profile_msg.hide()
239                self.profile_pbar.hide()
240                self.init_profile_dialog_button()
241                self.new_profile_window.show()
242               
243
244        #def add_new_profile_button_clicked     
245
246        def delete_profile_clicked(self,button,hbox):
247
248                self.msg_label.set_text("")
249                profile=hbox.get_children()[1].get_text().split("\n")[0]
250                profile=profile.decode("utf-8")
251                self.profiles_info.pop(profile)
252
253                delete=self.core.LliurexGoogleDriveManager.delete_profile(self.profiles_info,profile)
254                self.msg_label.show()
255
256                if delete:
257                        self.profile_msg.set_name("MSG_LABEL")
258                        self.msg_label.set_text(MSG_CHANGE_SUCCESS)
259                        self.profile_list_box.remove(hbox)
260                else:
261                        self.msg_label.set_name("MSG_ERROR_LABEL")
262                        self.msg_label.set_text(MSG_CHANGE_ERROR)
263
264                self.profiles_info=self.core.LliurexGoogleDriveManager.profiles_config.copy()
265
266
267        # def delete_profile_clicked
268               
269
270        def sync_profile_clicked(self,button,hbox):
271
272                self.msg_label.set_text("")
273                profile=hbox.get_children()[1].get_text().split("\n")[0]
274                mountpoint=hbox.get_children()[3].get_text()
275                status_info=self.core.LliurexGoogleDriveManager.check_mountpoint_status(mountpoint)
276                self.msg_label.show()
277                self.msg_label.set_name("MSG_LABEL")
278                if self.current_status[profile]==status_info["status"]:
279
280                        status_mod=self.core.LliurexGoogleDriveManager.sync_profile(profile,mountpoint)
281                        if status_mod["result"]:
282                                status_info=self.core.LliurexGoogleDriveManager.check_mountpoint_status(mountpoint)
283                                self.msg_label.set_text(MSG_CHANGE_SUCCESS)
284                                # if status_mod["action"]=="Mount":
285                                #       mount_image=Gtk.Image.new_from_file(MOUNT_ON_IMAGE)
286                                #       button.set_tooltip_text(_("Dismount profile"))
287                                #       button.set_name("MOUNT_ITEM_BUTTON")
288                                # else:
289                                #       mount_image=Gtk.Image.new_from_file(MOUNT_OFF_IMAGE)
290                                #       button.set_tooltip_text(_("Mount profile"))
291                                #       button.set_name("DELETE_ITEM_BUTTON")
292
293                        else:
294                                self.msg_label.set_name("MSG_ERROR_LABEL")
295                                self.msg_label.set_text(MSG_CHANGE_ERROR)       
296                               
297                       
298
299                info=self.item_status_info(status_info)
300                self.current_status[profile]=status_info["status"]
301                button.set_tooltip_text(info["tooltip"])
302                button.set_name(info["css"])           
303                hbox.get_children()[4].set_image(info["img"])
304
305       
306        #def mount_profile_clicked
307
308        def edit_profile_clicked(self,button,hbox):
309
310                self.edition=True
311                self.enable_entry_profile_dialog()
312
313                self.profile_to_edit=hbox               
314                profile=self.profile_to_edit.get_children()[1].get_text().split("\n")[0]
315                self.profile_entry.set_text(profile)
316                #self.profile_entry.set_editable(False)
317                email=self.profile_to_edit.get_children()[1].get_text().split("\n")[1]
318                self.email_entry.set_text(email)
319                #self.email_entry.set_editable(False)
320                mountpoint=self.profile_to_edit.get_children()[3].get_text()
321                self.mountpoint_entry.set_filename(mountpoint)
322                automount=self.profiles_info[profile.decode("utf-8")]["automount"]
323                self.automount_entry.set_active(automount)
324               
325
326                self.init_threads()
327
328                self.profile_msg.hide()
329                self.profile_pbar.hide()
330                self.msg_label.set_text("")
331                self.init_profile_dialog_button()
332                self.new_profile_window.show() 
333
334        #def edit_profile_clicked
335
336
337        def new_profile_button(self,profile_name,email,mountpoint):
338               
339                hbox=Gtk.HBox()
340                profile_image=Gtk.Image.new_from_file(PROFILE_IMAGE)
341                profile_image.set_margin_left(10)
342                profile_image.set_halign(Gtk.Align.CENTER)
343                profile_image.set_valign(Gtk.Align.CENTER)
344                profile_info="<span font='Roboto'><b>"+profile_name+"</b></span>\n"+"<span font='Roboto'>"+email+"</span>"
345                profile=Gtk.Label()
346                profile.set_markup(profile_info)
347                profile.set_margin_left(5)
348                profile.set_margin_right(15)
349                profile.set_margin_top(21)
350                profile.set_margin_bottom(21)
351                folder_image=Gtk.Image.new_from_file(FOLDER_IMAGE)
352                folder_image.set_margin_left(20)
353                folder_image.set_halign(Gtk.Align.CENTER)
354                folder_image.set_valign(Gtk.Align.CENTER)
355                folder=Gtk.Label()
356                folder.set_text(mountpoint)
357                folder.set_margin_left(10)
358                delete=Gtk.Button()
359                delete_image=Gtk.Image.new_from_file(DELETE_IMAGE)
360                delete.add(delete_image)
361                delete.set_halign(Gtk.Align.CENTER)
362                delete.set_valign(Gtk.Align.CENTER)
363                delete.set_name("DELETE_ITEM_BUTTON")
364                delete.connect("clicked",self.delete_profile_clicked,hbox)
365                delete.set_tooltip_text(_("Delete profile"))
366                edit=Gtk.Button()
367                edit_image=Gtk.Image.new_from_file(EDIT_IMAGE)
368                edit.add(edit_image)
369                edit.set_halign(Gtk.Align.CENTER)
370                edit.set_valign(Gtk.Align.CENTER)
371                edit.set_name("EDIT_ITEM_BUTTON")
372                edit.connect("clicked",self.edit_profile_clicked,hbox)
373                edit.set_tooltip_text(_("Edit profile"))
374                mount=Gtk.Button()
375               
376                status_info=self.core.LliurexGoogleDriveManager.check_mountpoint_status(mountpoint)
377                self.current_status[profile_name.encode("utf-8")]=status_info["status"]
378                print self.current_status
379                info=self.item_status_info(status_info)
380                # if status["status"]:
381                #       mount_image=Gtk.Image.new_from_file(MOUNT_ON_IMAGE)
382                #       mount.set_tooltip_text(_("Dismount profile"))
383                #       mount.set_name("MOUNT_ITEM_BUTTON")
384
385                # else:
386                #       mount_image=Gtk.Image.new_from_file(MOUNT_OFF_IMAGE)
387                #       mount.set_tooltip_text(_("Mount profile"))
388                #       mount.set_name("DELETE_ITEM_BUTTON")
389                mount_image=info["img"]
390                mount.set_tooltip_text(info["tooltip"])
391                mount.set_name(info["css"])
392                mount.add(mount_image)
393                mount.set_halign(Gtk.Align.CENTER)
394                mount.set_valign(Gtk.Align.CENTER)
395                #mount.set_name("MOUNT_ITEM_BUTTON")
396                mount.connect("clicked",self.sync_profile_clicked,hbox)
397               
398                hbox.pack_start(profile_image,False,False,0)
399                hbox.pack_start(profile,False,False,0)
400                hbox.pack_start(folder_image,False,False,0)
401                hbox.pack_start(folder,False,False,0)
402                hbox.pack_end(delete,False,False,10)
403                hbox.pack_end(edit,False,False,10)
404                hbox.pack_end(mount,False,False,10)
405                hbox.show_all()
406                hbox.set_name("PROFILE_BOX")
407                self.profile_list_box.pack_start(hbox,False,False,5)
408                self.profile_list_box.queue_draw()
409                hbox.queue_draw()
410               
411        #def new_profile_button
412
413        def item_status_info(self,status_info):
414       
415                size=status_info["size"]
416                used=status_info["used%"]
417
418                if status_info["status"]:
419                        img=Gtk.Image.new_from_file(MOUNT_ON_IMAGE)
420                        tooltip=_("Dismount now")
421                        css="MOUNT_ITEM_BUTTON"
422               
423                else:
424                        img=Gtk.Image.new_from_file(MOUNT_ON_IMAGE)
425                        tooltip=_("Mount now")
426                        css="DELETE_ITEM_BUTTON"       
427
428                return {"img":img ,"tooltip":tooltip, "css":css}       
429
430        #def item_status_info                   
431
432        def accept_add_profile_clicked(self,widget):
433
434                self.disable_entry_profile_dialog()
435               
436                profile=self.profile_entry.get_text()
437
438                self.new_profile=profile.strip().decode("utf-8")
439                email=self.email_entry.get_text()
440                self.new_email=email.strip()
441                self.new_mountpoint=self.mountpoint_entry.get_filename().decode("utf-8")
442                self.new_automount=self.automount_entry.get_state()
443
444                check_form=self.check_profile_info()
445                self.profile_msg.show()
446
447                if check_form["result"]:
448                        self.profiles_info[self.new_profile]={}
449                        self.profiles_info[self.new_profile]["email"]=self.new_email
450                        self.profiles_info[self.new_profile]["mountpoint"]=self.new_mountpoint
451                        self.profiles_info[self.new_profile]["automount"]=self.new_automount
452                        self.profile_msg.set_name("MSG_LABEL")
453                        if not self.edition:
454                                if not self.create_profile_t.launched:
455                                        self.profile_msg.set_text(_("Connecting with google to get account access..."))
456                                        self.profile_pbar.show()
457                                        self.retry=0
458                                        GLib.timeout_add(100,self.pulsate_add_profile)
459                               
460               
461                        else:
462                                if not self.edit_profile_t.launched:
463                                        self.profile_msg.set_text(_("Apply changes"))
464                                        self.profile_pbar.show()
465                                        GLib.timeout_add(100,self.pulsate_edit_profile) 
466                else:
467                        self.profile_msg.set_name("MSG_ERROR_LABEL")
468                        self.profile_msg.set_text(check_form["msg"])
469                        self.enable_entry_profile_dialog()
470                               
471                               
472               
473       
474        #def accept_add_profile_clicked
475
476        def check_profile_info(self):
477
478                msg_check=""
479
480                check_form=self.core.LliurexGoogleDriveManager.check_profile_info(self.new_profile,self.new_mountpoint,self.edition)
481
482                if not check_form["result"]:
483                        if check_form["code"]==1:
484                                msg_check=_("You must indicate a profile")
485
486                        elif check_form["code"]==2:
487                                msg_check=_("Profile can not contain blanks")
488
489                        elif check_form["code"]==3 :
490                                msg_check=_("Profile already exists")
491                       
492                        else:
493                                msg_check=_("Mounting point already used by another profile")   
494
495
496                return {"result":check_form["result"],"msg":msg_check}
497
498        #def check_profile_info
499
500        def pulsate_add_profile(self):
501
502                self.retry=self.retry+1
503                self.profile_pbar.pulse()
504               
505                if not self.create_profile_t.launched:
506                        self.accept_add_profile_button.hide()
507                        self.cancel_add_profile_button.hide()
508               
509                        self.create_profile_t.start()
510                        self.create_profile_t.launched=True
511
512                if not self.create_profile_t.is_alive():
513                        if not self.create_mountpoint_t.launched:
514                                self.profile_msg.set_name("MSG_LABEL")
515                                self.profile_msg.set_text(_("Creating profile... "))
516                                self.create_mountpoint_t.start()
517                                self.create_mountpoint_t.launched=True
518
519                        if self.create_mountpoint_t.done:
520                                self.profile_pbar.hide()
521                                if self.create_result:
522                                                self.new_profile_button(self.new_profile,self.new_email,self.new_mountpoint)
523                                                self.profiles_info=self.core.LliurexGoogleDriveManager.profiles_config.copy()   
524                                                self.profile_msg.set_text(_("Profile created successfully"))
525                                                self.change_cancel_button()
526                                               
527                                else:
528                                       
529                                        self.profile_msg.set_name("MSG_ERROR_LABEL")
530                                        self.profile_msg.set_text(_("An error ocurred creating profile. See log file for more information"))
531                                        self.cancel_add_profile_button.show()
532                                       
533                               
534                                return False   
535                                               
536                if self.create_profile_t.launched:
537                        if self.create_profile_t.is_alive():
538                                self.kill_create_profile()
539                                return True     
540                               
541
542                if self.create_mountpoint_t.launched:
543                        if not self.create_mountpoint_t.done:
544                                return True             
545
546        #def_pulsate_add_profile                                               
547
548        def kill_create_profile(self):
549       
550                if self.retry>MAX_RETRY_INTENTS:
551                        parent=psutil.Process(self.create_profile_t.pid)
552                        for child in parent.children(recursive=True):
553                                child.kill()
554                                self.create_profile_t.terminate()
555                                self.profile_msg.set_name("MSG_ERROR_LABEL")
556                                self.profile_msg.set_text(_("Error getting authorization"))
557
558                return True     
559
560        #def kill_create_profile       
561
562        def create_profile(self):
563
564                result=self.core.LliurexGoogleDriveManager.create_profile(self.new_profile)
565
566        #def create_profile     
567
568        def create_mountpoint(self):
569
570                self.create_result=self.core.LliurexGoogleDriveManager.create_mountpoint(self.profiles_info,self.new_profile)
571                self.create_mountpoint_t.done=True
572               
573        #def create_mountpoint 
574
575        def pulsate_edit_profile(self):
576
577                if not self.edit_profile_t.launched:
578                        self.edit_profile_t.start()
579                        self.edit_profile_t.launched=True
580                        self.accept_add_profile_button.hide()
581                        self.cancel_add_profile_button.hide()
582
583                if self.edit_profile_t.done:
584                        self.profile_pbar.hide()
585                        if self.edit_result:
586                                self.profile_msg.set_name("MSG_LABEL")
587                                self.profile_to_edit.get_children()[3].set_text(self.new_mountpoint)
588                                self.profiles_info=self.core.LliurexGoogleDriveManager.profiles_config.copy()   
589                                self.profile_msg.set_text(MSG_CHANGE_SUCCESS)
590                                self.change_cancel_button()
591                       
592                        else:
593                                self.profile_msg.set_name("MSG_ERROR_LABEL")
594                                self.profile_msg.set_text(MSG_CHANGE_ERROR)
595                                self.cancel_add_profile_button.show()
596                        return False
597                       
598
599                if self.edit_profile_t.launched:
600                        if not self.edit_profile_t.done:
601                                return True
602
603        #def pulsate_edit_profile                       
604
605        def edit_profile(self):
606
607                self.edit_result=self.core.LliurexGoogleDriveManager.edit_profile(self.profiles_info,self.new_profile)
608                self.edit_profile_t.done=True
609
610        #def edit_profile       
611
612        def cancel_add_profile_clicked(self,widget):
613               
614                self.new_profile_window.hide()
615
616       
617        #def cancel_add_profile_clicked
618               
619#class profilebox
Note: See TracBrowser for help on using the repository browser.