import os
from seleniumbase.core import log_helper
from seleniumbase.fixtures import constants
VISUAL_BASELINE_DIR = constants.VisualBaseline.STORAGE_FOLDER
abs_path = os.path.abspath(".")
visual_baseline_path = os.path.join(abs_path, VISUAL_BASELINE_DIR)
def get_visual_baseline_folder():
return visual_baseline_path
def visual_baseline_folder_setup():
"""Handle Logging"""
if not os.path.exists(visual_baseline_path):
try:
os.makedirs(visual_baseline_path)
except Exception:
pass # Should only be reachable during multi-threaded runs
def get_sbs_head():
# Uses caching to prevent extra method calls
SIDE_BY_SIDE_PNG = constants.SideBySide.get_favicon()
head = (
''
''
''
"Visual Comparison"
"" % (SIDE_BY_SIDE_PNG)
)
return head
def get_sbs_table_row(baseline="baseline.png", diff="baseline_diff.png"):
row = (
''
''
' | '
' | '
"
"
"" % (baseline, diff)
)
return row
def get_sbs_table_html(baseline="baseline.png", diff="baseline_diff.png"):
table_html = (
''
''
""
'| Baseline Screenshot | '
'Visual Diff Failure Screenshot | '
"
"
)
row = get_sbs_table_row(baseline, diff)
table_html += row
table_html += "
"
return table_html
def get_sbs_gen_by():
gen_by = (
'"
)
return gen_by
def get_sbs_header_text():
header_text = "SeleniumBase Visual Comparison"
return header_text
def get_sbs_header():
header_text = get_sbs_header_text()
header = '%s
' % header_text
return header
def get_sbs_footer():
footer = "
Last updated: "
timestamp, the_date, the_time = log_helper.get_master_time()
last_updated = "%s at %s" % (the_date, the_time)
footer = footer + "%s" % last_updated
gen_by = get_sbs_gen_by()
footer = footer + gen_by
return footer
def get_sbs_html(baseline="baseline.png", diff="baseline_diff.png"):
head = get_sbs_head()
header = get_sbs_header()
table_html = get_sbs_table_html(baseline, diff)
footer = get_sbs_footer()
the_html = (
''
+ head
+ ''
+ header
+ table_html
+ footer
+ ""
)
return the_html