added english localization
[meabook] / controller.py
1 """
2 Meabook controller class
3 """
4
5 from meabook.constants import *
6 from gettext import gettext as _
7 from meabook.caller import PhoneCaller
8
9
10 class MeabookController:
11     def __init__(self, model, view_class, renderer_class, config):
12         self.config = config
13         self.caller = PhoneCaller()
14         self.model = model
15         self.view = view_class(self, renderer_class(), self.config)
16         self.view.start()
17
18     def stop(self):
19         """Exit the program."""
20
21         self.config.save()
22         self.model.close()
23
24     def call(self, number):
25         """Make a phone call."""
26
27         self.caller.call(number)
28
29     def get_items(self, parent=0):
30         """Gets from model items with selected level."""
31
32         folders_list = list(self.get_all_folders(parent))
33         files_list = [(fields[FIELD_LOWLEVEL], internal_name, TYPE_FILE) for \
34             internal_name, fields in self.model.get_files([FIELD_LOWLEVEL], \
35             parent).items()]
36         return folders_list + files_list
37
38     def get_item(self, _id):
39         """Gets person info."""
40
41         info = []
42         entry = self.model.get_entry(_id)
43         entry_fields = entry.keys()
44         fields_to_show = self.config.get_fields()
45         ordered_fields = []
46         for field in self.config.get_order():
47             if field not in ordered_fields:
48                 ordered_fields.append(field)
49         for field in ordered_fields:
50             if (field in fields_to_show) and (field in entry_fields):
51                 info.append((field, entry[field]))
52         if not FIELD_IMAGE in fields_to_show and FIELD_IMAGE in entry.keys():
53             info.append((FIELD_IMAGE, entry[FIELD_IMAGE]))
54         return info
55
56     def get_all_folders(self, parent=0):
57         """Gets all filials."""
58
59         return [(name, internal_name, TYPE_DIRECTORY) for name, internal_name \
60             in self.model.get_folders(parent)]
61
62     def get_all_subfolders(self):
63         """Gets all departments."""
64
65         return [(name, internal_name, TYPE_DIRECTORY) for name, internal_name \
66             in self.model.get_folders()]
67
68     def get_files_by_pattern_ex(self, key, pattern=''):
69         """Gets all peoples matched pattern."""
70
71         files_dict = self.model.get_files_by_pattern([FIELD_LOWLEVEL, \
72             FIELD_TOPLEVEL, FIELD_MIDDLELEVEL], key, pattern, False)
73         return [("%s" % files_dict[internal_name][FIELD_LOWLEVEL], "%s / %s" % (
74             files_dict[internal_name][FIELD_TOPLEVEL], \
75             files_dict[internal_name][FIELD_MIDDLELEVEL]), \
76             internal_name, TYPE_FILE) for internal_name in files_dict]
77
78     def get_files_by_pattern(self, pattern=''):
79         """Gets all peoples matched pattern."""
80
81         files_dict = self.model.get_files_by_pattern([FIELD_LOWLEVEL, \
82             FIELD_TOPLEVEL, FIELD_MIDDLELEVEL], FIELD_LOWLEVEL, pattern, True)
83         return [("%s" % files_dict[internal_name][FIELD_LOWLEVEL], "%s / %s" % (
84             files_dict[internal_name][FIELD_TOPLEVEL], \
85             files_dict[internal_name][FIELD_MIDDLELEVEL]), \
86             internal_name, TYPE_FILE) for internal_name in files_dict]
87
88     def get_fields(self):
89         """Returns all exists fields from model."""
90
91         return self.model.get_fields()
92
93     def get_localized_fields(self):
94         """Returns dict with original and localized filed names."""
95
96         return dict([(_(field).decode('utf-8'), field) for field in \
97             self.model.get_fields()])
98
99     def show_about_dialog(self):
100         """Shows view About Dialog."""
101
102         self.view.create_about_dialog()
103
104     def show_configuration_dialog(self):
105         """Shows view Configuration Dialog."""
106
107         self.view.create_configuration_dialog(self, self.config.get_fields(), \
108             self.config.get_order())
109
110     def show_fields_configuration_dialog(self, parent):
111         """Shows FieldsConfigurationDialog."""
112
113         selected_fields = self.view.create_fields_configuration_dialog( \
114             self.get_localized_fields(), self.config.get_fields(), parent)
115         if selected_fields is not None:
116             self.config.set_fields(selected_fields)
117         return selected_fields
118
119     def show_fields_order_configuration_dialog(self):
120         """Shows FieldsOrderConfigurationDialog."""
121
122         selected_fields = self.view.create_fields_order_configuration_dialog( \
123             self.get_localized_fields(), self.config.get_order())
124         if selected_fields is not None:
125             self.config.set_order(selected_fields)
126         return selected_fields
127
128     def show_search_dialog(self):
129         """Shows view Search Dialog."""
130
131         self.view.create_search_dialog(self)
132
133     def show_import_dialog(self):
134         """Shows view IMport Dialog."""
135
136         result = self.view.create_import_dialog()
137         if not result:
138             return False
139
140         import_type, path = result
141         if import_type == 'ldif':
142             from meabook.parsers.ldif_parser import MeabookLDIFParser
143             parser = MeabookLDIFParser(open(path), self.model)
144             try:
145                 self.view.create_progress_dialog(_('Importing'))
146                 #FIXME: implement update_fraction
147                 parser.parse()
148             except:
149                 self.view.create_information_dialog(_('Import error'), \
150                     _('Unsupported file format!'))
151                 return False
152         elif import_type == 'db':
153             import shutil
154             self.model.close()
155             try:
156                 shutil.copy(path, self.model._path)
157             except:
158                 self.model.connect()
159                 self.view.create_information_dialog(_('Import error'), \
160                     _('Unable to copy database file!'))
161                 return False
162             else:
163                 self.model.connect()
164
165         self.config.set_fields(self.model.get_fields())
166         self.config.set_order(self.model.get_fields())
167         return True