forked from felsen/WMS
-
Notifications
You must be signed in to change notification settings - Fork 0
/
weather_test.py
132 lines (111 loc) · 4.89 KB
/
weather_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
# -*- coding: utf-8 -*-
# Author: Edison Suen
# Description: Scrape "The Weather Network" to extract the 7 day forecast
import requests
import pandas as pd
import re
from listofcountries import country_dict
from bs4 import BeautifulSoup
country = raw_input("Please enter the country: ")
country = country.upper()
province = raw_input("Please enter a province/state: ")
province = province.replace(' ', '-').lower()
city = raw_input("Please enter a city: ")
city = city.replace(' ', '-').lower()
units = raw_input("Metric or Imperial? ")
units = units.title()
print("\n")
def get_value(dic,value):
for name in dic:
if dic[name] == value:
return name.lower()
country_abbrev = get_value(country_dict,country)
base_url = "https://www.theweathernetwork.com/{}/weather/".format(country_abbrev)
headers = {'User-agent': "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36"}
website = requests.get((base_url+"{}"+"/"+"{}").format(province,city),headers=headers)
soup = BeautifulSoup(website.content, 'html.parser')
###### FOR TESTING PURPOSES ######
forecast = soup.find(id="seven-days-only")
items = forecast.find_all(class_="seven-day-column")
column = items[0]
period = column.find(class_ ="day_title").get_text()
outlook = column.find(class_="day_outlook").get_text()
feels_celcius = (column.find_all("span"))[4]
feels_feh = (column.find_all("span"))[5]
temp_celcius = column.find(class_="chart-daily-temp seven_days_metric seven_days_metric_c").get_text()
temp_feh = column.find(class_="chart-daily-temp seven_days_imperial seven_days_imperial_f").get_text()
night_celcius = column.find(class_="chart-daily-temp-low seven_days_metric seven_days_metric_c").get_text()
night_feh = column.find(class_="chart-daily-temp-low seven_days_imperial seven_days_imperial_f").get_text()
try:
rain_metric = column.find(class_="fx-details seven_days_metric rain seven_days_metric_kmh").get_text()
rain_imperial = column.find(class_="fx-details seven_days_imperial rain seven_days_imperial_mph").get_text()
except AttributeError:
pass
wind_metric = column.find(class_="fx-details seven_days_metric seven_days_metric_kmh wind")
wind_imperial = column.find(class_="fx-details seven_days_imperial seven_days_imperial_mph wind")
###### END TEST VARIABLES ######
###### CONDITIONALS ######
if units== "Metric":
days = [dt.get_text() for dt in forecast.select(".seven-day-column .day_title")]
temp = [t.get_text() for t in forecast.select(".seven-day-column .chart-daily-temp.seven_days_metric.seven_days_metric_c")]
temp_night = [tn.get_text() for tn in forecast.select(".seven-day-column .chart-daily-temp-low.seven_days_metric.seven_days_metric_c")]
rain = [r.get_text() for r in forecast.select(".seven-day-column .fx-details.seven_days_metric.rain.seven_days_metric_kmh")]
feels_arr = []
for feels in items:
feels_like = (feels.find_all("span"))[4]
feels_arr.append(feels_like)
for i in range(7):
feels_arr[i] = str(feels_arr[i])
feels_arr[i] = feels_arr[i]
elif units == "Imperial":
days = [dt.get_text() for dt in forecast.select(".seven-day-column .day_title")]
temp = [t.get_text() for t in forecast.select(".seven-day-column .chart-daily-temp.seven_days_imperial.seven_days_imperial_f")]
temp_night = [tn.get_text() for tn in forecast.select(".seven-day-column .chart-daily-temp-low.seven_days_imperial.seven_days_imperial_f")]
rain = [r.get_text() for r in forecast.select(".seven-day-column .fx-details.seven_days_imperial.rain.seven_days_imperial_mph")]
feels_arr = []
for feels in items:
feels_like = (feels.find_all("span"))[5]
feels_arr.append(feels_like)
for i in range(7):
feels_arr[i] = str(feels_arr[i])
feels_arr[i] = feels_arr[i]
###### END CONDITIONALS ######
###### DATA FRAME ######
try:
weather = pd.DataFrame({
"Days": days,
"Temp": temp,
"Night": temp_night,
"Rain ": rain,
" Feels like": feels_arr
})
except:
weather = pd.DataFrame({
"Days": days,
"Temp": temp,
"Night": temp_night,
" Feels like": feels_arr
})
try:
weather = weather[["Days", " Feels like", "Temp", "Night", "Rain "]]
except:
weather = weather[["Days", " Feels like", "Temp", "Night"]]
weather.index += 1
weather["Days"] = weather["Days"].map(lambda x: x.lstrip('\t'))
weather["Temp"] = weather["Temp"].map(lambda x: x.lstrip('\t'))
if units == "Metric":
weather["Night"] = weather["Night"].map(lambda x: x.lstrip('\t') + "C")
else:
weather["Night"] = weather["Night"].map(lambda x: x.lstrip('\t') + "F")
try:
weather["Rain "] = weather["Rain "].map(lambda x: x.lstrip('\t').rstrip('\t'))
except:
pass
if units == "Metric":
weather[" Feels like"] = weather[" Feels like"].map(lambda x: x.lstrip("<span>").rstrip("</span>")+u"\u00b0 " + "C")
else:
weather[" Feels like"] = weather[" Feels like"].map(lambda x: x.lstrip("<span>").rstrip("</span>")+u"\u00b0" + "F")
###### END DATA FRAME ######
###### RUN ######
print(weather)
###### END RUN ######