added pydrivebackend.py

pydrivebackend.py

@@ -0,0 +1,107 @@
+# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4 -*-
+#
+# Copyright 2015 Yigal Asnis 
+#
+# This file is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# It is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with duplicity; if not, write to the Free Software Foundation,
+# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+import string
+import duplicity.backend
+from duplicity.errors import BackendException
+
+class PyDriveBackend(duplicity.backend.Backend):
+    """Connect to remote store using PyDrive API"""
+
+    def __init__(self, parsed_url):
+        duplicity.backend.Backend.__init__(self, parsed_url)
+        try:
+            global pydrive
+            import httplib2
+            from apiclient.discovery import build
+            from oauth2client.client import SignedJwtAssertionCredentials
+            from pydrive.auth import GoogleAuth
+            from pydrive.drive import GoogleDrive
+        except ImportError:
+            raise BackendException('PyDrive backend requires PyDrive installation'
+                                   'python pip install PyDrive')
+
+        try: 
+            keyFilePath = parsed_url.keyfile
+            keyfile = open(keyFilePath[0], "r+")
+            key = keyfile.read();
+            keyfile.close()
+        except:
+            raise BackendException("Can't find/open/read the keyfile. Please read the README for configuration info.") 
+        credentials = SignedJwtAssertionCredentials(parsed_url.username + '@' + parsed_url.hostname, key, scope='https://www.googleapis.com/auth/drive')
+        credentials.authorize(httplib2.Http())
+        gauth = GoogleAuth()
+        gauth.credentials = credentials
+        self.drive = GoogleDrive(gauth)
+
+        # Dirty way to find root folder id
+        file_list = self.drive.ListFile({'q': "'Root' in parents"}).GetList()
+        if file_list:
+            parent_folder_id = file_list[0]['parents'][0]['id']
+        else:
+            file_in_root = self.drive.CreateFile({'title': 'i_am_in_root'})
+            file_in_root.Upload()
+            parent_folder_id = file_in_root['parents'][0]['id']
+
+        # Fetch destination folder entry (and create hierarchy if required).
+        folder_names = string.split(parsed_url.path, '/')
+        for folder_name in folder_names:
+            if not folder_name:
+                continue
+            file_list = self.drive.ListFile({'q': "'" + parent_folder_id + "' in parents"}).GetList()
+            folder =  next((item for item in file_list if item['title'] == folder_name and item['mimeType'] == 'application/vnd.google-apps.folder'), None)
+            if folder is None:
+                folder = self.drive.CreateFile({'title': folder_name, 'mimeType': "application/vnd.google-apps.folder", 'parents': [{'id': parent_folder_id}]})
+                folder.Upload()
+            parent_folder_id = folder['id']
+        self.folder = parent_folder_id
+
+    def FilesList(self):
+        return self.drive.ListFile({'q': "'" + self.folder + "' in parents"}).GetList()
+
+    def id_by_name(self, filename):
+        try:
+            return next(item for item in self.FilesList() if item['title'] == filename)['id']
+        except:
+            return ''
+
+    def _put(self, source_path, remote_filename):
+        drive_file = self.drive.CreateFile({'title': remote_filename, 'parents': [{"kind": "drive#fileLink", "id": self.folder}]})
+        drive_file.SetContentFile(source_path.name)
+        drive_file.Upload()
+
+    def _get(self, remote_filename, local_path):
+        drive_file = self.drive.CreateFile({'id': self.id_by_name(remote_filename)})
+        drive_file.GetContentFile(local_path.name)
+
+    def _list(self):
+        return [item['title'] for item in self.FilesList()]
+
+    def _delete(self, filename):
+        file_id = self.id_by_name(filename)
+        drive_file = self.drive.CreateFile({'id': file_id})
+        drive_file.auth.service.files().delete(fileId=drive_file['id']).execute()
+
+    def _query(self, filename):
+        try:
+            size = int((item for item in self.FilesList() if item['title'] == filename).next()['fileSize'])
+        except:
+            size = -1
+        return {'size': size}
+
+duplicity.backend.register_backend('pydrive', PyDriveBackend)