[Selenium] 현재 화면 Capture

import inspect
import os
import json
from selenium import webdriver

#Console log 저장 파일 이름을 해당 함수를  호출한 파일의 이름으로 생성하기 위함
Path=os.path.basename(inspect.getfile(inspect.currentframe()))
#파일 중 확장자를 제거한 이름을 확보하기 위함
filename=Path.split('.')

def capture(driver, filename):
    screenshot_name = filename[0] + '.png'
    driver.save_screenshot(screenshot_name)