download from an url

jeroenmeulenaar · Jan 31, 2013 · 73cbc29 · 73cbc29
1 parent 0dec814
commit 73cbc29
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -46,6 +46,8 @@ files = m.get_files()
 m.download_file(file_id, file_key)
 # upload a file
 m.uploadfile('/home/kim/mega/secret_plans')
+# download from an url
+m.download_from_url('https://mega.co.nz/#!wYo3AYZC!Zwi1f3ANtYwKNOc07fwuN1enOoRj4CreFouuGqi4D6Y')
 ```
 
 ## Running the tests

diff --git a/mega/mega.py b/mega/mega.py
@@ -5,6 +5,7 @@
 import binascii
 
 import requests
+from urlobject import URLObject
 from Crypto.Cipher import AES
 from Crypto.PublicKey import RSA
 from Crypto.Util import Counter
@@ -115,6 +116,11 @@ def get_files(self):
                 self.trashbin_id = file['h']
         return files_data
 
+    def download_from_url(self, url):
+        url_object = URLObject(url)
+        file_id, file_key = url_object.fragment[1:].split('!')
+        self.download_file(file_id, file_key, public=True)
+
     def download_file(self, file_id, file_key, public=False):
         if public:
             file_key = base64_to_a32(file_key)

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,3 @@
 requests>=1.1.0
+URLObject>=2.1.1
 pycrypto>=2.6