def get_records(request, max_records, action_verb='export'): ids, excl_ids = get_ids(request) sess_query = request.session.get('query') filters = [BaseCompany.id.in_(ids)] if ids else where_from_query({'sessquery': sess_query}) if excl_ids: filters.append(BaseCompany.id.notin_(excl_ids)) if not request.user.is_admin: filters.append(BaseCompany.activated == True) if 'alchemist_startup' in request.referer: filters.append(BaseCompany.is_alchemist == True) filters.append(BaseCompany.companytype == 'startup') elif 'companies' not in request.referer: company_typestr = get_company_typestr(urlparse(request.referer).path[1:]) filters.append(BaseCompany.companytype == company_typestr) if not (ids or (request.session.get('select_all_comp') and sess_query is not None)): request.session.flash('We didn\'t find anything to %s' % action_verb, 'error') return False if len(ids) > max_records: request.session.flash('You cannot %s more than %s companies' % (action_verb, max_records), 'error') return False records = DBSession.query(BaseCompany).filter(*filters).limit(max_records + 1).all() if len(records) > max_records: request.session.flash('You cannot %s more than %s companies' % (action_verb, max_records), 'error') return False return records
def find_company_suggestions(q, limit, request): show_type = should_show_types(request) where = [BaseCompany.name.ilike('%' + q + '%')] if not request.user.is_admin: where.append((BaseCompany.activated == True)) if show_type: from alchemist.companies.grid import get_company_typestr where.append(BaseCompany.companytype == get_company_typestr(show_type)) if show_type and 'alchemist' in show_type: where.append(BaseCompany.is_alchemist == True) query = DBSession.query(BaseCompany.id, BaseCompany.name, BaseCompany.companytype).filter(*where).limit( limit) return [transform(row, 'companies', True, row[2]) for row in query.all()]