def test_select_two_include_from_sinclude(self): qmodel = QueryModel() qmodel.sinclude(["column1", "column2"]) self.assertEqual(qmodel.select(), { "_id": False, "column1": True, "column2": True })
async def sync(self) -> List[dict]: logging.info("Syncing {} data...".format(self._entity_name)) cur = await self.find() if len(cur) > 0: gsidqm = QueryModel() gsidqm.sinclude(["gsis_id"]) cgsidd = await self.find(qm=gsidqm) cgsids = list(set([r["gsis_id"] for r in cgsidd])) schqm = QueryModel() schqm.cstart("finished", True) if len(cgsids) > 0: schqm.cand("gsis_id", cgsids, Operator.NIN) sch = await self._scheduleManager.find(qm=schqm) else: sch = await self._scheduleManager.find() return await self.save(self._queryAPI(sch))