def upload_asset(auth, config): """ Checks if a local asset has been uploaded before and if not, performs an upload operation on the single asset. :Args: - auth (:class:`.Credentials`): instance of the Credentials class as returned by authentication() - config (:class:`.Configuration`): instance of the Configuration class as returned by logging_mode() """ asset_mgr = FileManager(auth, cfg=config) asset_to_upload = asset_mgr.file_from_path(ASSET) if asset_to_upload.is_uploaded(): print("{0} already uploaded.".format(asset_to_upload)) return else: conf = "{0} MBs to be uploaded. Continue? (yes/no) ".format( len(asset_to_upload)/1024/1024) if input(conf)[0].lower() == 'y': try: asset_to_upload.upload() print("{0} uploaded successfully.".format(asset_to_upload)) except Exception as e: print("Upload failed: {0}".format(e)) else: print("Upload aborted.") return
def upload_assets(auth, config): """ Uploads a specified collection of assets either from a set created from scratch, or from the contents of existing directory. :Args: - auth (:class:`.Credentials`): instance of the Credentials class as returned by authentication() - config (:class:`.Configuration`): instance of the Configuration class as returned by logging_mode() """ asset_mgr = FileManager(auth, cfg=config) asset_to_add = asset_mgr.file_from_path(ASSET) # Creates FileCollection object asset_set = asset_mgr.create_file_set(asset_to_add) # Extends a FileCollection object with another FileCollection asset_set.extend(asset_mgr.files_from_dir(ASSET_DIR)) try: # force=true uploads the assets in the asset collection regardless of # whether they have been uploaded before. asset_set.upload(force=True) print("Assets uploaded successfully.") except RestCallException as e: print("failed: {0}".format(e))
def test_filemgr_file_from_path(self, mock_file, mock_api, mock_creds, mock_cfg): """Test file_from_path""" mgr = FileManager(mock_creds, cfg=mock_cfg) ufile = mgr.file_from_path("c:\\test.txt") mock_file.assert_called_with(mock.ANY, "c:\\test.txt") self.assertIsNotNone(ufile) ufile = mgr.file_from_path(None) mock_file.assert_called_with(mock.ANY, 'None') self.assertIsNotNone(ufile) ufile = mgr.file_from_path(42) mock_file.assert_called_with(mock.ANY, "42") self.assertIsNotNone(ufile)
def test_filemgr_file_from_path(self, mock_file, mock_api, mock_creds, mock_cfg): """Test file_from_path""" mgr = FileManager(mock_creds, cfg=mock_cfg) ufile = mgr.file_from_path("c:\\test.txt") mock_file.assert_called_with(mock.ANY, "c:\\test.txt") self.assertIsNotNone(ufile) ufile = mgr.file_from_path(None) mock_file.assert_called_with(mock.ANY, 'None') self.assertIsNotNone(ufile) ufile = mgr.file_from_path(42) mock_file.assert_called_with(mock.ANY, "42") self.assertIsNotNone(ufile)