def move_document_to_user_folder(resource_id, user): logger = LoggingHelper.getDebugLogger() client = GoogleDriveHelper.get_docs_client() dest_folder = user.get_profile().drive_folder_id if not dest_folder: logger.debug('Creating folder for {0}'.format(user.username)) dest_folder = GoogleDriveHelper.create_unistar_folder(user) user.get_profile().drive_folder_id = dest_folder user.get_profile().save() logger.debug('Moving {0} to {1}'.format(resource_id, GoogleDriveHelper._get_unistar_collection_name(user.username))) client.files().update(fileId=resource_id, body={"parents":[{'id':dest_folder}]}).execute()
def test_create_unistar_folder(self): response = {'id': 'abcdef'} requestBuilder = RequestMockBuilder({ 'drive.files.insert': (None, json.dumps(response)), }) GoogleDriveHelper.get_docs_client = Mock() GoogleDriveHelper.get_docs_client.return_value = apiclient.discovery.build( "drive", "v2", requestBuilder=requestBuilder) user = User.objects.create_user('arnaud.breton', '*****@*****.**') self.assertEqual(GoogleDriveHelper.create_unistar_folder(user), 'abcdef')
def test_create_unistar_folder(self): response = { 'id' : 'abcdef' } requestBuilder = RequestMockBuilder( { 'drive.files.insert': (None, json.dumps(response)), } ) GoogleDriveHelper.get_docs_client = Mock() GoogleDriveHelper.get_docs_client.return_value = apiclient.discovery.build("drive", "v2", requestBuilder=requestBuilder) user = User.objects.create_user('arnaud.breton','*****@*****.**') self.assertEqual(GoogleDriveHelper.create_unistar_folder(user), 'abcdef')
def move_document_to_user_folder(resource_id, user): logger = LoggingHelper.getDebugLogger() client = GoogleDriveHelper.get_docs_client() dest_folder = user.get_profile().drive_folder_id if not dest_folder: logger.debug('Creating folder for {0}'.format(user.username)) dest_folder = GoogleDriveHelper.create_unistar_folder(user) user.get_profile().drive_folder_id = dest_folder user.get_profile().save() logger.debug('Moving {0} to {1}'.format( resource_id, GoogleDriveHelper._get_unistar_collection_name(user.username))) client.files().update(fileId=resource_id, body={ "parents": [{ 'id': dest_folder }] }).execute()