import io
from PIL import Image
import pytesseract
import requests
from tweepy.streaming import StreamListener
from tweepy import OAuthHandler
from tweepy import Stream
import json
with open("config.json") as f:
config = json.load(f)
FOLLOW_IDS = config["TWITTER_FOLLOW_IDS"]
with open("secrets.json") as f:
keys = json.load(f)
CONSUMER_KEY = keys["TWITTER_CONSUMER_KEY"]
CONSUMER_SECRET = keys["TWITTER_CONSUMER_SECRET"]
ACCESS_KEY = keys["TWITTER_ACCESS_KEY"]
ACCESS_SECRET = keys["TWITTER_ACCESS_SECRET"]
class TwitterListener(StreamListener):
def __init__(self, callback):
self.callback = callback
def on_data(self, data):
tweet_json = json.loads(data)
try:
if tweet_json["user"]["id_str"] in FOLLOW_IDS:
print(tweet_json["text"])
self.callback(tweet_json)
except:
pass
class Twitter:
def __init__(self, tweet_callback=lambda x, y, z: x):
self.tweet_callback = tweet_callback
self.listener = TwitterListener(self.handle_tweet)
self.auth = OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
self.auth.set_access_token(ACCESS_KEY, ACCESS_SECRET)
self.stream = Stream(self.auth, self.listener)
self.stream.filter(follow=FOLLOW_IDS)
def handle_tweet(self, tweet_json):
screen_name = tweet_json["user"]["screen_name"]
id = tweet_json["id_str"]
text = tweet_json["text"].replace("\\", "")
# Get media if present
try:
urls = [x["media_url"].replace("\\", "") for x in tweet_json["entities"]["media"] if x["type"] == "photo"]
for url in urls:
response = requests.get(url)
img = Image.open(io.BytesIO(response.content))
# Extract text from image
img_text = pytesseract.image_to_string(img)
text += f' . {img_text}'
except KeyError:
pass
link = f'https://twitter.com/{screen_name}/status/{id}'
try:
self.tweet_callback(text, screen_name, link)
except:
pass
if __name__ == "__main__":
import time
twitter = Twitter()
while True:
time.sleep(1)
