x/stackalytics
Code Issues Proposed changes

Merge "Avoid dups in company drop-down"

Browse Source
This commit is contained in:
Jenkins
2014-08-19 10:04:27 +00:00
committed by Gerrit Code Review
parent f52f7e6855 e96773c3ef
commit bf6d4a88ef
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
stackalytics/dashboard/memory_storage.py
View File

@@ -145,9 +145,9 @@ class CachedMemoryStorage(MemoryStorage):
return set()
def get_index_keys_by_record_ids(self, index_name, record_ids):
return set([key
return set(key
for key, value in six.iteritems(self.indexes[index_name])
if value & record_ids])
if value & record_ids)
def get_record_ids(self):
return self.records.keys()

14
stackalytics/dashboard/web.py
View File

@@ -278,16 +278,16 @@ def get_contribution_json(records, **kwargs):
@decorators.record_filter(ignore=['company'])
def get_companies_json(record_ids, **kwargs):
memory_storage = vault.get_memory_storage()
companies = memory_storage.get_index_keys_by_record_ids(
'company_name', record_ids)
if kwargs['_params']['company']:
companies.add(kwargs['_params']['company'][0])
companies = set(memory_storage.get_original_company_name(company)
for company in memory_storage.get_index_keys_by_record_ids(
'company_name', record_ids))
result = [memory_storage.get_original_company_name(company)
for company in companies]
if kwargs['_params']['company']:
companies.add(memory_storage.get_original_company_name(
kwargs['_params']['company'][0]))
return [{'id': utils.safe_encode(c.lower()), 'text': c}
for c in sorted(result)]
for c in sorted(companies)]
@app.route('/api/1.0/modules')