vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Score based on the robot short name

Browse source
This commit is contained in:
Joseph Schorr 2015-04-20 12:51:47 -04:00
parent d714c00ddb
commit 16e05e83b1
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
endpoints/api/search.py
View file

@ -137,7 +137,7 @@ class FindRepositories(ApiResource):
def search_entity_view(username, entity): def search_entity_view(username, entity, get_short_name=None):
kind = 'user' kind = 'user'
avatar_data = avatar.get_data_for_user(entity) avatar_data = avatar.get_data_for_user(entity)
href = '/user/' + entity.username href = '/user/' + entity.username
@ -156,7 +156,7 @@ def search_entity_view(username, entity):
kind = 'robot' kind = 'robot'
avatar_data = None avatar_data = None
return { data = {
'kind': kind, 'kind': kind,
'avatar': avatar_data, 'avatar': avatar_data,
'name': entity.username, 'name': entity.username,
@ -164,6 +164,11 @@ def search_entity_view(username, entity):
'href': href 'href': href
} }
if get_short_name:
data['short_name'] = get_short_name(entity.username)
return data
def conduct_team_search(username, query, encountered_teams, results): def conduct_team_search(username, query, encountered_teams, results):
""" Finds the matching teams where the user is a member. """ """ Finds the matching teams where the user is a member. """
@ -242,9 +247,12 @@ def conduct_namespace_search(username, query, results):
def conduct_robot_search(username, query, results): def conduct_robot_search(username, query, results):
""" Finds matching robot accounts. """ """ Finds matching robot accounts. """
def get_short_name(name):
return parse_robot_username(name)[1]
matching_robots = model.get_matching_robots(query, username, limit=5) matching_robots = model.get_matching_robots(query, username, limit=5)
for robot in matching_robots: for robot in matching_robots:
results.append(search_entity_view(username, robot)) results.append(search_entity_view(username, robot, get_short_name))
@resource('/v1/find/all') @resource('/v1/find/all')
@ -282,6 +290,7 @@ class ConductSearch(ApiResource):
# Modify the results' scores via how close the query term is to each result's name. # Modify the results' scores via how close the query term is to each result's name.
for result in results: for result in results:
result['score'] = result['score'] * liquidmetal.score(result['name'], query) name = result.get('short_name', result['name'])
result['score'] = result['score'] * liquidmetal.score(name, query)
return {'results': sorted(results, key=itemgetter('score'), reverse=True)} return {'results': sorted(results, key=itemgetter('score'), reverse=True)}