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