copy_local_to_storage, both general and specialized-for-BasicFileStorage versions

This utility should allow for easy copying from a local filesystem to
the storage instance.
This commit is contained in:
Christopher Allan Webber 2011-11-20 22:01:43 -06:00
parent fe0a8f53e2
commit 98f6efb068
2 changed files with 22 additions and 0 deletions

View File

@ -169,6 +169,14 @@ class StorageInterface(object):
with file(dest_path, 'wb') as dest_file:
dest_file.write(source_file.read())
def copy_local_to_storage(self, filename, filepath):
"""
Copy this file from locally to the storage system.
"""
with self.get_file(filepath, 'wb') as dest_file:
with file(filename, 'rb') as source_file:
dest_file.write(source_file.read())
###########
# Utilities

View File

@ -20,6 +20,7 @@ from mediagoblin.storage import (
NoWebServing)
import os
import shutil
import urlparse
@ -76,3 +77,16 @@ class BasicFileStorage(StorageInterface):
def get_local_path(self, filepath):
return self._resolve_filepath(filepath)
def copy_local_to_storage(self, filename, filepath):
"""
Copy this file from locally to the storage system.
"""
# Make directories if necessary
if len(filepath) > 1:
directory = self._resolve_filepath(filepath[:-1])
if not os.path.exists(directory):
os.makedirs(directory)
shutil.copy(
filename, self.get_local_path(filepath))