6f897c1a2eab0adf8a34daf773d313efffd33cfa
[meabook] / controller.py
1 """
2 Meabook controller class
3 """
4
5 from meabook.constants import *
6 from meabook.caller import PhoneCaller
7 from gettext import gettext as _
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['cn'], internal_name, 'f') for internal_name, \
34             fields in self.model.get_files(['cn'], parent).items()]
35         return folders_list + files_list
36
37     def get_item(self, _id):
38         """Gets person info."""
39
40         info = []
41         entry = self.model.get_entry(_id)
42         entry_fields = entry.keys()
43         fields_to_show = self.config.get_fields()
44         ordered_fields = []
45         for field in self.config.get_order():
46             if field not in ordered_fields:
47                 ordered_fields.append(field)
48         for field in ordered_fields:
49             if (field in fields_to_show) and (field in entry_fields):
50                 info.append((field, entry[field]))
51         if not 'image' in fields_to_show and 'image' in entry.keys():
52             info.append(('image', entry['image']))
53         return info
54
55     def get_all_folders(self, parent=0):
56         """Gets all filials."""
57
58         return [(name, internal_name, 'd') for name, internal_name in \
59             self.model.get_folders(parent)]
60
61     def get_all_subfolders(self):
62         """Gets all departments."""
63
64         return [(name, internal_name, 'd') for name, internal_name in \
65             self.model.get_folders()]
66
67     def get_files_by_pattern_ex(self, key, pattern=''):
68         """Gets all peoples matched pattern."""
69
70         files_dict = self.model.get_files_by_pattern(['cn', 'o', 'ou'], \
71             key, pattern, False)
72         return [("%s" % files_dict[internal_name]['cn'], "%s / %s" % (
73             files_dict[internal_name]['o'], files_dict[internal_name]['ou']), \
74             internal_name, TYPE_FILE) for internal_name in files_dict]
75
76     def get_files_by_pattern(self, pattern=''):
77         """Gets all peoples matched pattern."""
78
79         files_dict = self.model.get_files_by_pattern(['cn', 'o', 'ou'], \
80             'cn', pattern, True)
81         return [("%s" % files_dict[internal_name]['cn'], "%s / %s" % (
82             files_dict[internal_name]['o'], files_dict[internal_name]['ou']), \
83             internal_name, TYPE_FILE) for internal_name in files_dict]
84
85     def get_fields(self):
86         """Returns all exists fields from model."""
87
88         return self.model.get_fields()
89
90     def get_localized_fields(self):
91         """Returns dict with original and localized filed names."""
92
93         return dict([(_(field).decode('utf-8'), field) for field in \
94             self.model.get_fields()])
95
96     def show_about_dialog(self):
97         """Shows view About Dialog."""
98
99         self.view.create_about_dialog()
100
101     def show_configuration_dialog(self):
102         """Shows view Configuration Dialog."""
103
104         self.view.create_configuration_dialog(self, self.config.get_fields(), \
105             self.config.get_order())
106
107     def show_fields_configuration_dialog(self, parent):
108         """Shows FieldsConfigurationDialog."""
109
110         selected_fields = self.view.create_fields_configuration_dialog( \
111             self.get_localized_fields(), self.config.get_fields(), parent)
112         if selected_fields is not None:
113             self.config.set_fields(selected_fields)
114         return selected_fields
115
116     def show_fields_order_configuration_dialog(self):
117         """Shows FieldsOrderConfigurationDialog."""
118
119         selected_fields = self.view.create_fields_order_configuration_dialog( \
120             self.get_localized_fields(), self.config.get_order())
121         if selected_fields is not None:
122             self.config.set_order(selected_fields)
123         return selected_fields
124
125     def show_search_dialog(self):
126         """Shows view Search Dialog."""
127
128         self.view.create_search_dialog(self)
129
130     def show_import_dialog(self):
131         """Shows view IMport Dialog."""
132
133         path = self.view.create_import_dialog()
134         if not path:
135             return False
136
137         from meabook.parsers.ldif_parser import MeabookLDIFParser
138         parser = MeabookLDIFParser(open(path), self.model)
139         try:
140             self.view.create_progress_dialog(_('Importing'))
141             #FIXME: implement update_fraction
142             parser.parse()
143             self.config.set_fields(self.model.get_fields())
144             self.config.set_order(self.model.get_fields())
145         except:
146             self.view.create_information_dialog(_('Import error'), \
147                 _('Unsupported file format!'))
148         else:
149             return True