def fetch_worksheet_ids(self, group_id, worksheet_ids=None): query = Worksheet.all() query.what("uid") query.filter("group_id =", group_id) wids = [] worksheets = query.fetch() for ws in worksheets: wids.append(ws.key()) if worksheet_ids != None: wids = list(set(wids).intersection(set(worksheet_ids))) return wids
def execute(self): rtn = self._new_rtn() worksheets = TaskService.get_instance().fetch_worksheets(self._get_user_id()) std = stdModel() workbasket = Worksheet() workbasket.uid = model.EMPTY_UID workbasket.ws_name = i18n.get_i18n_message(conf.get_preferred_language(), "task_label_workbasket") workbasket.taskcount = TaskService.get_instance().get_taskcount(False, self._get_user_id(), worksheet_id=model.EMPTY_UID) workbasket.creator_id = self._get_user_id() worksheets.insert(0, workbasket) trash = Worksheet() trash.uid = -100 trash.ws_name = i18n.get_i18n_message(conf.get_preferred_language(), "task_label_recyclebin") trash.taskcount = TaskService.get_instance().get_taskcount(True, self._get_user_id()) trash.creator_id = self._get_user_id() worksheets.append(trash) std.worksheets = worksheets std.contactGroups = ContactService.get_instance().fetch_my_groups(self._get_user_id()) stmworksheets = TaskService.get_instance().fetch_worksheets(self._get_user_id(), sharetome=True) std.worksheets.extend(stmworksheets) std.worksheetComponents = {} std.worksheetVersions = {} std.worksheetContacts = {} for worksheet in std.worksheets: if worksheet.group_id != None and worksheet.group_id != model.EMPTY_UID: std.worksheetComponents[worksheet.key()] = ContactService.get_instance().fetch_groupcomponents(worksheet.group_id) std.worksheetVersions[worksheet.key()] = ContactService.get_instance().fetch_groupversions(worksheet.group_id) std.worksheetContacts[worksheet.key()] = ContactService.get_instance().fetch_contacts_by_group(worksheet.group_id) else: myself = ContactService.get_instance().get_myself(self._get_user_id()) if myself is not None: std.worksheetContacts[worksheet.key()] = [myself] rtn.set_data(std) return rtn.to_json()