Update functions.py

functions.py

@@ -101,3 +101,89 @@ def get_model(project, model_name, evaluation_metric, sort_metrics_by):
 
     return model
 
+
+
+def get_weather_json(date, WEATHER_API_KEY):
+    return requests.get(f'https://weather.visualcrossing.com/VisualCrossingWebServices/rest/services/timeline/helsinki/{date}?unitGroup=metric&include=days&key={WEATHER_API_KEY}&contentType=json').json()
+
+
+def get_weather_data(date):
+    WEATHER_API_KEY = os.getenv('WEATHER_API_KEY')
+    json = get_weather_json(date, WEATHER_API_KEY)
+    data = json['days'][0]
+
+    return [
+        json['address'].capitalize(),
+        data['datetime'],
+        data['tempmax'],
+        data['tempmin'],
+        data['temp'],
+        data['feelslikemax'],
+        data['feelslikemin'],
+        data['feelslike'],
+        data['dew'],
+        data['humidity'],
+        data['precip'],
+        data['precipprob'],
+        data['precipcover'],
+        data['snow'],
+        data['snowdepth'],
+        data['windgust'],
+        data['windspeed'],
+        data['winddir'],
+        data['pressure'],
+        data['cloudcover'],
+        data['visibility'],
+        data['solarradiation'],
+        data['solarenergy'],
+        data['uvindex'],
+        data['conditions']
+    ]
+
+
+def get_weather_df(data):
+    col_names = [
+        'city',
+        'date',
+        'tempmax',
+        'tempmin',
+        'temp',
+        'feelslikemax',
+        'feelslikemin',
+        'feelslike',
+        'dew',
+        'humidity',
+        'precip',
+        'precipprob',
+        'precipcover',
+        'snow',
+        'snowdepth',
+        'windgust',
+        'windspeed',
+        'winddir',
+        'pressure',
+        'cloudcover',
+        'visibility',
+        'solarradiation',
+        'solarenergy',
+        'uvindex',
+        'conditions'
+    ]
+
+    new_data = pd.DataFrame(
+        data,
+        columns=col_names
+    )
+    new_data.date = new_data.date.apply(timestamp_2_time1)
+
+    return new_data
+
+def timestamp_2_time1(x):
+    dt_obj = datetime.strptime(str(x), '%Y-%m-%d')
+    dt_obj = dt_obj.timestamp() * 1000
+    return int(dt_obj)
+
+def timestamp_2_time(x):
+    dt_obj = datetime.strptime(str(x), '%m/%d/%Y')
+    dt_obj = dt_obj.timestamp() * 1000
+    return int(dt_obj)