def enable_download_in_headless_chrome(driver, download_dir):
  #add missing support for chrome "send_command"  to selenium webdriver
  driver.command_executor._commands["send_command"] = ("POST", '/session/$sessionId/chromium/send_command')

  params = {'cmd': 'Page.setDownloadBehavior', 'params': {'behavior': 'allow', 'downloadPath': download_dir}}
  driver.execute("send_command", params)

enable_download_in_headless_chrome(driver, os.getcwd())