Upload files with Selenium (Windows version)

In case when you have basic HTML input for file it’s very easy, just run .find_element_by_id('id_photo').send_keys(<full path to file>). But very often we don’t have such inputs or they are hidden and we cann’t send keys to them. For such cases with using Windows we can do it with help of pywinauto or other automation GUI utilities.

So, if you have open file dialog and run test on windows, I have solution. With help of pywinauto we can put path to file into dialog and click "Open". To connect pywinauto to browser we should have process id, for Firefox we already have right id in driver instance but if you use Chrome, driver has only pid of browser but you need pid of tab, in my example I have only one tab and in code I just get first child of browser. Also we should do few tries to fill out dialog or run some sleep to wait for dialog will be opened.

Next you can found simple example how to upload file.

#!/usr/bin/python
import psutil
from unittest import TestCase
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from pywinauto.application import Application


class UploadTestCase(TestCase):
    def setUp(self):
        self.browser = webdriver.Chrome()
        self.wait = WebDriverWait(self.browser, 10)

    def tearDown(self):
        self.browser.close()

    def get_pid(self):
        if isinstance(self.browser, webdriver.Chrome):
            process = psutil.Process(self.browser.service.process.pid)
            return process.children()[0].pid              # process id of browser tab
        if isinstance(self.browser, webdriver.Firefox):
            return self.browser.service.process.pid       # process id of browser

        assert Exception('driver does not supported')

    def upload_file(self, filename):
        path = os.path.join(
            os.path.realpath('.'),
            'fixtures',
            filename
        )
        assert os.path.exists(path)

        for i in range(10):
            app = Application()
            app.connect(process=self.get_pid())  # connect to browser
            dialog = app.top_window_()           # get active top window (Open dialog)
            if not dialog.Edit.Exists():         # check if Edit field is exists
                time.sleep(1)                    # if no do again in 1 second (waiting for dialog after click)
                continue
            dialog.Edit.TypeKeys('"{}"'.format(path))   # put file path
            dialog['&OpenButton'].Click()               # click Open button

            return

        raise Exception('"Open File" dialog not found')

    def test_upload_small_logo(self):
        el_logo_upload = self.wait.until(
            EC.presence_of_element_located(
                (By.NAME, ‘upload-logo')
            )
        )

        el_logo_upload.click()
        self.upload_file('logo_small.png')

        # checking

2016-08-08 10:00:20 test selenium

Posts

2018 (2)
- September (1)
  - How to build your own blockchain for a financial product
- January (1)
  - Debugging Python Applications with pdb
2017 (2)
- April (1)
  - How to Improve Django Performance. Optimization Tips
- March (1)
  - Django-Classifier or What Have I Done?
2016 (5)
- December (1)
  - How to speak from Arduino to Python and back
- October (1)
  - Office information gadget mounted on toy
- September (1)
  - Django package: django-classifier
- August (2)
  - Upload files with Selenium (Windows version)
  - How to change choices for field in Django-Rest-Framework
2015 (2)
- July (1)
  - How I take photos of water crown
- January (1)
  - Waiting for finishing asynchronous calls in AngularJS
2014 (2)
- October (1)
  - How to use PostgreSQL partitions with Django
- September (1)
  - Sony Action Cam firmware update error on Mac OS X

Upload files with Selenium (Windows version)

Tags

Posts

See also