Refactor visitors

This commit is contained in:
Tim Van Baak 2020-04-11 11:45:21 -07:00
parent a9a025094a
commit 751b0cf485
1 changed files with 33 additions and 20 deletions

View File

@ -5,7 +5,39 @@ for verification against constraints.
import re import re
class FeatureCounter(): class RenderableVisitor():
"""Default implementation of the visitor pattern"""
def TextSpan(self, span):
return self
def LineBreak(self, span):
return self
def ParsedArticle(self, span):
span.recurse(self)
return self
def BodyParagraph(self, span):
span.recurse(self)
return self
def SignatureParagraph(self, span):
span.recurse(self)
return self
def BoldSpan(self, span):
span.recurse(self)
return self
def ItalicSpan(self, span):
span.recurse(self)
return self
def CitationSpan(self, span):
span.recurse(self)
return self
class GetCitations(RenderableVisitor):
def __init__(self):
self.citations = []
def CitationSpan(self, span):
self.citations.append(self.cite_target)
return self
class FeatureCounter(RenderableVisitor):
def __init__(self): def __init__(self):
self.word_count = 0 self.word_count = 0
self.citation_count = 0 self.citation_count = 0
@ -15,30 +47,11 @@ class FeatureCounter():
self.word_count += len(re.split('\s+', span.innertext.strip())) self.word_count += len(re.split('\s+', span.innertext.strip()))
return self return self
def LineBreak(self, span):
return self
def ParsedArticle(self, span):
span.recurse(self)
return self
def BodyParagraph(self, span):
span.recurse(self)
return self
def SignatureParagraph(self, span): def SignatureParagraph(self, span):
self.has_signature = True self.has_signature = True
span.recurse(self) span.recurse(self)
return self return self
def BoldSpan(self, span):
span.recurse(self)
return self
def ItalicSpan(self, span):
span.recurse(self)
return self
def CitationSpan(self, span): def CitationSpan(self, span):
self.citation_count += 1 self.citation_count += 1
span.recurse(self) span.recurse(self)