class LinksParser(HTMLParser.HTMLParser):
def __init__(self):
HTMLParser.HTMLParser.__init__(self)
self.recording = 0
self.data = []
def handle_starttag(self, tag, attributes):
if tag != 'div':
return
if self.recording:
self.recording += 1
return
for name, value in attributes:
if name == 'id' and value == 'remository':
break
else:
return
self.recording = 1
def handle_endtag(self, tag):
if tag == 'div' and self.recording:
self.recording -= 1
def handle_data(self, data):
if self.recording:
self.data.append(data)
self.recording
подсчитывает количество вложенных тегов div
, начиная с «запускающего». Когда мы находимся в поддереве с корневым тегом запуска, мы накапливаем данные в self.data
.
Данные в конце синтаксического анализа остаются в self.data
(список строк, возможно, пустой, если триггерный тег не встречен). Ваш код из-за пределов класса может получить доступ к списку непосредственно из экземпляра в конце разбора, или вы можете добавить соответствующие методы доступа для этой цели, в зависимости от того, что именно является вашей целью.
Класс можно легко сделать более общим, используя вместо константных литеральных строк, показанных в приведенном выше коде, 'div'
, 'id'
и 'remository'
, атрибуты экземпляра self.tag
, self.attname
и self.attvalue
, установленный __init__
из переданных ему аргументов - я избежал этого дешевого этапа обобщения в приведенном выше коде, чтобы не затенять основные точки (отслеживать количество вложенных тегов и накапливать данные в списке, когда состояние записи активно).