Mercurial > hg > smartopen

annotate smartopen/handlers.py @ 11:ba9058605c5a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
add a wiktionary handler
author Jeff Hammel <jhammel@mozilla.com>
date Wed, 22 Dec 2010 17:22:04 -0800
parents a963acb1713d
children bb995bdf82e2
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
1 import address
10
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
2 import string
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
3 import urllib
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
4 import urllib2
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
5
11
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
6 class Location(object):
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
7 """
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
8 generic class for locations
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
9 """
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
10
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
11 def __init__(self, baseurl=""):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
12 self.baseurl = baseurl
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
13
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
14 def url(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
15 return self.baseurl + self.process(query)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
16
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
17 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
18 return query
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
19
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
20 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
21 return True
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
22
11
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
23 def exists(self, URL):
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
24 """does a URL exist?"""
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
25 try:
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
26 urllib.urlopen(URL)
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
27 return True
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
28 except IOError:
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
29 return False
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
30
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
31 class URL(Location):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
32 """a straight URL"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
33
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
34 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
35 if '://' in query:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
36 return query
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
37 return 'http://' + query
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
38
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
39 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
40 """try to open the url"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
41 if ' ' in query or '\n' in query:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
42 return False
11
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
43 return self.exists(self.process(query))
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
44
1
10fc4904c10f now can pass data
k0s <k0scist@gmail.com>
parents: 0
diff changeset
45 class GoogleMaps(Location):
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
46 """try to google-maps the address"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
47
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
48 def __init__(self):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
49 gmapsurl='http://maps.google.com/maps?f=q&hl=en&q='
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
50 Location.__init__(self, gmapsurl)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
51
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
52 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
53 theaddress = address.normalizeaddress(query)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
54 if not theaddress:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
55 return theaddress
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
56 return urllib.quote_plus(theaddress)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
57
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
58 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
59 return bool(self.process(query))
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
60
1
10fc4904c10f now can pass data
k0s <k0scist@gmail.com>
parents: 0
diff changeset
61 class Trac(Location):
2
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
62 def __init__(self, baseurl):
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
63 baseurl = baseurl.strip('/') + '/'
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
64 Location.__init__(self, baseurl)
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
65
2
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
66 def process(self, query):
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
67 if query[0] == 'r':
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
68 if query[1:].isdigit():
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
69 return 'changeset/' + str(query[1:])
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
70 if query[0] == '#':
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
71 if query[1:].isdigit():
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
72 return 'ticket/' + str(query[1:])
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
73
2
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
74 def test(self, query):
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
75 return bool(self.process(query))
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
76
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
77
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
78 class Wikipedia(Location):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
79 """try to open the query in wikipedia"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
80 def __init__(self):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
81 wikiurl = 'http://en.wikipedia.org/wiki/'
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
82 Location.__init__(self, wikiurl)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
83
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
84 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
85 return urllib.quote_plus('_'.join(query.split()))
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
86
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
87 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
88 'test to see if the article exists'
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
89
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
90 # need a phony user agent so wikipedia won't know we're a bot
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
91 headers = {}
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
92 headers['User-Agent'] = 'Mozilla/5.0 (Windows; U; Windows NT 5.1; en-GB; rv:1.8.0.4) Gecko/20060508 Firefox/1.5.0.4'
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
93
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
94 request = urllib2.Request(self.url(query), None, headers)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
95 try:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
96 f = urllib2.urlopen(request).read()
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
97 except urllib2.HTTPError, e:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
98 return False
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
99
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
100 if 'Wikipedia does not have an article with this exact name' in f:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
101 return False
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
102 return True
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
103
11
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
104 class Wiktionary(Location):
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
105 def __init__(self):
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
106 baseurl = 'http://en.wiktionary.org/wiki/'
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
107 Location.__init__(self, baseurl)
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
108 def test(self, query):
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
109 for c in (' ', '\n', '/'):
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
110 if c in query:
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
111 return False
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
112 if self.exists(self.url(query)):
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
113 return True
ba9058605c5a add a wiktionary handler
Jeff Hammel <jhammel@mozilla.com>
parents: 10
diff changeset
114
7
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
115 class Bugzilla(Location):
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
116 def __init__(self):
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
117 baseurl = 'https://bugzilla.mozilla.org/show_bug.cgi?id='
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
118 Location.__init__(self, baseurl)
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
119
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
120 def test(self, query):
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
121 try:
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
122 int(query)
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
123 return True
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
124 except:
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
125 return False
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
126
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
127 class Google(Location):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
128 def __init__(self):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
129 googleurl = 'http://www.google.com/search?hl=en&q='
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
130 Location.__init__(self, googleurl)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
131
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
132 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
133 return urllib.quote_plus(query)
7
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
134
8
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
135 class FedEx(Location):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
136 def __init__(self):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
137 baseurl = 'http://www.fedex.com/Tracking?cntry_code=us&language=english&tracknumber_list='
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
138 Location.__init__(self, baseurl)
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
139
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
140 def process(self, query):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
141 if query.count(' ') == 2:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
142 query = ''.join(query.split(' '))
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
143 return query
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
144
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
145 def test(self, query):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
146 query = self.process(query)
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
147 if len(query) != 12:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
148 return False
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
149 try:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
150 int(query)
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
151 except ValueError:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
152 return False
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
153 return True
10
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
154
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
155 class MercurialRevision(Location):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
156 def __init__(self):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
157 baseurl = 'http://hg.mozilla.org/mozilla-central/rev/'
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
158 Location.__init__(self, baseurl)
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
159
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
160 def test(self, query):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
161 query = set(query)
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
162 if query.issubset(string.digits + 'abcdef'):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
163 return True
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
164 return False