Zed 7 лет назад
Родитель
Сommit
d5214e2f77
1 измененных файлов с 9 добавлено и 2 удалено
  1. 9 2
      src/parserutils.nim

+ 9 - 2
src/parserutils.nim

@@ -19,11 +19,18 @@ proc selectText*(node: XmlNode; selector: string): string =
   let res = node.querySelector(selector)
   let res = node.querySelector(selector)
   result = if res == nil: "" else: res.innerText()
   result = if res == nil: "" else: res.innerText()
 
 
+proc getHeader(profile: XmlNode): XmlNode =
+  result = profile.querySelector(".permalink-header")
+  if result.isNil:
+    result = profile.querySelector(".stream-item-header")
+  if result.isNil:
+    result = profile.querySelector(".ProfileCard-userFields")
+
 proc isVerified*(profile: XmlNode): bool =
 proc isVerified*(profile: XmlNode): bool =
-  profile.selectText(".Icon.Icon--verified").len > 0
+  getHeader(profile).selectText(".Icon.Icon--verified").len > 0
 
 
 proc isProtected*(profile: XmlNode): bool =
 proc isProtected*(profile: XmlNode): bool =
-  profile.selectText(".Icon.Icon--protected").len > 0
+  getHeader(profile).selectText(".Icon.Icon--protected").len > 0
 
 
 proc getName*(profile: XmlNode; selector: string): string =
 proc getName*(profile: XmlNode; selector: string): string =
   profile.selectText(selector).strip()
   profile.selectText(selector).strip()