I have written a simple script that searches twitter for keywords and saves them to a csv file if they contain those words. It can be found on my github here.
How can I improve this code to generally be more efficient and be up to coding standards ?
"""
Script that goes through english tweets that are filtered by security words and posted in the last one hour and stores the polarity, id, date time, query, username and text into a csv file.
"""
import tweepy
import datetime, time, csv, codecs
from textblob import TextBlob
import cleanit
##setting authorization stuff for twitter##
consumer_key = "xxx"
consumer_secret = "xxx"
access_token = "xxx"
access_token_secret = "xxx"
auth = tweepy.OAuthHandler(consumer_key, consumer_secret)
auth.set_access_token(access_token, access_token_secret)
api = tweepy.API(auth)
##initializing lists##
big_list = []
text_list = []
id_list = []
name_list = []
created_list = []
query_list = []
polarityy = []
t = 0
#use words in this list as search terms for tweepy.cursor function
security_words = ['phishing','dos','botnet','xss','smb','wannacry','heartbleed','ransomware','trojan','spyware','exploit','virus','malware','mitm']
# if word in security words list and double_meaning_words list if text also contains word from gen words list, if it does store if not discard
double_meaning_words = ['petya','smb','dos','infosec','hacker','backdoor']
gen_words = ["attack","security","hit","detected","protected","injection","data","exploit", "router", 'ransomware', 'phishing', 'wannacry', 'security']
def storing_data(stat):
##store id,username,datetime,text and polarity for filtered tweets in csv##
text_list.append(str(cleanit.tweet_cleaner_updated(status.text)).encode("utf-8"))
id_list.append(str(status.id)) # append id number to list
name_list.append(str(status.user.screen_name)) # append user name to list
created_list.append((status.created_at).strftime('%c')) # append date time to list
analysis = TextBlob(status.text)
analysis = analysis.sentiment.polarity # use textblob on text to get sentiment score of text
if analysis >= -1 and analysis <= 0: # append sentiment score to list
polarityy.append("4")
else:
polarityy.append("0")
def rejects(stat):
##store tweets which do not pass filters into csv##
with open('rejects.csv', "a", newline='', encoding='utf-8') as rejectfile:
logger = csv.writer(rejectfile)
logger.writerow([status.text])
while True:
print ('running', datetime.datetime.now())
with open('sec_tweet_dataset_5.csv', "a", newline='', encoding='utf-8') as logfile:
logger = csv.writer(logfile)
for i in security_words:
alex = []
for status in tweepy.Cursor(api.search, i,lang="en").items(40): #search twitter for word in security word list in english
if (status.retweeted == False) or ('RT @' not in status.text): #is tweet is retweeted dont store it
if i in double_meaning_words and i in status.text: #if search term being used from security words list also in double meaning words check if it also contains word -
for words in gen_words: # - from gen_words list. If it does continue to storing if not dont store.
if words in status.text:
storing_data(status)
break
else:
rejects(status)
else:
storing_data(status)
rejects(status)
while t < len(polarityy):
alex = ([polarityy[t],id_list[t],created_list[t],name_list[t],text_list[int(t)]])
t += 1
logger.writerow(alex)
time.sleep(1800)