Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sam4u3 merged 1 commit into from
Jul 18, 2022
Merged

Fix: Fixed issue when taking screenshots and images are displayed cro… #18

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
61 changes: 50 additions & 11 deletions Screenshot/Screenshot_Clipping.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
"""
pass

@staticmethod
# Take temporary screenshot of the web page to get the size of the image
def __get_screen_size(driver: WebDriver) -> dict:
driver.get_screenshot_as_file('screenshot.png')
image = Image.open('screenshot.png')
width, height = image.size

return {'width': width, 'height': height}

def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
"""
Take full screenshot of web page
Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
elements: List of Xpath of elements to hide from web pages
is_load_at_runtime: Page Load at runtime
load_wait_time: The Wait time while loading full screen
multi_images: The flag is uses to capture multiple images and create single image in vertical format

Returns:
str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
time.sleep(2)
rectangles = []


i = 0
while i < total_height:
ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
rectangles.append((ii, i, top_width, top_height))
ii = ii + viewport_width
i = i + viewport_height
stitched_image = Image.new('RGB', (total_width, total_height))
stitched_image = None
previous = None
part = 0

# This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
init_canvas = True
# Get the screenshot and save the dimensions of the image, this will be useful for create a
# correct canvas with correct dimensions
# and not show images with a wrong size or cut
size_screenshot = self.__get_screen_size(driver)
# Get temporary sum of height of the total pages
height_canvas = size_screenshot['height'] * len(rectangles)

# Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
# Or assign the height of the total screenshot
if multi_images:
# Assign the height of the total screenshot at canvas even if not used at the moment
stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
else:
# Assign the dimensions corresponding to the size of the uniq screenshot
stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

# This constant is used top offset the image in the canvas
jump = round(size_screenshot['height'] / 2)
# With take multiple screenshots this value is used to adjust the position of the image in the canvas
# This value know update every time the screenshot is taken
top_offset = jump

for rectangle in rectangles:
if not previous is None:
if previous is not None:
driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
time.sleep(3)
time.sleep(1)
self.hide_elements(driver, elements)

file_name = "part_{0}.png".format(part)
driver.get_screenshot_as_file(file_name)
screenshot = Image.open(file_name)

if rectangle[1] + viewport_height > total_height:
offset = (rectangle[0], total_height - viewport_height)
offset = (rectangle[0], rectangle[1] + top_offset)
else:
offset = (rectangle[0], rectangle[1])
if init_canvas:
offset = (rectangle[0], rectangle[1])
init_canvas = False
else:
offset = (rectangle[0], rectangle[1] + top_offset)
top_offset = jump + top_offset

stitched_image.paste(screenshot, offset)
del screenshot
os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
stitched_image.save(save_path)
return save_path

def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
"""
Usage:
Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
Raises:
N/A
"""
image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
location = element.location
size = element.size
x = location['x']
Expand Down
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all 
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix");
}
} catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms");
}
} catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links");
}
} catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); }
})();
(function(){
try {
var __m = "youtube.com";
var __re = new RegExp('^' + "youtube\\.com" + '

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default");
}
} catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers");
}
} catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode");
}
} catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); }
})();
})();

Skip to content
Fix: Fixed issue when taking screenshots and images are displayed cro…#18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

 Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and
privacy statement. We’ll occasionally send you account related emails.

Already on GitHub?
Sign in
to your account

 Merged

sam4u3
merged 1 commit into

from

Jul 18, 2022
 Merged


Fix: Fixed issue when taking screenshots and images are displayed cro…
#18
 Changes from all commits
Commits

Show all changes


1 commit


Select commit


File filter

Filter by extension
Filter by extension
Conversations

Failed to load comments. 
Loading
Jump to

Jump to file


Failed to load files. 
Loading
 Diff view

 Diff view


61 changes: 50 additions & 11 deletions
Screenshot/Screenshot_Clipping.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
import os
import time
import uuid

from PIL import Image
from selenium import webdriver
Expand DownExpand Up@@ -31,8 +30,17 @@ def __init__(self):
 """
 pass

 @staticmethod
 # Take temporary screenshot of the web page to get the size of the image
 def __get_screen_size(driver: WebDriver) -> dict:
 driver.get_screenshot_as_file('screenshot.png')
 image = Image.open('screenshot.png')
 width, height = image.size

 return {'width': width, 'height': height}

 def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: str = 'selenium_full_screenshot.png',
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5) -> str:
 elements: list = None, is_load_at_runtime:bool = False,load_wait_time: int = 5, multi_images: bool = False) -> str:
 """
 Take full screenshot of web page
 Args:
Expand All@@ -42,6 +50,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 elements: List of Xpath of elements to hide from web pages
 is_load_at_runtime: Page Load at runtime
 load_wait_time: The Wait time while loading full screen
 multi_images: The flag is uses to capture multiple images and create single image in vertical format

 Returns:
 str : The path of image
Expand DownExpand Up@@ -78,7 +87,6 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 time.sleep(2)
 rectangles = []


 i = 0
 while i < total_height:
 ii = 0
Expand All@@ -92,23 +100,54 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 rectangles.append((ii, i, top_width, top_height))
 ii = ii + viewport_width
 i = i + viewport_height
 stitched_image = Image.new('RGB', (total_width, total_height))
 stitched_image = None
 previous = None
 part = 0

 # This value indicates if is the first part of the screenshot, or it is the rest of the screenshot
 init_canvas = True
 # Get the screenshot and save the dimensions of the image, this will be useful for create a
 # correct canvas with correct dimensions
 # and not show images with a wrong size or cut
 size_screenshot = self.__get_screen_size(driver)
 # Get temporary sum of height of the total pages
 height_canvas = size_screenshot['height'] * len(rectangles)

 # Compare if the screenshot it's only one part and assign the correct dimensions of the screenshot size
 # Or assign the height of the total screenshot
 if multi_images:
 # Assign the height of the total screenshot at canvas even if not used at the moment
 stitched_image = Image.new('RGB', (size_screenshot['width'], height_canvas))
 else:
 # Assign the dimensions corresponding to the size of the uniq screenshot
 stitched_image = Image.new('RGB', (size_screenshot['width'], size_screenshot['height']))

 # This constant is used top offset the image in the canvas
 jump = round(size_screenshot['height'] / 2)
 # With take multiple screenshots this value is used to adjust the position of the image in the canvas
 # This value know update every time the screenshot is taken
 top_offset = jump

 for rectangle in rectangles:
 if not previous is None:
 if previous is not None:
 driver.execute_script("window.scrollTo({0}, {1})".format(rectangle[0], rectangle[1]))
 time.sleep(3)
 time.sleep(1)
 self.hide_elements(driver, elements)

 file_name = "part_{0}.png".format(part)
 driver.get_screenshot_as_file(file_name)
 screenshot = Image.open(file_name)

 if rectangle[1] + viewport_height > total_height:
 offset = (rectangle[0], total_height - viewport_height)
 offset = (rectangle[0], rectangle[1] + top_offset)
 else:
 offset = (rectangle[0], rectangle[1])
 if init_canvas:
 offset = (rectangle[0], rectangle[1])
 init_canvas = False
 else:
 offset = (rectangle[0], rectangle[1] + top_offset)
 top_offset = jump + top_offset

 stitched_image.paste(screenshot, offset)
 del screenshot
 os.remove(file_name)
Expand All@@ -118,7 +157,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
 stitched_image.save(save_path)
 return save_path

 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 def get_element(self, driver: WebDriver, element: WebElement, save_location: str, image_name: str = 'cropped_screenshot.png', to_hide: list = None) -> str:
 """
 Usage:
 Capture Element screenshot as a image
Expand All@@ -131,7 +170,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
 Raises:
 N/A
 """
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
 image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide, multi_images=True)
 location = element.location
 size = element.size
 x = location['x']
Expand Down