Webscreenshot – 截取网站列表的简单脚本

Webscreenshot 是一个简单易用的 Python 脚本，旨在为用户提供一种方便的方式来截取网站截图，同时生成网站列表。

功能特色

• 快速而准确地捕捉屏幕上的网站截图
• 自动生成网站列表文件，以便用户了解他们的目标

如何使用 Webscreenshot

1. 首先，您需要下载 Webscreenshot 的代码并在您的系统上安装它；
2. 在命令行中指定要截取的网站列表文件和输出文件夹；
3. 运行脚本，等待程序完成。

python3 webscreenshot.py -i /path/to/input/file -o /path/to/output/directory

代码示例

#!/usr/bin/env python3
import argparse
import os

def get_args():
    parser = argparse.ArgumentParser(description='Take screenshots of websites and generate a list.')
    parser.add_argument('-i', '--input', type=str, required=True,
                        help='Path to input file containing list of URLs.')
    parser.add_argument('-o', '--output', type=str, required=True,
                        help='Path to output directory where screenshots will be saved.')
    return parser.parse_args()

def run_command(cmd):
    print(f'[INFO] Running: {cmd}')
    os.system(cmd)

def main():
    args = get_args()

    with open(args.input, 'r') as f:
        urls = f.readlines()

    for url in urls:
        url = url.strip()
        filename = f'{url.replace("http://", "").replace("https://", "")}.png'
        output_file = os.path.join(args.output, filename)
        cmd = f'google-chrome --headless --disable-gpu --screenshot={output_file} --window-size=1280,720 {url}'
        run_command(cmd)

    print(f'[INFO] Screenshots saved to {args.output} directory.')
    with open(os.path.join(args.output, 'list.txt'), 'w') as f:
        f.write('\n'.join(urls))

if __name__ == '__main__':
    main()

总结

Webscreenshot 是一个易于使用的 Python 脚本，旨在为用户提供方便的方式来截取网站截图，并生成网站列表文件。该脚本具有快速而准确地捕捉屏幕上的网站截图和自动生成网站列表文件的特点。用户只需使用命令行指定要截取的网站列表文件和输出文件夹即可在控制台中运行脚本。