def query(self, *args, **kw): if self.cacheTypeDescriptions: typeDescs = self.typeDescs else: typeDescs = {} if len(args) == 1: # full query string queryString = args[0] elif len(args) == 2: # BBB: fields, sObjectType queryString = 'select %s from %s' % (args[0], args[1]) if 'conditionalExpression' in kw: # BBB: fields, sObjectType, conditionExpression as kwarg queryString += ' where %s' % (kw['conditionalExpression']) elif len(args) == 3: # BBB: fields, sObjectType, conditionExpression as positional arg whereClause = args[2] and (' where %s' % args[2]) or '' queryString = 'select %s from %s%s' % (args[0], args[1], whereClause) else: raise RuntimeError, "Wrong number of arguments to query method." res = BaseClient.query(self, queryString) # calculate the union of the sets of record types from each record types = reduce(lambda a,b: a|b, [getRecordTypes(r) for r in res[_tPartnerNS.records:]], set()) new_types = types - set(typeDescs.keys()) if new_types: typeDescs.update(self.queryTypesDescriptions(new_types)) data = QueryRecordSet(records=[self._extractRecord(r, typeDescs) for r in res[_tPartnerNS.records:]], done=_bool(res[_tPartnerNS.done]), size=int(str(res[_tPartnerNS.size])), queryLocator = str(res[_tPartnerNS.queryLocator])) return data
def query(self, *args, **kw): if self.cacheTypeDescriptions: typeDescs = self.typeDescs else: typeDescs = {} if len(args) == 1: # full query string queryString = args[0] elif len(args) == 2: # BBB: fields, sObjectType queryString = 'select %s from %s' % (args[0], args[1]) if 'conditionalExpression' in kw: # BBB: fields, sObjectType, # conditionExpression as kwarg queryString += ' where %s' % (kw['conditionalExpression']) elif len(args) == 3: # BBB: fields, sObjectType, conditionExpression # as positional arg whereClause = args[2] and (' where %s' % args[2]) or '' queryString = 'select %s from %s%s' % ( args[0], args[1], whereClause ) else: raise RuntimeError("Wrong number of arguments to query method.") res = BaseClient.query(self, queryString) # calculate the union of the sets of record types from each record types = reduce(lambda a, b: a | b, [getRecordTypes(r) for r in res[_tPartnerNS.records,]], set()) new_types = types - set(typeDescs.keys()) if new_types: typeDescs.update(self.queryTypesDescriptions(new_types)) data = QueryRecordSet( records=[self._extractRecord(r, typeDescs) for r in res[_tPartnerNS.records,]], done=_bool(res[_tPartnerNS.done]), size=int(str(res[_tPartnerNS.size])), queryLocator=str(res[_tPartnerNS.queryLocator]) ) return data