initial commit

Author: casarock

.gitignore

@@ -0,0 +1,4 @@
+.vscode
+*.png
+*.code-workspace
+__pycache__

config.yaml

@@ -0,0 +1,10 @@
+csvUrl: 'https://raw.githubusercontent.com/CSSEGISandData/COVID-19/master/csse_covid_19_data/csse_covid_19_time_series/time_series_19-covid-Confirmed.csv'
+
+TwitterAPI:
+  consumer:
+    key: "xxxxxxxxxxxxxxxxx"
+    secret: "xxxxxxxxxxxxxxxxxxxxxxxx"
+
+  access:
+    token: "xxxxxxxxxxxxxxxxxxxx"
+    secret: "xxxxxxxxxxxxxxxxxxxxxxxxxxx"

coronaStats.py

@@ -0,0 +1,88 @@
+import pandas as pd
+import matplotlib.pyplot as plt
+
+class CoronaStats:
+
+    def __init__(self, csv_url):    
+        self.population = {'germany': 83019213,  'overall': 7418000000}
+        self.csv_url = csv_url
+        self.get_clean_data()
+
+    def get_clean_data(self):
+        df = pd.read_csv(self.csv_url)
+        df = df.drop(columns=['Lat', 'Long'])
+
+        df_group = df.groupby(['Country/Region'])
+
+        germany = df_group.get_group('Germany')
+
+        overall = df.agg(['sum'])
+        germany = germany.agg(['sum'])
+
+        self.overall = self.clean_Data(overall, 'overall', ['Country/Region'])
+        self.germany = self.clean_Data(germany, 'germany', ['Province/State', 'Country/Region'])
+
+    def clean_Data(self, data_frame, country, drop_columns):
+        cleaned_data_frame = data_frame.drop(columns = drop_columns)
+        cleaned_data_frame = cleaned_data_frame.T
+        
+        cleaned_data_frame['related'] = (100/self.population[country]) * cleaned_data_frame['sum']
+
+        return cleaned_data_frame
+
+    def plot_diagram_related(self, filename='confirmed_compared_related'):
+        diagram_file = self.plot_diagram('related', 
+                                         'Date', 
+                                         'Confirmed infections related to population (%)', 
+                                         'Confirmed COVID-19 - Related to population', 
+                                         filename)
+        
+        return diagram_file
+
+    def plot_diagram(self, yvalue, xlabel, ylabel, title, filename):
+        plt.figure(figsize=(8,5))
+        ax = plt.gca()
+
+        self.overall.plot(kind='line', y=yvalue, ax=ax, label="Overall", color='black')
+        self.germany.plot(kind='line', y=yvalue, ax=ax, label="Germany", color='red')
+
+        plt.xlabel(xlabel)
+        plt.ylabel(ylabel)
+        plt.title(title)
+        plt.savefig(filename + '.png', dpi=300)
+        
+        return filename + '.png' 
+    
+    def get_actual_infected(self):
+        germany_changed = self.germany.tail(2)['sum'][1] - self.germany.tail(2)['sum'][0]
+        overall_changed = self.overall.tail(2)['sum'][1] - self.overall.tail(2)['sum'][0]
+
+        germany = '+' if germany_changed > 0 else '-'
+        overall = '+' if overall_changed > 0 else '-'
+
+        germany += str(germany_changed)
+        overall += str(overall_changed)
+
+        return {
+            'germany': [
+                self.germany.tail(1)['sum'][0],
+                germany
+            ],
+            'overall': [
+                self.overall.tail(1)['sum'][0],
+                overall
+            ]
+        }
+
+    def get_latest_update_date(self):
+        [month, day, year] = self.overall.index[-1].split('/', 3)
+        date = '.'.join([day, month, '20'+year])
+        return date
+    
+    def get_weekly_growth(self):
+        frame = self.germany.tail(7)
+        actual = frame['sum'][-1]
+        days_before = frame['sum'][0]
+        growth_rate = (actual*100/days_before) - 100
+
+        return growth_rate

statsbot.py

@@ -0,0 +1,46 @@
+import tweepy
+import yaml
+import coronaStats as st
+
+def load_config():
+    with open("config.yaml", 'r') as stream:
+        try:
+            config = yaml.safe_load(stream)
+            return config
+
+        except yaml.YAMLError as exc:
+            print(exc)
+            exit
+            
+def tweetStats(text, config, image = None):
+    config = load_config()
+
+    # Authenticate to Twitter
+    auth = tweepy.OAuthHandler(config['TwitterAPI']['consumer']['key'], config['TwitterAPI']['consumer']['secret'])
+    auth.set_access_token(config['TwitterAPI']['access']['token'], config['TwitterAPI']['access']['secret'])
+    
+    api = tweepy.API(auth)
+
+    #Create a tweet
+    if image:
+        api.update_with_media(image, text)
+    else:
+        api.update_status(text)
+
+config = load_config()
+myfilename = 'compared_related'
+stats = st.CoronaStats(config['csvUrl'])
+myfile = stats.plot_diagram_related(myfilename)
+infected = stats.get_actual_infected()
+date = stats.get_latest_update_date()
+growth_germany = stats.get_weekly_growth()
+
+tweet_text = "Am " + date + " gab es " + str(infected['overall'][0]) + " (" + infected['overall'][1] +") erfasste Infektionen weltweit.\n"
+tweet_text += "In Deutschland: " + str(infected['germany'][0]) + " (" + infected['germany'][1] +") Infektionen.\n\n"
+tweet_text += "Wachstumsrate in Deutschland der letzten 7 Tage: " + "{0:.2f}".format(growth_germany) + "%\n\n"
+tweet_text += "Dashboard: https://my-covid-dash.herokuapp.com/"
+tweet_text += "\n\n#corona #mycoronastats #COVID19deutschland"
+
+tweetStats(tweet_text, config, myfile)
+
+exit