1505a67990c312bc993847f4e2eb9de6badb0ac8
[meabook] / controller.py
1 """
2 Meabook controller class
3 """
4
5 from meabook.constants import *
6 import gettext
7
8 _ = gettext.gettext
9
10
11 class MeabookController:
12     def __init__(self, model, view_class, renderer_class, config):
13         self.config = config
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 get_items(self, parent=0):
25         """Gets from model items with selected level."""
26
27         folders_list = list(self.get_all_folders(parent))
28         files_list = [(fields['cn'], internal_name, 'f') for internal_name, \
29             fields in self.model.get_files(['cn'], parent).items()]
30         return folders_list + files_list
31
32     def get_item(self, _id):
33         """Gets person info."""
34
35         info = []
36         entry = self.model.get_entry(_id)
37         entry_fields = entry.keys()
38         fields_to_show = self.config.get_fields()
39         ordered_fields = []
40         for field in self.config.get_order():
41             if field not in ordered_fields:
42                 ordered_fields.append(field)
43         for field in ordered_fields:
44             if (field in fields_to_show) and (field in entry_fields):
45                 info.append((field, entry[field]))
46         if not 'image' in fields_to_show and 'image' in entry.keys():
47             info.append(('image', entry['image']))
48         return info
49
50     def get_all_folders(self, parent=0):
51         """Gets all filials."""
52
53         return [(name, internal_name, 'd') for name, internal_name in \
54             self.model.get_folders(parent)]
55
56     def get_all_subfolders(self):
57         """Gets all departments."""
58
59         return [(name, internal_name, 'd') for name, internal_name in \
60             self.model.get_folders()]
61
62     def get_all_files(self):
63         """Gets all peoples."""
64
65         # FIXME: read field names from Config
66         files_dict = self.model.get_files(['cn'])
67         return [(files_dict[internal_name]['cn'], internal_name, TYPE_FILE) \
68             for internal_name in files_dict]
69
70     def get_all_files_by_pattern(self, pattern, separated=False):
71         """Gets all peoples matched pattern."""
72
73         if not separated:
74             return [(name, internal_name, TYPE_FILE) for internal_name, name \
75                 in self.model.get_files_by_pattern(pattern)]
76         else:
77             files_dict = self.model.get_files(['cn', 'o', 'ou'], pattern=pattern)
78             return [("%s" % files_dict[internal_name]['cn'], "%s / %s" % (
79                 files_dict[internal_name]['o'], files_dict[internal_name]['ou']), \
80                 internal_name, TYPE_FILE) for internal_name in files_dict]
81
82     def get_fields(self):
83         """Returns all exists fields from model."""
84
85         return self.model.get_fields()
86
87     def show_about_dialog(self):
88         """Shows view About Dialog."""
89
90         self.view.create_about_dialog()
91
92     def show_configuration_dialog(self):
93         """Shows view Configuration Dialog."""
94
95         self.view.create_configuration_dialog(self, self.config)
96
97     def show_search_dialog(self):
98         """Shows view Search Dialog."""
99
100         self.view.create_search_dialog(self)
101
102     def show_import_dialog(self):
103         """Shows view IMport Dialog."""
104
105         path = self.view.create_import_dialog()
106         if not path:
107             return False
108
109         from meabook.parsers.ldif_parser import MeabookLDIFParser
110         parser = MeabookLDIFParser(open(path), self.model)
111         try:
112             self.view.create_progress_dialog(_('Importing'))
113             #FIXME: implement update_fraction
114             parser.parse()
115             self.config.set_fields(self.model.get_fields())
116             self.config.set_order(self.model.get_fields())
117         except:
118             self.view.create_information_dialog(_('Import error'), \
119                 _('Unsupported file format!'))
120         else:
121             return True