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()
