LandmarkMatcher.py

   1 #
   2 # vi: set softtabstop=4 shiftwidth=4 tabstop=8 expandtab:
   3
   4 """Matches partial landmark names to full landmark names using a
   5    prioritized list.
   6
   7    It turns out that you can send a partial name to the travel planner
   8    web app and it'll figure out what you mean, but there's no
   9    prioritization of partial matches: BAYSHORE maps to
  10    BAYSHORE CATHOLIC SCHOOL before it maps to BAYSHORE SHOPPING CENTRE.
  11
  12    We do our own prioritization by using landmarkLeech.py to grab
  13    an offline copy of the landmark list in landmarks.txt, and
  14    listing the categories (transit station, shopping centre, etc.) in
  15    priority order in landmarkCategoryPriority.txt."""
  16
  17 from PlannerExceptions import *
  18
  19 class LandmarkMatcher:
  20     def __init__(self, listfile="landmarks.txt",
  21                  priofile="landmarkCategoryPriority.txt"):
  22
  23         self.all = set()
  24
  25         # List of categories; each one contains a list of landmarks
  26         self.categories = []
  27
  28         # Map of category to landmark list.
  29         self.categoryMap = {}
  30
  31         self._loadLandmarks(listfile)
  32         self._loadCategoryPrio(priofile)
  33
  34     def _loadLandmarks(self, listfile):
  35         listfp = open(listfile, "r")
  36         for line in listfp:
  37             parts = line.rstrip().split("|")
  38             landmark = parts[1].upper()
  39             self.all.add(landmark)
  40
  41             if not parts[0] in self.categoryMap:
  42                 self.categoryMap[parts[0]] = []
  43             self.categoryMap[parts[0]].append(landmark)
  44
  45     def _loadCategoryPrio(self, priofile):
  46         priofp = open(priofile, "r")
  47         for line in priofp:
  48             self.categories.append(self.categoryMap[line.rstrip()])
  49
  50     def match(self, str):
  51         str = str.upper()
  52         if str in self.all:
  53             return str
  54
  55         for catlist in self.categories:
  56             for landmark in catlist:
  57                 if landmark.startswith(str):
  58                     return landmark
  59
  60         raise InvalidLandmarkException("Couldn't find landmark '%s' "
  61                                        "in local database." % str)