file-medium_sel_4.py

import base64
import requests

uri = "http://localhost:3333/session"

header = {
  "content-type": "application/json",
}

# Creates a new session
res = requests.post(uri, json={
  "capabilities": {
    "alwaysMatch": {
      "acceptInsecureCerts": True
    }
  }
}, headers=header)

# Stores Session ID
seddionId = res.json().get("value").get("sessionId")

# Browse to a URL
requests.post(f"{uri}/{seddionId}/url", json={
  "url": "https://www.google.com",
}, headers=header)

# Take a screenshot
res = requests.get(f"{uri}/{seddionId}/screenshot", headers=header)
image = res.json().get("value")

# Saves the image to a screenshot.png file
with open("screenshot.png", "wb") as fh:
  fh.write(base64.b64decode(image.encode("ASCII")))