Silias-Public
/
Sonnendach
2
0
Fork 1
Code Issues Pull Requests Projects Releases 4 Wiki Activity

Merge pull request #1 from romanschenk37/Screenshot 

Screenshot
This commit is contained in:
romanschenk37 2022-03-15 08:17:22 +01:00 committed by GitHub
parent f8382284f8 4b11d0e4de
commit e3eb20d61f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 48 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
Main.py
View File

@ -6,12 +6,13 @@ from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from PIL import Image
import os
import qrcode
from tkinter import filedialog
import time
DropDownLabels = ["Street: ", "Number: ", "Postal code: ", "City: ", "Sonnendach URL: ", "Eignung", "Screenshot Filename: "]
DropDownLabels = ["Street: ", "Number: ", "Postal code: ", "City: ", "Sonnendach URL: ", "Eignung", "Image Filename: ", "PV Production 50", "PV Production 75", "PV Production 100", "Value Electricity production"]
file_split_char = ","
OptionList = []
exit = False
@ -26,16 +27,21 @@ outputtext = "Welcome to application Sonnendach\n"
columnIndexes = []
def search_adresses(adress_list, filename_adresslist, driver):
global file_split_char
global stopThread
global columnIndexes
if(not os.path.exists("screenshots")):
os.makedirs("screenshots", exist_ok=False)
if (not os.path.exists("qrcodes")):
os.makedirs("qrcodes", exist_ok=False)
for i in range(len(adress_list)):
line = adress_list[i]
adress = line.split(file_split_char)
if((line != adress_list[0]) & ((adress[20][0:11] != "screenshots"))):
if((line != adress_list[0]) & ((adress[columnIndexes[6]] == ""))):
print(adress)
search_string = adress[columnIndexes[0]] + " " + adress[columnIndexes[1]] + " " + adress[columnIndexes[2]] + " " + adress[columnIndexes[3]]
search_bar = driver.find_element(By.ID, "searchTypeahead1")
@ -50,19 +56,32 @@ def search_adresses(adress_list, filename_adresslist, driver):
if(suggestion.text == search_string):
found = True
suggestion.click()
driver.implicitly_wait(120)
driver.implicitly_wait(10)
break
if(found):
time.sleep(2)
# TODO: Validation
url = driver.current_url
eignung = driver.find_element(By.ID, "eignung")
image_filename = "screenshots/" + eignung.text + " - " + search_string + ".png"
pv_Production50 = driver.find_element(By.ID, "pv50")
pv_Production75 = driver.find_element(By.ID, "pv75")
pv_Production100 = driver.find_element(By.ID, "pv100")
value_electricity_production = driver.find_elements(By.XPATH, "//h2[@id='TitelSolarstrom']//strong")[2]
image_filename = eignung.text + " - " + search_string + ".png"
adress[columnIndexes[4]] = url
adress[columnIndexes[5]] = eignung.text
adress[columnIndexes[6]] = image_filename
# TODO: 1000er Trennzeichen und " Franken" entfernen
adress[columnIndexes[7]] = pv_Production50.text.replace("'", "")
adress[columnIndexes[8]] = pv_Production75.text.replace("'", "")
adress[columnIndexes[9]] = pv_Production100.text.replace("'", "")
adress[columnIndexes[10]] = value_electricity_production.text.replace("'", "").replace(" Franken", "")
adress_file = open(filename_adresslist, "w")
new_line_string = ""
@ -75,20 +94,30 @@ def search_adresses(adress_list, filename_adresslist, driver):
adress_file.write(new_adress_list)
adress_file.close()
# 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("qrcodes/" + image_filename)
# Create Screenshot
featureElement1 = driver.find_element(By.XPATH,
"//section[@id='one']//div[@class='container']//div[@class='row 150%']")
featureElement2 = driver.find_element(By.XPATH,
"//section[@id='one']")
location = featureElement1.location
size1 = featureElement1.size
size2 = featureElement2.size
x = location["x"]
y = 5 # location["y"]
w = x + size2["width"]
h = y + size1["height"]
area = (x, y, w, h)
time.sleep(1)
driver.save_screenshot("screenshots/" + image_filename)
time.sleep(0.2)
Image.open("screenshots/" + image_filename).crop(area).save("screenshots/" + image_filename)
#featureElement = driver.find_element(By.XPATH, "// section[contains(string(),’START SCREENSHOT TESTING’)]")
#location = featureElement.location
#size = featureElement.size
driver.save_screenshot(image_filename)
#x = location["x"]
#y = location["y"]
#w = x + size["width"]
#h = y + size["height"]
#fullImg = Image.open(image_filename)
#cropImg = fullImg.crop(x, y, w, h)
#cropImg.save(image_filename)
# TODO Screenshot schneiden
print(image_filename + " was saved.")
@ -219,6 +248,7 @@ def command_exit():
root = tkinter.Tk()
root.protocol("WM_DELETE_WINDOW", command_exit)
root.wm_title("Sonnendach")
mainText = tkinter.Label(root, text=outputtext, width=80)
mainText.grid(row=1, column=1, padx=10, pady=3)