class StreetMatcher(BaseMatcher):
def __init__(self, patterns_str, lookingEntityName, replaceMatches = False, replacementString = '', matchLimit = 2):
BaseMatcher.__init__(self, patterns_str, lookingEntityName, replaceMatches, replacementString)
self.foundEntities = OrderedSet()
self.limit = matchLimit
@baseMatcher_cut_log
def cut(self, match):
for i in range(0,len(match.groups())):
street_name = self.findStreet(match.group(i+1))
if street_name:
self.foundEntities.append(street_name)
#if (len(self.foundEntities) == self.limit):
# break
return self.replacementString
@streetMatcher__matchWithoutCut
def _matchWithoutCut(self, input, pattern):
for pretenders in pattern.findall(input):
if pattern.groups == 1:
pretenders = [pretenders]
for pretender in pretenders:
street_name = self.findStreet(pretender)
if street_name:
self.foundEntities.add(street_name)
return input
@streetMatcher_find_street
def findStreet(self, name):
streets = Street.objects.filter(name__contains=name)
if streets:
return streets[0].name
else:
street_freaks = StreetFreak.objects.filter(name__contains=name)
if street_freaks:
return street_freaks[0].street.name
@baseMatcher_match_log
def match(self, input):
self.foundEntities = OrderedSet()
if self.replaceMatches:
in_match = self._matchWithCut
else:
in_match = self._matchWithoutCut
for pattern in self.patterns:
input = in_match(input, pattern)
if (len(self.foundEntities) >= self.limit):
break
return (self.foundEntities, input)
def match(self, input):
self.foundEntities = OrderedSet()
if self.replaceMatches:
in_match = self._matchWithCut
else:
in_match = self._matchWithoutCut
for pattern in self.patterns:
input = in_match(input, pattern)
if (len(self.foundEntities) >= self.limit):
break
return (self.foundEntities, input)
def __init__(self, patterns_str, lookingEntityName, replaceMatches = False, replacementString = '', matchLimit = 2):
BaseMatcher.__init__(self, patterns_str, lookingEntityName, replaceMatches, replacementString)
self.foundEntities = OrderedSet()
self.limit = matchLimit