Mercurial > hg > smartopen

annotate smartopen/handlers.py @ 10:a963acb1713d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
add a new handler: MercurialRevision
author Jeff Hammel <jhammel@mozilla.com>
date Tue, 21 Sep 2010 12:01:34 -0700
parents 44950f376e98
children ba9058605c5a
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
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
6 class Location:
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
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
23
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
24 class URL(Location):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
25 """a straight URL"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
26
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
27 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
28 if '://' in query:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
29 return query
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
30 return 'http://' + query
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
31
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
32 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
33 """try to open the url"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
34
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
35 if ' ' in query or '\n' in query:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
36 return False
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
37
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
38 try:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
39 site = urllib.urlopen(self.process(query))
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
40 except IOError:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
41 return False
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
42 return True
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
43
1
10fc4904c10f now can pass data
k0s <k0scist@gmail.com>
parents: 0
diff changeset
44 class GoogleMaps(Location):
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
45 """try to google-maps the address"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
46
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
47 def __init__(self):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
48 gmapsurl='http://maps.google.com/maps?f=q&hl=en&q='
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
49 Location.__init__(self, gmapsurl)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
50
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
51 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
52 theaddress = address.normalizeaddress(query)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
53 if not theaddress:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
54 return theaddress
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
55 return urllib.quote_plus(theaddress)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
56
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
57 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
58 return bool(self.process(query))
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
59
1
10fc4904c10f now can pass data
k0s <k0scist@gmail.com>
parents: 0
diff changeset
60 class Trac(Location):
2
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
61 def __init__(self, baseurl):
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
62 baseurl = baseurl.strip('/') + '/'
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
63 Location.__init__(self, baseurl)
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
64
2
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
65 def process(self, query):
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
66 if query[0] == 'r':
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
67 if query[1:].isdigit():
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
68 return 'changeset/' + str(query[1:])
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
69 if query[0] == '#':
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
70 if query[1:].isdigit():
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
71 return 'ticket/' + str(query[1:])
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
72
2
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
73 def test(self, query):
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
74 return bool(self.process(query))
f084e152dd47 add Trac handler
k0s <k0scist@gmail.com>
parents: 1
diff changeset
75
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
76
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
77 class Wikipedia(Location):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
78 """try to open the query in wikipedia"""
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
79 def __init__(self):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
80 wikiurl = 'http://en.wikipedia.org/wiki/'
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
81 Location.__init__(self, wikiurl)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
82
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
83 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
84 return urllib.quote_plus('_'.join(query.split()))
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
85
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
86 def test(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
87 'test to see if the article exists'
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
88
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
89 # 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
90 headers = {}
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
91 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
92
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
93 request = urllib2.Request(self.url(query), None, headers)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
94 try:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
95 f = urllib2.urlopen(request).read()
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
96 except urllib2.HTTPError, e:
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
97 return False
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
98
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
99 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
100 return False
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
101 return True
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
102
7
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
103 class Bugzilla(Location):
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
104 def __init__(self):
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
105 baseurl = 'https://bugzilla.mozilla.org/show_bug.cgi?id='
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
106 Location.__init__(self, baseurl)
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
107
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
108 def test(self, query):
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
109 try:
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
110 int(query)
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
111 return True
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
112 except:
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
113 return False
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
114
0
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
115 class Google(Location):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
116 def __init__(self):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
117 googleurl = 'http://www.google.com/search?hl=en&q='
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
118 Location.__init__(self, googleurl)
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
119
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
120 def process(self, query):
d6fa501af82f initial commit of smartopen
k0s <k0scist@gmail.com>
parents:
diff changeset
121 return urllib.quote_plus(query)
7
d5343f681ef9 add bugzilla handler
Jeff Hammel <k0scist@gmail.com>
parents: 2
diff changeset
122
8
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
123 class FedEx(Location):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
124 def __init__(self):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
125 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
126 Location.__init__(self, baseurl)
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
127
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
128 def process(self, query):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
129 if query.count(' ') == 2:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
130 query = ''.join(query.split(' '))
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
131 return query
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
132
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
133 def test(self, query):
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
134 query = self.process(query)
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
135 if len(query) != 12:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
136 return False
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
137 try:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
138 int(query)
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
139 except ValueError:
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
140 return False
44950f376e98 add FedEx tracking number handler
Jeff Hammel <jhammel@mozilla.com>
parents: 7
diff changeset
141 return True
10
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
142
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
143 class MercurialRevision(Location):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
144 def __init__(self):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
145 baseurl = 'http://hg.mozilla.org/mozilla-central/rev/'
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
146 Location.__init__(self, baseurl)
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
147
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
148 def test(self, query):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
149 query = set(query)
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
150 if query.issubset(string.digits + 'abcdef'):
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
151 return True
a963acb1713d add a new handler: MercurialRevision
Jeff Hammel <jhammel@mozilla.com>
parents: 8
diff changeset
152 return False