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

Last change on this file since 3959 was 3959, checked in by jrpelegrina, 3 years ago

Fix some files

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