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)
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):
