Merge pull request #33 from Frameio/jh/add-download

Add Download function

Author: jhodges10

frameioclient/client.py

@@ -2,6 +2,7 @@
 from .upload import FrameioUploader
 from requests.adapters import HTTPAdapter
 from requests.packages.urllib3.util.retry import Retry
+from .download import FrameioDownloader
 
 class PaginatedResponse(object):
   def __init__(self, results=[], page=0, page_size=0, total=0, total_pages=0):
@@ -232,11 +233,26 @@ def upload(self, asset, file):
 
       Example::
 
-        client.upload(asset, open('example.mp4'))
+        client.upload(asset, open('example.mp4')) // TODO fix this example (bad way of opening file)
     """
     uploader = FrameioUploader(asset, file)
     uploader.upload()
-    
+  
+  def download(self, asset, download_folder):
+    """
+    Download an asset. The method will exit once the file is downloaded.
+
+    :Args:
+      asset (object): The asset object.
+      download_folder (path): The location to download the file to.
+
+      Example::
+
+        client.download(asset, "~./Downloads")
+    """
+    downloader = FrameioDownloader(asset, download_folder)
+    downloader.download()
+
   def get_comment(self, comment_id, **kwargs):
     """
     Get a comment.

frameioclient/download.py

@@ -0,0 +1,20 @@
+import requests
+import math
+import os
+
+class FrameioDownloader(object):
+  def __init__(self, asset, download_folder):
+    self.asset = asset
+    self.download_folder = download_folder
+
+  def download(self):
+    # TODO add test to chere here to make sure the URL we download is coming from our S3 buckets
+    # Grab the filename from the passed dictionary and store it.
+    original_filename = self.asset['name']
+    final_destination = os.path.join(self.download_folder, original_filename)
+    
+    # Grab the origial_url from the passed dictionary, then download the file
+    url = self.asset['original']
+    r = requests.get(url)
+    open(final_destination, 'wb').write(r.content)
+