- 109
- 60
Кто может помочь или обьяснить почему не работает, скрипт не мой, взял на гитхабе, но ничего не работает. Можете кидаться тапками только помогите
Анализ:
from PIL import Image, ImageDraw
from selenium import webdriver
import os
import sys
class ScreenAnalysis:
STAGING_URL = 'http://www.yahoo.com' #эталон страницы
PRODUCTION_URL = 'http://www.yahoo.com' #тестируемая
driver = None
def __init__(self): #вызов функций
self.set_up()
self.capture_screens()
self.analyze()
self.clean_up()
def set_up(self):
self.driver = None
def clean_up(self): #функция закрытия
self.driver.close()
def capture_screens(self): #название скрина
self.screenshot(self.STAGING_URL, 'screen_staging.png')
self.screenshot(self.PRODUCTION_URL, 'screen_production.png')
def screenshot(self, url, file_name): #размер окна и сохранение
print( "Capturing", url, "screenshot as", file_name, "...")
self.driver.get(url)
self.driver.set_window_size(1920, 1080)
self.driver.save_screenshot(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'screenshots', file_name))
self.driver.get_screenshot_as_png()
print("Done.")
def analyze(self): # сравнение скришотов и разбиение на квадраты
screenshot_staging = Image.open("screenshots/screen_staging.png")
screenshot_production = Image.open("screenshots/screen_production.png")
columns = 60
rows = 80
screen_width, screen_height = screenshot_staging.size
block_width = ((screen_width - 1) // columns) + 1 # this is just a division ceiling
block_height = ((screen_height - 1) // rows) + 1
for y in range(0, screen_height, block_height+1):
for x in range(0, screen_width, block_width+1):
region_staging = self.process_region(screenshot_staging, x, y, block_width, block_height)
region_production = self.process_region(screenshot_production, x, y, block_width, block_height)
if region_staging is not None and region_production is not None and region_production != region_staging:
draw = ImageDraw.Draw(screenshot_staging)
draw.rectangle((x, y, x+block_width, y+block_height), outline = "red")
screenshot_staging.save("result.png")
def process_region(self, image, x, y, width, height): # чувствительность при сравнении
region_total = 0
# This can be used as the sensitivity factor, the larger it is the less sensitive the comparison
factor = 100
for coordinateY in range(y, y+height):
for coordinateX in range(x, x+width):
try:
pixel = image.getpixel((coordinateX, coordinateY))
region_total += sum(pixel)/4
except:
return
return region_total/factor
ScreenAnalysis()
