Extend answer to Exercise 1-31

zzzzzwyu · May 17, 2017 · 7110bd9 · 7110bd9
1 parent 113a61b
commit 7110bd9
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/1-31answer.txt b/1-31answer.txt
@@ -1,4 +1,16 @@
 We can use the following 'filter':
 
-removedHash = re.sub(r'#\S+', '', tweet)
-scrubbedTweet = re.sub(r'^RT( @[^\s@]+)+|^\.+', '', removedHash).lstrip()
+def filter(tweet, meta=False):
+	findHash = re.findall(r'#\S+', tweet)
+	elements = {}
+	if findHash and meta:
+		elements['hashtags'] = hashtag.lstrip('@') for hashtag in findHash
+	findRT = re.search(r'^RT(?: @[^\s@]+)+', removedHash)
+	if findRT and meta:
+		users = findRT.group().split()
+		users.remove('RT')
+		elements['RT'] = tuple(users)
+	scrubbedTweet = re.sub(r'(^RT( @[^\s@]+)+|^\.+)|#\S+', '', tweet).lstrip()
+	if meta:
+		return (scrubbedTweet, elements)
+	return scrubbedTweet