searchcode_code.py 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. ## Searchcode (It)
  2. #
  3. # @website https://searchcode.com/
  4. # @provide-api yes (https://searchcode.com/api/)
  5. #
  6. # @using-api yes
  7. # @results JSON
  8. # @stable yes
  9. # @parse url, title, content
  10. from urllib import urlencode
  11. from json import loads
  12. import cgi
  13. import re
  14. # engine dependent config
  15. categories = ['it']
  16. paging = True
  17. # search-url
  18. url = 'https://searchcode.com/'
  19. search_url = url+'api/codesearch_I/?{query}&p={pageno}'
  20. # do search-request
  21. def request(query, params):
  22. params['url'] = search_url.format(query=urlencode({'q': query}),
  23. pageno=params['pageno']-1)
  24. return params
  25. # get response from search-request
  26. def response(resp):
  27. results = []
  28. search_results = loads(resp.text)
  29. # parse results
  30. for result in search_results['results']:
  31. href = result['url']
  32. title = "" + result['name'] + " - " + result['filename']
  33. content = result['repo'] + "<br />"
  34. lines = dict()
  35. for line, code in result['lines'].items():
  36. lines[int(line)] = code
  37. content = content + '<pre class="code-formatter"><table class="code">'
  38. for line, code in sorted(lines.items()):
  39. content = content + '<tr><td class="line-number" style="padding-right:5px;">'
  40. content = content + str(line) + '</td><td class="code-snippet">'
  41. # Replace every two spaces with ' &nbps;' to keep formatting while allowing the browser to break the line if necessary
  42. content = content + cgi.escape(code).replace('\t', ' ').replace(' ', '&nbsp; ').replace(' ', ' &nbsp;')
  43. content = content + "</td></tr>"
  44. content = content + "</table></pre>"
  45. # append result
  46. results.append({'url': href,
  47. 'title': title,
  48. 'content': content})
  49. # return results
  50. return results