如何使此代碼通用

Question

我有一個用例告訴，當一個標籤是link，它的屬性是rel=dns-prefetch和prefetch然後只是說，預先解析dns啓用。

我做了一個標誌爲pre_resolve_dns_enabled並將其設置爲true，如下所示。

class Extractor(HTMLParser): 

    def __init__(self): 
     HTMLParser.__init__(self) 
     self.pre_resolve_dns_enabled = False 

    def feed(self, data): 
     HTMLParser.feed(self,data) 

    def handle_starttag(self, tag, attrs): 
     if tag == 'link' and ('rel', 'dns-prefetch') in attrs: 
      self.pre_resolve_dns_enabled = True 
     if tag == 'link' and ('rel', 'prefetch') in attrs: 
      self.prefetch_enabled = True 

而且我寫了兩種方法來返回狀態。

def is_pre_resolve_dns_enabled(self): 
    return self.pre_resolve_dns_enabled 

def is_prefetch_enabled(self): 
    return self.prefetch_enabled 

有沒有一種辦法可以有效地使handle_starttag方法通用的，所以，我沒有硬編碼的標籤和查詢這樣的attars和擺脫is_pre_resolve_dns_enabled和is_prefetch_enabled。

Answer 1

如果你想和BS4要做到這一點，您可以使用此：

from bs4 import BeautifulSoup 
soup = BeautifulSoup(html) # Or some xml-like structure 
# This will select all link tags which have a rel attribute with value dns-prefetch. 
if len(soup.select('link[rel=dns-prefetch]')) > 0: 
    self.pre_resolve_dns_enabled = True 
if len(soup.select('link[rel=prefetch]')) > 0: 
    self.prefetch_enabled = True