Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-08-07 22:12:15 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

experimenting with captureScreenshot

Browse source
This commit is contained in:
Noah Levitt 2015-06-16 18:42:21 -07:00
parent f254e2eec1
commit d8a962b29e
2 changed files with 32 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

12
bin/browse-url
View file

@ -6,6 +6,8 @@ import os
import sys
import logging
import umbra
import re
import datetime
arg_parser = argparse.ArgumentParser(prog=os.path.basename(__file__),
description='browse-url - open urls in chrome/chromium and run behaviors',
@ -24,7 +26,15 @@ args = arg_parser.parse_args(args=sys.argv[1:])
logging.basicConfig(stream=sys.stdout, level=args.log_level,
format='%(asctime)s %(process)d %(levelname)s %(threadName)s %(name)s.%(funcName)s(%(filename)s:%(lineno)d) %(message)s')
with umbra.Browser(chrome_exe=args.chrome_exe) as browser:
for url in args.urls:
browser.browse_page(url)
def on_screenshot(screenshot_png):
filename = "{}-{:%Y%m%d%H%M%S}.png".format(re.sub(r"\W", "_", url), datetime.datetime.now())
with open(filename, mode='wb') as png_out:
png_out.write(screenshot_png)
logging.info("wrote screenshot to {}".format(filename))
browser.browse_page(url, on_screenshot=on_screenshot)