Advertisement
Guest User

Search.py class

a guest
Jan 30th, 2016
123
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.21 KB | None | 0 0
  1. def search_database(search_query):
  2.     #search_result = es.search(index='posts', q='full name:"Ronaldo"')
  3.     #search_result = es.search(index='players', q=search_query)
  4.    
  5.     query = {
  6.         "query": {
  7.             "match": {
  8.                 "full name": search_query,
  9.             }
  10.         }
  11.     }
  12.  
  13.     search_result = es.search(index=index, body=query)
  14.     return_list = []
  15.  
  16.     if search_result['timed_out'] is True:
  17.         playerDTO = PlayerDTO(123, "Search timed out", " ", 1 ,2)
  18.         return_list.append(playerDTO)
  19.  
  20.     elif len(search_result['hits']['hits']) == 0:
  21.         playerDTO = PlayerDTO(123, "No search results", " ", 1 ,2)
  22.         return_list.append(playerDTO)
  23.     else:
  24.         for player in search_result['hits']['hits']:
  25.             playerDTO = PlayerDTO(player['_source']['player id'],
  26.                                   player['_source']['first name'],
  27.                                   player['_source']['last name'],
  28.                                   player['_source']['team id'],
  29.                                   player['_source']['league id'])
  30.             #playerDTO.team_id = 0
  31.  
  32.             print(playerDTO)
  33.             return_list.append(playerDTO)
  34.  
  35.     return return_list
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement