def get_namespaces(self): # Skip auth-ing to db in test operations if not self.argv: return ['4952435991248896_1'] query_pb = query_pb2.Query() helper.set_kind(query_pb, "__namespace__") client = apache_helper.get_datastore(PROJECT) namespace_entities = apache_helper.fetch_entities( PROJECT, '', query_pb, client) namespaces = [] for n in namespace_entities: # Get namespace name or id key_path = n.key.path[-1] if key_path.HasField('id'): name_or_id = key_path.id else: name_or_id = key_path.name # Avoid duplicates and test namespaces if len(str(name_or_id)) > 1 and name_or_id not in namespaces: namespaces.append(name_or_id) return namespaces
def process(self, query, *args, **kwargs): # Returns an iterator of entities that reads in batches. entities = helper.fetch_entities(self._project, self._datastore_namespace, query, self._datastore) return entities
def process(self, query, *args, **kwargs): # Returns an iterator of entities that reads in batches. entities = helper.fetch_entities(self._project, self._datastore_namespace, query, self._datastore) return entities