171b2bff06cb894f5c97539d82ecaf6df04deb1b
[meabook] / controller.py
1 """
2 Meabook controller class
3 """
4
5 class MeabookController:
6     def __init__(self, model, view_class, config):
7         self.config = config
8         self.model = model
9         self.view = view_class(self)
10         self.view.start()
11
12     def stop(self):
13         """Exit the program."""
14
15         self.config.save()
16         self.model.close()
17
18     def get_items(self, parent=0):
19         """Gets from model items with selected level."""
20
21         folders_list = list(self.get_all_folders(parent))
22         files_list = [(fields['cn'], internal_name, 'fi') for internal_name, \
23             fields in self.model.get_files(['cn'], parent).items()]
24         return folders_list + files_list
25
26     def get_item(self, _id):
27         """Gets person info."""
28
29         info = []
30         entry = self.model.get_entry(_id)
31         entry_fields = entry.keys()
32         ordered_fields = []
33         fields_to_show = self.config.get_fields()
34         for field in dict().fromkeys(self.config.get_order()).keys():
35             if (field in fields_to_show) and (field in entry_fields):
36                 info.append((field, entry[field]))
37         return info
38
39     def get_all_folders(self, parent=0):
40         """Gets all filials."""
41
42         return [(name, internal_name, 'fo') for name, internal_name in \
43             self.model.get_folders(parent)]
44
45     def get_all_subfolders(self):
46         """Gets all departments."""
47
48         return [(name, internal_name, 'fo') for name, internal_name in \
49             self.model.get_folders()]
50
51     def get_all_files(self):
52         """Gets all peoples."""
53
54         # FIXME: read field names from Config
55         files_dict = self.model.get_files(['cn'])
56         return [(files_dict[internal_name]['cn'], internal_name, 'file') for \
57             internal_name in files_dict]
58
59     def get_fields(self):
60         """Returns all exists fields from model."""
61
62         return self.model.get_fields()
63
64     def show_about_dialog(self):
65         """Shows view About Dialog."""
66
67         self.view.create_about_dialog()
68
69     def show_configuration_dialog(self):
70         """Shows view Configuration Dialog."""
71
72         self.view.create_configuration_dialog(self, self.config)
73
74     def show_import_dialog(self):
75         """Shows view IMport Dialog."""
76
77         path = self.view.create_import_dialog()
78         if not path:
79             return False
80
81         from meabook.parsers.ldif_parser import MeabookLDIFParser
82         parser = MeabookLDIFParser(open(path), self.model)
83         try:
84             self.view.create_progress_dialog()
85             parser.parse()
86         except:
87             self.view.create_information_dialog(_('Import error'), \
88                 _('Unsupported file format!'))