Support for upload to FileDropper in ffx-backup.py

ffx-backup.py

@@ -14,8 +14,8 @@ EXCLUDES = [
 
 import time, os, os.path, re, sys, tarfile
 
-if len(sys.argv) != 2:
-    print "Usage: %s path_to_profile" % sys.argv[0]
+if len(sys.argv) != 2 and len(sys.argv) != 4:
+    print "Usage: %s path_to_profile [filedropper_username filedropper_password]" % sys.argv[0]
     sys.exit(1)
 
 profile_path = sys.argv[1]
@@ -23,6 +23,11 @@ profile_name = os.path.basename(profile_path)
 date = time.strftime("%Y%m%d_%H%M%S")
 archive_name = "%s-%s.tar.bz2" % (profile_name, date)
 
+fd_user = None
+fd_pass = None
+if len(sys.argv) == 4:
+    fd_user, fd_pass = sys.argv[2], sys.argv[3]
+
 # Compile exclusion regexps
 EXCLUDES_RE = []
 for regexp in EXCLUDES:
@@ -52,3 +57,32 @@ tar.close()
 # Display archive name and sizes
 print "Profile saved to %s" % archive_name
 print "%.1f MB --> %.1f MB" % (total_size/(1024**2), os.path.getsize(archive_name)/(1024**2))
+
+# Upload to FileDropper if needed
+if fd_user is not None and fd_pass is not None:
+    import FileDropper
+
+    fd = FileDropper.FileDropper()
+    fd.login(fd_user, fd_pass)
+
+    # Upload the file
+    url = fd.upload(archive_name)
+
+    # Get the file ID
+    lst = fd.list()
+    file_id = -1
+    for line in lst:
+        if line[6] == url:
+            file_id = line[1]
+            break
+    if file_id == -1:
+        # File not found in list!
+        raise Exception("Error in upload, could not get file ID, permissions not set")
+
+    # Set the permissions
+    fd.set_perm(file_id, FileDropper.FD_PERM_PRIVATE)
+
+    # Logout
+    fd.logout()
+
+    print "File uploaded to FileDropper: %s" % url