forked from Silias-Public/Sonnendach
implemented alway use api to get Data. Open Sonnendach pages by url
This commit is contained in:
schrom01
parent
af460021b0
commit
b620704a54
300
Sonnendach.py
300
Sonnendach.py
|
|
@ -16,7 +16,7 @@ from selenium.webdriver.common.action_chains import ActionChains
|
|||
import json
|
||||
import requests
|
||||
|
||||
DropDownLabels = ["Street: ", "Number: ", "Postal code: ", "City: ", "Sonnendach URL: ", "Suitability", "Image Filename Map: ", "Image Filename Production: ", "Image Filename qrcode: ", "PV Production 50", "PV Production 75", "PV Production 100", "Value Electricity production", "Roof area"]
|
||||
DropDownLabels = ["Street: ", "Number: ", "Postal code: ", "City: ", "coordinate X", "coordinate Y", "Sonnendach URL: ", "Suitability", "Image Filename Map: ", "Image Filename Production: ", "Image Filename qrcode: ", "PV Production 50", "PV Production 75", "PV Production 100", "Value Electricity production", "Roof area"]
|
||||
file_split_char = ","
|
||||
OptionList = []
|
||||
exit = False
|
||||
|
|
@ -49,117 +49,25 @@ def search_adresses(adress_list, filename_adresslist, driver, mainText):
|
|||
address_count = len(adress_list)
|
||||
#Search for each adress in the list and create screenshots and qrcodes
|
||||
for i in range(address_count):
|
||||
url = "not-found"
|
||||
suitability = "not-found"
|
||||
pv_Production50 = "not-found"
|
||||
pv_Production75 = "not-found"
|
||||
pv_Production100 = "not-found"
|
||||
value_electricity_production = "not-found"
|
||||
roof_area = "not-found"
|
||||
found = True
|
||||
line = adress_list[i]
|
||||
adress = line.split(file_split_char)
|
||||
if((line != adress_list[0]) & (adress[columnIndexes[4]] == "") & (len(adress[columnIndexes[0]]) > 0) & (len(adress[columnIndexes[1]]) > 0) & (len(adress[columnIndexes[2]]) > 0) & (len(adress[columnIndexes[3]]) > 0)):
|
||||
search_string = adress[columnIndexes[0]] + " " + adress[columnIndexes[1]] + " " + adress[columnIndexes[2]] + " " + adress[columnIndexes[3]]
|
||||
|
||||
if(createImages.get() > 0):
|
||||
search_bar = driver.find_element(By.ID, "searchTypeahead1")
|
||||
search_bar.send_keys(Keys.CONTROL + "a")
|
||||
search_bar.send_keys(Keys.DELETE)
|
||||
search_bar.send_keys(search_string)
|
||||
driver.implicitly_wait(10)
|
||||
time.sleep(0.7)
|
||||
found = False
|
||||
suggestions = driver.find_elements(By.XPATH, "//div[@class='tt-suggestion tt-selectable']")
|
||||
for suggestion in suggestions:
|
||||
if(suggestion.text == search_string):
|
||||
found = True
|
||||
suggestion.click()
|
||||
driver.implicitly_wait(10)
|
||||
break
|
||||
|
||||
if(found):
|
||||
time.sleep(0.7)
|
||||
url = driver.current_url
|
||||
suitability = driver.find_element(By.ID, "eignung").text
|
||||
pv_Production50 = driver.find_element(By.ID, "pv50").text.replace("'", "")
|
||||
pv_Production75 = driver.find_element(By.ID, "pv75").text.replace("'", "")
|
||||
pv_Production100 = driver.find_element(By.ID, "pv100").text.replace("'", "")
|
||||
value_electricity_production = driver.find_elements(By.XPATH, "//h2[@id='TitelSolarstrom']//strong")[2].text.replace("'", "").replace(" Franken", "")
|
||||
roof_area = driver.find_element(By.ID, "areaOutput").text
|
||||
|
||||
time.sleep(2)
|
||||
image_filename = suitability + " - " + search_string
|
||||
image_folder_map = "screenshots/"
|
||||
image_filename_map = image_filename + " map" + ".png"
|
||||
image_folder_production = "screenshots/"
|
||||
image_filename_production = image_filename + " production" + ".png"
|
||||
image_folder_qrcode = "qrcodes/"
|
||||
image_filename_qrcode = image_filename + " qrcode" + ".png"
|
||||
|
||||
# Create QR-Code
|
||||
qr = qrcode.QRCode(version=1, box_size=10, border=5)
|
||||
qr.add_data(url)
|
||||
qr.make(fit=True)
|
||||
qr.make_image(fill='black', back_color='white').save(image_folder_qrcode + image_filename_qrcode)
|
||||
|
||||
# Create Screenshot of Map
|
||||
actions = ActionChains(driver)
|
||||
try:
|
||||
driver.execute_script("""var l = document.getElementsByClassName("ol-zoom ol-unselectable ol-control")[0];
|
||||
l.parentNode.removeChild(l);""")
|
||||
except:
|
||||
pass
|
||||
|
||||
mapElement = driver.find_element(By.XPATH,
|
||||
"//div[@id='map']//div[@class='ol-viewport']")
|
||||
location = mapElement.location
|
||||
size = mapElement.size
|
||||
x = location["x"]
|
||||
y = location["y"]
|
||||
w = x + size["width"]
|
||||
h = size["height"]
|
||||
area = (x, 0, w, h)
|
||||
driver.execute_script("window.scrollTo(0, " + str(y) + ")")
|
||||
time.sleep(1)
|
||||
driver.save_screenshot(image_folder_map + image_filename_map)
|
||||
time.sleep(0.2)
|
||||
Image.open(image_folder_map + image_filename_map).crop(area).save(
|
||||
image_folder_map + image_filename_map)
|
||||
|
||||
# Create Screenshot of Chart
|
||||
try:
|
||||
chartElement = driver.find_elements(By.XPATH, "//div[@id='chart']//*")[0]
|
||||
location = chartElement.location
|
||||
size = chartElement.size
|
||||
x = location["x"]
|
||||
y = location["y"]
|
||||
w = x + size["width"]
|
||||
h = size["height"]
|
||||
area = (x, 0, w, h)
|
||||
driver.execute_script("window.scrollTo(0, " + str(y) + ")")
|
||||
|
||||
time.sleep(1)
|
||||
driver.save_screenshot(image_folder_production + image_filename_production)
|
||||
time.sleep(0.2)
|
||||
Image.open(image_folder_production + image_filename_production).crop(area).save(
|
||||
image_folder_production + image_filename_production)
|
||||
except:
|
||||
image_filename_production = "not-found"
|
||||
|
||||
print("saved Address: " + search_string)
|
||||
else:
|
||||
url = "not-found"
|
||||
suitability = "not-found"
|
||||
image_filename_map = "not-found"
|
||||
image_filename_production = "not-found"
|
||||
image_filename_qrcode = "not-found"
|
||||
pv_Production50 = "not-found"
|
||||
pv_Production75 = "not-found"
|
||||
pv_Production100 = "not-found"
|
||||
value_electricity_production = "not-found"
|
||||
roof_area = "not-found"
|
||||
|
||||
print("not found: " + search_string)
|
||||
|
||||
else: # If no images must be created
|
||||
if((line != adress_list[0]) and (adress[columnIndexes[6]] == "") and (((len(adress[columnIndexes[0]]) > 0) and (len(adress[columnIndexes[1]]) > 0) and (len(adress[columnIndexes[2]]) > 0) and (len(adress[columnIndexes[3]]) > 0)) or ((len(adress[columnIndexes[4]]) > 0) and (len(adress[columnIndexes[5]]) > 0)))):
|
||||
address_string = adress[columnIndexes[0]] + " " + adress[columnIndexes[1]] + " " + adress[columnIndexes[2]] + " " + adress[columnIndexes[3]]
|
||||
if(not ((len(adress[columnIndexes[4]]) > 0) and (len(adress[columnIndexes[5]]) > 0))):
|
||||
# Get Coordinates for Address
|
||||
try:
|
||||
# Address to coordinates
|
||||
params = {
|
||||
"type": "locations",
|
||||
"searchText": search_string,
|
||||
"searchText": address_string,
|
||||
"lang": "de"}
|
||||
locationsResponse = requests.get('https://api3.geo.admin.ch/rest/services/api/SearchServer',
|
||||
params=params)
|
||||
|
|
@ -167,10 +75,20 @@ def search_adresses(adress_list, filename_adresslist, driver, mainText):
|
|||
firstLocation = locations['results'][0]['attrs']
|
||||
|
||||
# you need the coordinates from other services, once you have them
|
||||
coordinates = (firstLocation['y'], firstLocation['x']) # example coordinates
|
||||
coordinates = (float(firstLocation['y']), float(firstLocation['x'])) # example coordinates
|
||||
except:
|
||||
found = False
|
||||
else:
|
||||
# Get Coordinates from File
|
||||
coordinate_x = float(adress[columnIndexes[4]]) % 1000000
|
||||
coordinate_y = float(adress[columnIndexes[5]]) % 1000000
|
||||
coordinates = (coordinate_x, coordinate_y)
|
||||
|
||||
if(found):
|
||||
# Get Building ID from coordinates
|
||||
try:
|
||||
coordinates_str = ','.join(map(str, coordinates))
|
||||
tolerance = 1
|
||||
|
||||
# this is wrong, bbox should be computed using tolerance and pixel size, but there are no problems for the
|
||||
# purpose of this exercise
|
||||
bbox = (coordinates[0] - tolerance, coordinates[1] - tolerance, coordinates[0] + tolerance,
|
||||
|
|
@ -195,7 +113,11 @@ def search_adresses(adress_list, filename_adresslist, driver, mainText):
|
|||
# for result in results:
|
||||
# print(result)
|
||||
building_id = (response_one_roof_decoded)['results'][0]['attributes']['building_id']
|
||||
except:
|
||||
found = False
|
||||
|
||||
if(found):
|
||||
try:
|
||||
# use the building_id of previous request to get all roofs associated with this building_id
|
||||
params = {
|
||||
'layer': 'ch.bfe.solarenergie-eignung-daecher',
|
||||
|
|
@ -213,52 +135,45 @@ def search_adresses(adress_list, filename_adresslist, driver, mainText):
|
|||
best_roof = roof
|
||||
url = "https://www.uvek-gis.admin.ch/BFE/sonnendach/index.html?featureId=" + str(best_roof['featureId']) + "&lang=de"
|
||||
suitability = str(best_roof['attributes']['klasse'])
|
||||
image_filename_map = "not-created"
|
||||
image_filename_production = "not-created"
|
||||
image_filename_qrcode = "not-created"
|
||||
pv_Production100 = str(float(best_roof['attributes']['gstrahlung'])*0.2*0.8)
|
||||
pv_Production75 = str(float(pv_Production100) * 0.75)
|
||||
pv_Production50 = str(float(pv_Production100) * 0.5)
|
||||
value_electricity_production = str(best_roof['attributes']['finanzertrag'])
|
||||
roof_area = str(best_roof['attributes']['flaeche'])
|
||||
print("saved Address: " + search_string)
|
||||
except:
|
||||
url = "not-found"
|
||||
suitability = "not-found"
|
||||
image_filename_map = "not-found"
|
||||
image_filename_production = "not-found"
|
||||
image_filename_qrcode = "not-found"
|
||||
pv_Production50 = "not-found"
|
||||
pv_Production75 = "not-found"
|
||||
pv_Production100 = "not-found"
|
||||
value_electricity_production = "not-found"
|
||||
roof_area = "not-found"
|
||||
found = False
|
||||
|
||||
roof_is_suitable = True # TODO decide if roof is suitable
|
||||
|
||||
if(found and createImages.get() > 0 and roof_is_suitable):
|
||||
image_filename = suitability + " - " + address_string
|
||||
image_folder_map = "screenshots/"
|
||||
image_filename_map = image_filename + " map" + ".png"
|
||||
image_folder_production = "screenshots/"
|
||||
image_filename_production = image_filename + " production" + ".png"
|
||||
image_folder_qrcode = "qrcodes/"
|
||||
image_filename_qrcode = image_filename + " qrcode" + ".png"
|
||||
create_images(address_string, driver, image_filename_map, image_filename_production,
|
||||
image_filename_qrcode, image_folder_map, image_folder_production, image_folder_qrcode,
|
||||
url)
|
||||
|
||||
else:
|
||||
image_filename_map = "not-created"
|
||||
image_filename_production = "not-created"
|
||||
image_filename_qrcode = "not-created"
|
||||
|
||||
save_information_to_addresslist(adress, adress_list, columnIndexes, coordinates, file_split_char, i,
|
||||
image_filename_map, image_filename_production, image_filename_qrcode,
|
||||
pv_Production100, pv_Production50, pv_Production75, roof_area, suitability,
|
||||
url, value_electricity_production)
|
||||
|
||||
save_addresslist_in_file(adress_list, filename_adresslist)
|
||||
|
||||
print("saved Address: " + address_string)
|
||||
print("Address " + str(i) + " of " + str(address_count))
|
||||
timediff = relativedelta(datetime.datetime.now(), startTime)
|
||||
print("Process is running since %d years %d months %d days %d hours %d minutes %d seconds" % (timediff.years, timediff.months, timediff.days, timediff.hours, timediff.minutes, timediff.seconds))
|
||||
|
||||
|
||||
#Write back into file
|
||||
adress[columnIndexes[4]] = url
|
||||
adress[columnIndexes[5]] = suitability
|
||||
adress[columnIndexes[6]] = image_filename_map
|
||||
adress[columnIndexes[7]] = image_filename_production
|
||||
adress[columnIndexes[8]] = image_filename_qrcode
|
||||
adress[columnIndexes[9]] = pv_Production50
|
||||
adress[columnIndexes[10]] = pv_Production75
|
||||
adress[columnIndexes[11]] = pv_Production100
|
||||
adress[columnIndexes[12]] = value_electricity_production
|
||||
adress[columnIndexes[13]] = roof_area
|
||||
adress_file = open(filename_adresslist, "w", encoding="utf-8")
|
||||
new_line_string = ""
|
||||
for j in adress:
|
||||
new_line_string = new_line_string + j + file_split_char
|
||||
adress_list[i] = new_line_string
|
||||
new_adress_list = ""
|
||||
for j in adress_list:
|
||||
new_adress_list = new_adress_list + (j) + "\n"
|
||||
adress_file.write(new_adress_list)
|
||||
adress_file.close()
|
||||
print("Process is running since %d years %d months %d days %d hours %d minutes %d seconds" % (
|
||||
timediff.years, timediff.months, timediff.days, timediff.hours, timediff.minutes, timediff.seconds))
|
||||
|
||||
|
||||
if(stopThread == True):
|
||||
|
|
@ -270,6 +185,96 @@ def search_adresses(adress_list, filename_adresslist, driver, mainText):
|
|||
print("closing Thread")
|
||||
return
|
||||
|
||||
|
||||
def save_information_to_addresslist(adress, adress_list, columnIndexes, coordinates, file_split_char, list_index,
|
||||
image_filename_map, image_filename_production, image_filename_qrcode,
|
||||
pv_Production100, pv_Production50, pv_Production75, roof_area, suitability, url,
|
||||
value_electricity_production):
|
||||
# Save Information in Addresslist
|
||||
adress[columnIndexes[4]] = str(coordinates[0])
|
||||
adress[columnIndexes[5]] = str(coordinates[1])
|
||||
adress[columnIndexes[6]] = url
|
||||
adress[columnIndexes[7]] = suitability
|
||||
adress[columnIndexes[8]] = image_filename_map
|
||||
adress[columnIndexes[9]] = image_filename_production
|
||||
adress[columnIndexes[10]] = image_filename_qrcode
|
||||
adress[columnIndexes[11]] = pv_Production50
|
||||
adress[columnIndexes[12]] = pv_Production75
|
||||
adress[columnIndexes[13]] = pv_Production100
|
||||
adress[columnIndexes[14]] = value_electricity_production
|
||||
adress[columnIndexes[15]] = roof_area
|
||||
new_line_string = ""
|
||||
for j in adress:
|
||||
new_line_string = new_line_string + j + file_split_char
|
||||
adress_list[list_index] = new_line_string
|
||||
|
||||
|
||||
def save_addresslist_in_file(adress_list, filename_adresslist):
|
||||
new_adress_list = ""
|
||||
for j in adress_list:
|
||||
new_adress_list = new_adress_list + (j) + "\n"
|
||||
# save File
|
||||
adress_file = open(filename_adresslist, "w", encoding="utf-8")
|
||||
adress_file.write(new_adress_list)
|
||||
adress_file.close()
|
||||
|
||||
|
||||
def create_images(address_string, driver, image_filename_map, image_filename_production, image_filename_qrcode,
|
||||
image_folder_map, image_folder_production, image_folder_qrcode, url):
|
||||
driver.get(url)
|
||||
time.sleep(2)
|
||||
|
||||
# Create QR-Code
|
||||
qr = qrcode.QRCode(version=1, box_size=10, border=5)
|
||||
qr.add_data(url)
|
||||
qr.make(fit=True)
|
||||
qr.make_image(fill='black', back_color='white').save(image_folder_qrcode + image_filename_qrcode)
|
||||
|
||||
# Create Screenshot of Map
|
||||
actions = ActionChains(driver)
|
||||
try:
|
||||
driver.execute_script("""var l = document.getElementsByClassName("ol-zoom ol-unselectable ol-control")[0];
|
||||
l.parentNode.removeChild(l);""")
|
||||
except:
|
||||
pass
|
||||
|
||||
mapElement = driver.find_element(By.XPATH,
|
||||
"//div[@id='map']//div[@class='ol-viewport']")
|
||||
location = mapElement.location
|
||||
size = mapElement.size
|
||||
x = location["x"]
|
||||
y = location["y"]
|
||||
w = x + size["width"]
|
||||
h = size["height"]
|
||||
area = (x, 0, w, h)
|
||||
driver.execute_script("window.scrollTo(0, " + str(y) + ")")
|
||||
time.sleep(1)
|
||||
driver.save_screenshot(image_folder_map + image_filename_map)
|
||||
time.sleep(0.2)
|
||||
Image.open(image_folder_map + image_filename_map).crop(area).save(
|
||||
image_folder_map + image_filename_map)
|
||||
|
||||
# Create Screenshot of Chart
|
||||
try:
|
||||
chartElement = driver.find_elements(By.XPATH, "//div[@id='chart']//*")[0]
|
||||
location = chartElement.location
|
||||
size = chartElement.size
|
||||
x = location["x"]
|
||||
y = location["y"]
|
||||
w = x + size["width"]
|
||||
h = size["height"]
|
||||
area = (x, 0, w, h)
|
||||
driver.execute_script("window.scrollTo(0, " + str(y) + ")")
|
||||
|
||||
time.sleep(1)
|
||||
driver.save_screenshot(image_folder_production + image_filename_production)
|
||||
time.sleep(0.2)
|
||||
Image.open(image_folder_production + image_filename_production).crop(area).save(
|
||||
image_folder_production + image_filename_production)
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
def read_adresslist(filename_adresslist):
|
||||
returnvalues = []
|
||||
try:
|
||||
|
|
@ -323,7 +328,7 @@ def command():
|
|||
thread_search_adresses.join()
|
||||
except:
|
||||
pass
|
||||
print("Thread is closed")
|
||||
print("Threads are closed")
|
||||
root.quit()
|
||||
print("Application is closed")
|
||||
driver.quit()
|
||||
|
|
@ -388,7 +393,6 @@ def command_exit():
|
|||
|
||||
|
||||
|
||||
|
||||
root = tkinter.Tk()
|
||||
root.protocol("WM_DELETE_WINDOW", command_exit)
|
||||
root.wm_title("Sonnendach")
|
||||
|
|
|
|||
|
|
@ -22,6 +22,7 @@ tolerance=1
|
|||
# purpose of this exercise
|
||||
bbox=(coordinates[0]-tolerance, coordinates[1]-tolerance, coordinates[0]+tolerance,coordinates[1]+tolerance)
|
||||
bbox_str=','.join(map(str,bbox))
|
||||
# retrieve one roof and the building_id associated
|
||||
params={
|
||||
"geometryType":"esriGeometryPoint",
|
||||
"returnGeometry":"true",
|
||||
|
|
@ -33,12 +34,12 @@ params={
|
|||
"order":"distance",
|
||||
"lang":"de",
|
||||
}
|
||||
# retrieve one roof and the building_id associated
|
||||
print(coordinates_str)
|
||||
response_one_roof = requests.get('https://api3.geo.admin.ch/rest/services/api/MapServer/identify', params=params)
|
||||
response_one_roof_decoded = json.loads(response_one_roof.content.decode())
|
||||
results = (response_one_roof_decoded['results'])
|
||||
# for result in results:
|
||||
# print(result['attributes']['building_id'])
|
||||
for result in results:
|
||||
print(result['attributes']['building_id'])
|
||||
building_id=(response_one_roof_decoded)['results'][0]['attributes']['building_id']
|
||||
|
||||
# use the building_id of previous request to get all roofs associated with this building_id
|
||||
|
|
@ -49,7 +50,6 @@ params = {
|
|||
'searchText': str(building_id),
|
||||
'returnGeometry': 'false'
|
||||
}
|
||||
|
||||
response_all_roofs = requests.get('https://api3.geo.admin.ch/rest/services/api/MapServer/find', params=params)
|
||||
response_all_roofs_decoded = json.loads(response_all_roofs.content.decode())['results']
|
||||
for roof in response_all_roofs_decoded:
|
||||
|
|
|
|||
Loading…
Reference in New Issue