Merge lp:~elopio/selenium-simple-test/download into lp:selenium-simple-test

[Leo Arias]

Added the download function.

Added the download function.
This branch lacks the self test.

[Vincent Ladeuil (vila)]

> This branch lacks the self test.

Or several.

Just reading quickly:

68 + if 'Content-Dispoisition' in info

'Content-Disposition' would have been caught by tests ;)

But more generally, sst is about delegating that kind of stuff to the browser (there are *many* weird cases you don't want to even try to support that are already properly handled there).

You can even write tests using 'file:///' urls

review: Needs Fixing

[Vincent Ladeuil (vila)]

@Leo: ping

Are you still working on this proposal (mark it wip) or not (mark it rejected) ?

;)

review: Needs Information

[Leo Arias (elopio)]

Marked as work in progress.
It's currently not needed, but it will be in the future. However, it should probably be moved to the test that's requiring it, not necessarily on SST.

Unmerged revisions

291. By Leo Arias on 2012-08-09

Typo.

290. By Leo Arias on 2012-08-09

Typo.

289. By Leo Arias on 2012-08-09

Added the name from the content disposition.

288. By Leo Arias on 2012-08-09

Merged with trunk.

287. By Leo Arias on 2012-08-02

Cleaned the download function a little.

286. By Leo Arias on 2012-07-31

Typo.

285. By Leo Arias on 2012-07-31

Fixed the call to urlibopen.

284. By Leo Arias on 2012-07-31

Removed the print statements.

283. By Leo Arias on 2012-07-31

Fixed the get_cookies call.

282. By Leo Arias on 2012-07-31

Fixed typo.

=== modified file 'src/sst/actions.py'
--- src/sst/actions.py	2012-07-17 13:43:15 +0000
+++ src/sst/actions.py	2012-08-09 17:35:20 +0000
@@ -41,6 +41,7 @@
 import os
 import re
 import time
+import urllib2
 
 from datetime import datetime
 from pdb import set_trace as debug
@@ -68,10 +69,10 @@
     'assert_table_headers', 'assert_table_row_contains_text', 'assert_text',
     'assert_text_contains', 'assert_textfield', 'assert_title',
     'assert_title_contains', 'assert_url', 'assert_url_contains', 'check_flags',
-    'click_link', 'close_window', 'debug', 'dismiss_alert', 'end_test',
-    'click_button', 'click_element', 'exists_element', 'fails', 'get_argument',
-    'get_base_url', 'get_cookies', 'get_current_url', 'get_element',
-    'get_element_by_css', 'get_element_by_xpath', 'get_elements',
+    'click_link', 'close_window', 'debug', 'dismiss_alert', 'download_file', 
+    'end_test', 'click_button', 'click_element', 'exists_element', 'fails', 
+    'get_argument', 'get_base_url', 'get_cookies', 'get_current_url', 
+    'get_element', 'get_element_by_css', 'get_element_by_xpath', 'get_elements',
     'get_elements_by_css', 'get_elements_by_xpath', 'get_link_url',
     'get_page_source', 'go_back', 'go_to', 'refresh', 'reset_base_url',
     'run_test', 'set_base_url', 'set_checkbox_value', 'set_dropdown_value',
@@ -1395,3 +1396,45 @@
 def get_cookies():
     """Gets the cookies of current session (set of dicts)."""
     return browser.get_cookies()
+
+
+def download_file(url, download_path):
+    """
+    Download the file stored in the URL specified and save it
+    to the download path."""
+    session_id = _get_session_id_cookie()
+    headers = _make_cookie_header_with_session_id(session_id)
+    web_file, file_name = _get_file_from_url(url, headers)
+    file_path = os.path.join(download_path, file_name)
+    with open(file_path, "wb") as local_file:
+        local_file.write(web_file.read())
+
+
+def _get_session_id_cookie():
+    cookies = get_cookies()
+    session = cookies[0]
+    return session['value']
+
+
+def _make_cookie_header_with_session_id(session_id):
+    session_header = 'sessionid=' + session_id
+    return {'Cookie': session_header}
+
+
+def _get_file_from_url(url, headers):
+    request = urllib2.Request(url, None, headers)
+    web_file = urllib2.urlopen(request)
+    file_name = _get_suggested_file_name(request)
+    if not file_name:
+        file_name = os.path.basename(url)
+    return web_file, file_name
+    
+
+def _get_suggested_file_name(request):
+    info = request.info()
+    file_name = None
+    if 'Content-Dispoisition' in info
+        content_disposition = info['Content-Disposition']
+        file_name = content_disposition.split('filename=')[1]
+        file_name = file_name.strip('"\'')
+    return file_name