lookup_with_google_and_link.py

import sublime, sublime_plugin
import os, sys, re

# import modules in current directory
dist_dir = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, dist_dir)

# Python 2/3 compatible
import chardet, pystache

try:
	# Python 3 (ST3)
	from urllib.request import Request, urlopen
	from urllib.error import URLError
	from urllib.parse import urlencode
except ImportError:
	# Python 2 (ST2)
	from urllib2 import Request, URLError, urlopen
	from urllib import urlencode

def preemptive_imports():
	""" needed to ensure ability to import these classes later within functions, due to the way ST2 loads plug-in modules """
	from chardet import universaldetector
preemptive_imports()

class LookupWithGoogleAndLinkCommand(sublime_plugin.TextCommand):

	def get_link_with_title(self, phrase):
		try:
			url = "http://www.google.com/search?%s&btnI=I'm+Feeling+Lucky" % urlencode({'q': phrase})
			req = Request(url, headers={'User-Agent' : "Sublime Text 2 Hyperlink Helper"})
			f = urlopen(req)
			url = f.geturl()
			content = f.read()
			decoded_content = content.decode(chardet.detect(content)['encoding'])
			title = re.search(r"<title>([^<>]*)</title>", decoded_content, re.I).group(1)
			title = title.strip()
			return url, title, phrase
		except URLError as e:
			sublime.error_message("Error fetching Google search result: %s" % str(e))
			return None

	def run(self, edit):
		nonempty_sels = []

		# set aside nonempty selections
		for s in reversed(self.view.sel()):
			if not s.empty():
				nonempty_sels.append(s)
				self.view.sel().subtract(s)

		# expand remaining (empty) selections to words
		self.view.run_command("expand_selection", {"to": "word"})

		# add nonempty selections back in
		for s in nonempty_sels:
			self.view.sel().add(s)
		
		# apply the links
		for s in reversed(self.view.sel()):
			if not s.empty():
				txt = self.view.substr(s)
				link = self.get_link_with_title(txt)
				if not link:
					continue
				self.view.replace(edit, s, pystache.render(self.view.settings().get('hyperlink_helper_link_format'), {'url': link[0], 'title?': {'title': link[1]}, 'input': link[2]}))