Fix nil tweet time crash

HamzaAlwan · Oct 19, 2019 · d5061bd · d5061bd
1 parent e73963f
commit d5061bd
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/parserutils.nim b/src/parserutils.nim
@@ -80,15 +80,12 @@ proc getTweetText*(tweet: XmlNode): string =
     link = text.selectAttr("a.twitter-timeline-link.u-hidden", "data-expanded-url")
   parseText(text, if quote != nil: link else: "")
 
-proc getTime(tweet: XmlNode): XmlNode =
-  tweet.select(".js-short-timestamp")
-
 proc getTimestamp*(tweet: XmlNode): Time =
-  let time = getTime(tweet).attr("data-time")
+  let time = tweet.selectAttr(".js-short-timestamp", "data-time")
   fromUnix(if time.len > 0: parseInt(time) else: 0)
 
 proc getShortTime*(tweet: XmlNode): string =
-  getTime(tweet).innerText()
+  tweet.selectText(".js-short-timestamp")
 
 proc getDate*(node: XmlNode; selector: string): Time =
   let date = node.select(selector)