def test_register(self): """ Register/Unregister works """ registry.unregister("stock", StockAutocompleteProvider) providers = registry.get_all_by_autocompleter("stock") self.assertEqual(len(providers), 0) # Have to leave things the way models.py init-ed them for other tests! registry.register("stock", StockAutocompleteProvider) providers = registry.get_all_by_autocompleter("stock") self.assertEqual(len(providers), 1)
return { 'EV': 'Enterprise Value', } @classmethod def get_one_way_phrase_aliases(cls): return { 'Revenue': 'Turnover', } @classmethod def get_iterator(cls): return calc_info.calc_dicts registry.register("faceted_stock", FacetedStockAutocompleteProvider) registry.register("stock", StockAutocompleteProvider) registry.register("mixed", StockAutocompleteProvider) registry.register("mixed", IndicatorAutocompleteProvider) registry.register("mixed", CalcAutocompleteProvider) registry.register("ind_stock", IndicatorAutocompleteProvider) registry.register("ind_stock", StockAutocompleteProvider) registry.register("mixed", StockAutocompleteProvider) registry.register("indicator", IndicatorAutocompleteProvider) registry.register("indicator_aliased", IndicatorAliasedAutocompleteProvider) registry.register("indicator_selective", IndicatorSelectiveAutocompleteProvider) registry.register("metric", CalcAutocompleteProvider) registry.register("metric_aliased", CalcAliasedAutocompleteProvider) registry.register("facet_stock_no_facet_ind", FacetedStockAutocompleteProvider) registry.register("facet_stock_no_facet_ind", IndicatorAutocompleteProvider)
return { 'EV': 'Enterprise Value', } @classmethod def get_one_way_phrase_aliases(cls): return { 'Revenue': 'Turnover', } @classmethod def get_iterator(cls): return calc_info.calc_dicts registry.register("faceted_stock", FacetedStockAutocompleteProvider) registry.register("stock", StockAutocompleteProvider) registry.register("mixed", StockAutocompleteProvider) registry.register("mixed", IndicatorAutocompleteProvider) registry.register("mixed", CalcAutocompleteProvider) registry.register("ind_stock", IndicatorAutocompleteProvider) registry.register("ind_stock", StockAutocompleteProvider) registry.register("mixed", StockAutocompleteProvider) registry.register("indicator", IndicatorAutocompleteProvider) registry.register("indicator_aliased", IndicatorAliasedAutocompleteProvider) registry.register("indicator_selective", IndicatorSelectiveAutocompleteProvider) registry.register("metric", CalcAutocompleteProvider) registry.register("metric_aliased", CalcAliasedAutocompleteProvider) registry.register("facet_stock_no_facet_ind", FacetedStockAutocompleteProvider)
def get_score(self): """ Larger companies should end up higher in search results. """ return self.obj.market_cap def get_data(self): return { 'type': 'stock', 'id': self.obj.id, 'score': self.get_score(), 'display_name': u'%s (%s)' % (self.obj.name, self.obj.symbol), 'search_name': self.obj.symbol, } registry.register("stock", StockAutocompleteProvider) registry.register("mixed", StockAutocompleteProvider) class Indicator(models.Model): name = models.CharField(max_length=200, unique=True) internal_name = models.CharField(max_length=200, unique=True) score = models.FloatField(null=True, blank=True) class IndicatorAutocompleteProvider(AutocompleterProvider): model = Indicator provider_name = "ind" def get_obj_id(self):