wehub-resource-sync
39 lines
1.0 KiB
Python
Executable File
39 lines
1.0 KiB
Python
Executable File
#!/root/python3.11/bin/python3
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from argparse import ArgumentParser
|
|
from pathlib import Path
|
|
|
|
lib_path = str(Path(__file__).resolve().parent.parent / "lib")
|
|
sys.path.insert(0, lib_path)
|
|
|
|
from web_browser_config import ClientConfig
|
|
from web_browser_utils import (
|
|
ScreenshotMode,
|
|
_handle_screenshot,
|
|
_print_response_with_metadata,
|
|
send_request,
|
|
)
|
|
|
|
config = ClientConfig()
|
|
|
|
|
|
def screenshot():
|
|
"""Capture a screenshot and handle it according to the default config.screenshot_mode."""
|
|
response = send_request(config.port, "screenshot", "GET")
|
|
if response is None:
|
|
return
|
|
screenshot_data = response["screenshot"]
|
|
_print_response_with_metadata(response)
|
|
if config.screenshot_mode == ScreenshotMode.SAVE:
|
|
_handle_screenshot(screenshot_data, ScreenshotMode.SAVE)
|
|
else:
|
|
_handle_screenshot(screenshot_data, ScreenshotMode.PRINT)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = ArgumentParser()
|
|
args = parser.parse_args()
|
|
screenshot()
|