mixing it up a little bit

Author: fmiskovic

bot.py

@@ -1,4 +1,5 @@
 import logging.config
+import random
 import time
 
 from news import news_grabber, news_api_auth
@@ -17,24 +18,27 @@
 followers_handler = followers.FollowersHandler(tw_api)
 
 
-def tweet_something_about(keyword):
-    news = news_grabber.get_news(keyword, category='business')
-    if len(news) == 0:
-        logger.info(f'There are no news about %s at the moment', keyword)
-        return
+def get_news(query=None, category='business', country=None):
+    return news_grabber.get_news(query=query, category=category, country=country)
 
-    logger.info('Tweeting about ' + keyword)
 
-    for n in news:
-        tw_handler.post_new_tweet(n.description, n.url)
-        logging.info(f'Taking a %d minutes break...', INTERVAL)
-        time.sleep(INTERVAL)
-        logger.info('OK I am ready to continue. Lets tweet something...')
+def tweet_news(news_items, hash_tag=None):
+    for n in news_items:
+        result = tw_handler.post_new_tweet(text=n.description, source=n.url, hashtag=hash_tag)
+        if result is not None:
+            logging.info(f'Taking a %s minutes break...', INTERVAL / 60)
+            time.sleep(INTERVAL)
+            logger.info('OK I am ready to continue. Lets tweet something again...')
 
 
+keywords = ['cryptocurrency', 'ethereum', 'bitcoin', 'cryptocoin', 'litecoin']
+
 while True:
     # follow new followers
     followers_handler.follow_followers()
 
-    # check for cryptocurrency news and tweet about it
-    tweet_something_about('cryptocurrency')
+    keyword = random.choice(keywords)
+    news = get_news(query=keyword)
+    if len(news) > 0:
+        logger.info('Tweet attempt about ' + keyword)
+        tweet_news(news_items=news, hash_tag='#cryptocurrencynews #' + keyword)

news/logging.config

@@ -10,14 +10,13 @@ def __init__(self, news_api):
         """Init news api client"""
         self.news_api = news_api
 
-        logging.config.fileConfig('logging.config')
-        self.logger = logging.getLogger('news')
+        self.logger = logging.getLogger(__name__)
         self.logger.info('Initialized NewsApiClient')
 
-    def get_news(self, query, category=None):
+    def get_news(self, query, category=None, country=None):
         """get latest breaking news"""
         news = []
-        response = self.news_api.get_top_headlines(query, category=category)
+        response = self.news_api.get_top_headlines(query, category=category, country=country)
         if response['status'] == 'ok':
             self.logger.info('Querying news went ok')
             for article in response['articles']:

news/news_grabber.py

@@ -13,28 +13,28 @@ def tearDown(self) -> None:
         super().tearDown()
 
     def test_get_crypto_news(self):
-        news = self.grabber.get_news('cryptocurrency', category='business')
+        news = self.grabber.get_news(query='cryptocurrency', category='business')
         print('crypto')
         for n in news:
             print(n)
         self.assertTrue(len(news) > 0)
 
     def test_get_crypto_tech_news(self):
-        news = self.grabber.get_news('cryptocurrency', category='business')
+        news = self.grabber.get_news(query='cryptocurrency', category='business')
         print('crypto')
         for n in news:
             print(n)
         self.assertTrue(len(news) > 0)
 
     def test_get_bitcoin_news(self):
-        news = self.grabber.get_news('bitcoin')
+        news = self.grabber.get_news(query='bitcoin')
         print('bitcoin')
         for n in news:
             print(n)
         self.assertTrue(len(news) > 0)
 
     def test_get_news_non_existing(self):
-        news = self.grabber.get_news('fqwsdzx')
+        news = self.grabber.get_news(query='fqwsdzx')
         self.assertTrue(len(news) == 0)
 
 

test/test_news_grabber.py

@@ -15,15 +15,15 @@ def tearDown(self) -> None:
     def test_create_post(self):
         source = "https://slashdot.org/story/19/05/19/1837257/bitcoin-roars-back-surges-50-in-30-days"
         text = "Bitcoin 'Roars Back', Surges 50% in 30 Days"
-        tweet = tweets.create_post(text, source)
+        tweet = tweets.create_post(text=text, source=source)
         print(tweet)
         self.assertIsNotNone(tweet)
 
     def test_tweet_news(self):
         source = "https://slashdot.org/story/19/05/19/1837257/bitcoin-roars-back-surges-50-in-30-days"
         text = "Bitcoin 'Roars Back', Surges 50% in 30 Days"
 
-        response = self.post_bot.post_new_tweet(text, source)
+        response = self.post_bot.post_new_tweet(text=text, source=source)
         self.assertIsNotNone(response)
 
     def test_get_latest_tweet(self):
@@ -39,7 +39,7 @@ def test_get_latest_tweet_url(self):
         text = tweet.text
 
         sub = text[0:text.find('\n\n')]
-        logger.info(f'Substring is: %s', sub)
+        print(f'Substring is: %s', sub)
         self.assertIsNotNone(sub)
         self.assertTrue(len(sub) > 0)
 

test/test_twitter_post.py

@@ -10,8 +10,7 @@ def __init__(self, tw_api):
         self.api = tw_api
         self.me = self.api.me()
 
-        logging.config.fileConfig('logging.config')
-        self.logger = logging.getLogger('tw')
+        self.logger = logging.getLogger(__name__)
         self.logger.info('Initializing FollowersHandler')
 
     def follow_followers(self):

twitter/followers.py

@@ -3,13 +3,12 @@
 from twitter import tweet_similarity
 
 
-def create_post(text, source):
-    hashtags = '#cryptocurrencynews #cryptocurrency'
-    if len(text) >= 140:
-        sub_text = text[0:136] + '...'
-        return sub_text + '\n\n' + hashtags + '\n\n' + source
+def create_post(text, source, *hashtag):
+    if len(text) >= 170:
+        sub_text = text[0:166] + '...'
+        return sub_text + '\n\n' + hashtag + '\n\n' + source
     else:
-        return text + '\n\n' + hashtags + '\n\n' + source
+        return text + '\n\n' + hashtag + '\n\n' + source
 
 
 class TweetsHandler:
@@ -19,11 +18,10 @@ def __init__(self, tw_api):
         self.api = tw_api
         self.me = self.api.me()
 
-        logging.config.fileConfig('logging.config')
-        self.logger = logging.getLogger('tw')
+        self.logger = logging.getLogger(__name__)
         self.logger.info('Initialized TweetsHandler')
 
-    def post_new_tweet(self, text, source):
+    def post_new_tweet(self, text, source, hashtag=None):
         """Tweet some news"""
         latest_tweets = self.get_tweets_list(100)
         if len(latest_tweets) > 0:
@@ -34,9 +32,10 @@ def post_new_tweet(self, text, source):
 
         try:
             self.logger.info('Posting a new tweet...')
-            return self.api.update_status(create_post(text, source))
+            return self.api.update_status(create_post(text, source, hashtag))
         except Exception as e:
             self.logger.error('Failed to post new tweet: %s', e, exc_info=1)
+            return None
 
     def get_latest_tweet(self):
         self.logger.info('Getting my last tweet...')