diff --git a/src/twi_parser.py b/src/twi_parser.py index 09ef416..6fb0060 100644 --- a/src/twi_parser.py +++ b/src/twi_parser.py @@ -171,6 +171,15 @@ def parse_tweet(tweet): data["card"], _photo = parse_card(tweet["card"]) if _photo: data["media"].append(_photo) + if "note_tweet" in tweet: + note = tweet["note_tweet"]["note_tweet_results"]["result"] + data["full_text"] = note["text"] + data["entities"] = [] + for e in ["user_mentions", "hashtags", "urls"]: + for m in note["entity_set"].get(e, []): + data["entities"].append(parse_entities(m)) + data["entities"].sort(key=lambda x: x["indices"][0]) + return data