twblue/src/extra/autocompletionUsers/settings.py

# -*- coding: utf-8 -*-
import storage
import wx
import config
import wx_settings
import manage
import output
from mysc.thread_utils import call_threaded

class autocompletionSettings(object):
 def __init__(self, window):
  super(autocompletionSettings, self).__init__()
  self.window = window
  self.dialog = wx_settings.autocompletionSettingsDialog()
  self.dialog.friends_buffer.SetValue(config.main["mysc"]["save_friends_in_autocompletion_db"])
  self.dialog.followers_buffer.SetValue(config.main["mysc"]["save_followers_in_autocompletion_db"])
  self.dialog.viewList.Bind(wx.EVT_BUTTON, self.view_list)
  if self.dialog.ShowModal() == wx.ID_OK:
   call_threaded(self.add_users_to_database)

 def add_users_to_database(self):
  config.main["mysc"]["save_friends_in_autocompletion_db"] = self.dialog.friends_buffer.GetValue()
  config.main["mysc"]["save_followers_in_autocompletion_db"] = self.dialog.friends_buffer.GetValue()
  output.speak(_(u"Updating database... You can close this window now. A message will tell you when the process finishes."))
  database = storage.storage()
  if self.dialog.followers_buffer.GetValue() == True:
   buffer = self.window.search_buffer("people", "followers")
   for i in buffer.db.settings[buffer.name_buffer]:
    database.set_user(i["screen_name"], i["name"], 1)
  else:
   database.remove_by_buffer(1)
  if self.dialog.friends_buffer.GetValue() == True:
   buffer = self.window.search_buffer("people", "friends")
   for i in buffer.db.settings[buffer.name_buffer]:
    database.set_user(i["screen_name"], i["name"], 2)
  else:
   database.remove_by_buffer(2)
  wx_settings.show_success_dialog()
  self.dialog.Destroy()
  
 def view_list(self, ev):
  q = manage.autocompletionManage(self.window)


def execute_at_startup(window):
 database = storage.storage()
 if config.main["mysc"]["save_followers_in_autocompletion_db"] == True and config.main["other_buffers"]["show_followers"] == True:
  buffer = window.search_buffer("people", "followers")
  for i in buffer.db.settings[buffer.name_buffer]:
   database.set_user(i["screen_name"], i["name"], 1)
 else:
  database.remove_by_buffer(1)
 if config.main["mysc"]["save_friends_in_autocompletion_db"] == True and config.main["other_buffers"]["show_friends"] == True:
  buffer = window.search_buffer("people", "friends")
  for i in buffer.db.settings[buffer.name_buffer]:
   database.set_user(i["screen_name"], i["name"], 2)
 else:
  database.remove_by_buffer(2)