week6/week6.py

   1 #!/usr/bin/env python
   2
   3 import urllib
   4
   5 from enthought.traits.api import HasTraits,Str,Float,Button
   6 from enthought.traits.ui.api import View,Item,ButtonEditor
   7 from yahoo.search.image import ImageSearch
   8
   9 import pylab as pl
  10
  11 def yahoosearch(query):
  12         ''' Yahoo searches for an image and returns a URL, or an empty string if something goes wrong '''
  13         try:
  14                 search = ImageSearch(app_id="junk",query=query)
  15                 search.results = 1
  16                 for res in search.parse_results():
  17                         return res.Url
  18         except:
  19                 pass
  20
  21         return ""
  22
  23 def fetchlink(url):
  24         ''' Fetch file from link and return path to local file '''
  25         return urllib.urlretrieve(url)[0]
  26
  27 class ImgSearch(HasTraits):
  28         query = Str("Enter search")
  29         run_query = Button()
  30         url = Str
  31         localfile = Str
  32         def _run_query_fired(self):
  33                 self.url = yahoosearch(self.query)
  34                 self.localfile = fetchlink(self.url)
  35
  36                 pl.open(self.localfile)
  37
  38         view = View('query','url','localfile',Item('run_query',show_label=False))
  39
  40 c = ImgSearch()
  41 c.configure_traits()
  42