It's quite simple, plz try this:
from selenium import webdriverdriver = webdriver.Chrome()def take_screenshot(name): driver.get_screenshot_as_file(f"./screenshot/{name}.png")#Or with datetime:def take_screenshot_with_dt(name): datetime.now() screenDatetime = datetime.now().strftime('%d-%m-%Y-%H-%M') driver.get_screenshot_as_file(f"./screenshot/{name}-{screenDatetime}.png")take_screenshot("screenshotName")take_screenshot_with_dt("screenshotNameDt")