65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
from pypvwatts import PVWatts
|
|
from pywatts.db import *
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
def fetch_data(from_lon, to_lon, from_lat, to_lat, step_size=1):
|
|
my_api_key = ''
|
|
with open(str(Path.home()) + '/pvwatts_api_key.txt', 'r') as file:
|
|
my_api_key = file.readline()
|
|
# Strip newline
|
|
my_api_key = my_api_key[:-1]
|
|
|
|
p = PVWatts(api_key=my_api_key)
|
|
|
|
results = []
|
|
|
|
for longitude in range(from_lon, to_lon, step_size):
|
|
for latitude in range(from_lat, to_lat, step_size):
|
|
try:
|
|
results.append(p.request(
|
|
format='JSON', system_capacity=4, module_type=1, array_type=1,
|
|
azimuth=190, tilt=40, dataset='intl', timeframe='hourly',
|
|
losses=10, lon=longitude, lat=latitude
|
|
))
|
|
except:
|
|
print("Could not fetch all data")
|
|
return results
|
|
|
|
return results
|
|
|
|
|
|
def store_data(result_list):
|
|
db.connect()
|
|
|
|
if not db.table_exists(WeatherStation):
|
|
db.create_tables([WeatherStation, Result])
|
|
|
|
for result in result_list:
|
|
if result.errors:
|
|
continue
|
|
|
|
if WeatherStation.select().where(WeatherStation.id == result.station_info['location']).exists():
|
|
print("Data for station %s already in database. Skipping." % result.station_info['location'])
|
|
continue
|
|
|
|
station = WeatherStation.create(
|
|
longitude=result.station_info['lon'],
|
|
latitude=result.station_info['lat'],
|
|
location=result.station_info['location'],
|
|
elevation=result.station_info['elev'],
|
|
city=result.station_info['city'],
|
|
state=result.station_info['state'],
|
|
id=result.station_info['location'],
|
|
)
|
|
|
|
Result.create(
|
|
station=station,
|
|
dc_output=result.outputs['dc'],
|
|
temperature=result.outputs['tamb'],
|
|
wind_speed=result.outputs['wspd']
|
|
)
|
|
|
|
db.close()
|