Spaces:
Runtime error
Runtime error
wira.indra
commited on
Commit
·
d49f00b
1
Parent(s):
faf61e8
add twitter feature
Browse files- twitter_scraper.py +47 -0
twitter_scraper.py
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import snscrape.modules.twitter as sntwitter
|
2 |
+
import datetime as dt
|
3 |
+
import pandas as pd
|
4 |
+
import sys
|
5 |
+
import os
|
6 |
+
import re
|
7 |
+
import tqdm
|
8 |
+
|
9 |
+
|
10 |
+
def scrape_tweets(query, max_tweets=10, output_path="./scraper/output/" ):
|
11 |
+
tweets_list = []
|
12 |
+
|
13 |
+
for i,tweet in tqdm(enumerate(sntwitter.TwitterSearchScraper(query).get_items())):
|
14 |
+
if max_tweets != -1 and i >= int(max_tweets):
|
15 |
+
break
|
16 |
+
tweets_list.append([tweet.date, tweet.id, tweet.content, tweet.user.username, tweet.likeCount, tweet.retweetCount, tweet.replyCount, tweet.quoteCount, tweet.url, tweet.lang])
|
17 |
+
|
18 |
+
df = pd.DataFrame(tweets_list, columns=['Datetime', 'Tweet Id', 'Text', 'Username', 'Likes', 'Retweets', 'Replies', 'Quotes', 'URL', 'Language'])
|
19 |
+
df = df[df["Language"] == "in"]
|
20 |
+
return df
|
21 |
+
|
22 |
+
def remove_unnecessary_char(text):
|
23 |
+
text = re.sub("\[USERNAME\]", " ", text)
|
24 |
+
text = re.sub("\[URL\]", " ", text)
|
25 |
+
text = re.sub("\[SENSITIVE-NO\]", " ", text)
|
26 |
+
text = re.sub(' +', ' ', text)
|
27 |
+
return text
|
28 |
+
|
29 |
+
def preprocess_tweet(text):
|
30 |
+
text = re.sub('\n',' ',text)
|
31 |
+
text = re.sub('^(\@\w+ ?)+',' ',text)
|
32 |
+
text = re.sub(r'\@\w+',' ',text)
|
33 |
+
text = re.sub('((www\.[^\s]+)|(https?://[^\s]+)|(http?://[^\s]+))',' ',text)
|
34 |
+
text = re.sub('/', ' ', text)
|
35 |
+
text = re.sub(' +', ' ', text)
|
36 |
+
return text
|
37 |
+
|
38 |
+
def remove_nonaplhanumeric(text):
|
39 |
+
text = re.sub('[^0-9a-zA-Z]+', ' ', text)
|
40 |
+
return text
|
41 |
+
|
42 |
+
def preprocess_text(text):
|
43 |
+
text = preprocess_tweet(text)
|
44 |
+
text = remove_unnecessary_char(text)
|
45 |
+
text = remove_nonaplhanumeric(text)
|
46 |
+
text = text.lower()
|
47 |
+
return text
|