Add save_data_as() method

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -1629,6 +1629,14 @@ def save_file_as(self, file_url, new_file_name, destination_folder=None):
         page_utils._download_file_to(
             file_url, destination_folder, new_file_name)
 
+    def save_data_as(self, data, file_name, destination_folder=None):
+        """ Saves the data specified to a file of the name specified.
+            If no destination folder is specified, the default one is used.
+            (The default downloads folder = "./downloaded_files") """
+        if not destination_folder:
+            destination_folder = constants.Files.DOWNLOADS_FOLDER
+        page_utils._save_data_as(data, destination_folder, file_name)
+
     def get_downloads_folder(self):
         """ Returns the OS path of the Downloads Folder.
             (Works with Chrome and Firefox only, for now.) """

seleniumbase/fixtures/page_utils.py

@@ -1,6 +1,7 @@
 """
 This module contains useful utility methods.
 """
+import codecs
 import re
 import requests
 
@@ -76,6 +77,12 @@ def _download_file_to(file_url, destination_folder, new_file_name=None):
         code.write(r.content)
 
 
+def _save_data_as(data, destination_folder, file_name):
+    out_file = codecs.open(destination_folder + '/' + file_name, "w+")
+    out_file.writelines(data)
+    out_file.close()
+
+
 def _jq_format(code):
     """
     DEPRECATED - Use re.escape() instead, which performs the intended action.