michael_sayapin/holiday-cn
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify code

Browse Source
This commit is contained in:
NateScarlet 2019-03-09 20:29:39 +08:00
parent 3ded7ffe81
commit afcc7c3972
No known key found for this signature in database
GPG Key ID: 5C242793B070309C
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
fetch_holidays.py
View File

@ -94,12 +94,6 @@ class DescriptionParser:
self.year = year self.year = year
self.date_history = list() self.date_history = list()
def memorize_date(self, value: date):
self.date_history.append(value)
def clear_memory(self):
del self.date_history[:]
def parse(self) -> Iterator[dict]: def parse(self) -> Iterator[dict]:
"""Generator for description parsing result. """Generator for description parsing result.
@ -107,7 +101,7 @@ class DescriptionParser:
year (int): Context year year (int): Context year
""" """
self.clear_memory() del self.date_history[:]
for i in re.split('[，。；]', self.description): for i in re.split('[，。；]', self.description):
for j in SentenceParser(self, i).parse(): for j in SentenceParser(self, i).parse():
yield j yield j
@ -167,7 +161,7 @@ class SentenceParser:
for i in method(self, text): for i in method(self, text):
count += 1 count += 1
is_seen = i in self.parent.date_history is_seen = i in self.parent.date_history
self.parent.memorize_date(i) self.parent.date_history.append(i)
if is_seen: if is_seen:
continue continue
yield i yield i