def __init__(self, path):
self.store = os.path.join(path, 'objects')
if not os.path.isdir(self.store):
raise InvalidStore("objects is not a directory")
db = os.path.join(path, 'database')
if not os.path.isfile(db):
raise InvalidStore("database is not a file")
self.metadata = MetadataStore(db)
def create_store(path):
if os.path.exists(path):
if not os.path.isdir(path) or os.listdir(path):
raise CreationError("Path is not a directory or is not empty")
exists = True
else:
exists = False
try:
if not exists:
os.mkdir(path)
os.mkdir(os.path.join(path, 'objects'))
except OSError as e: # pragma: no cover
raise CreationError("Could not create directories: %s: %s" % (
e.__class__.__name__, e.message))
MetadataStore.create_db(os.path.join(path, 'database'))
def create_store(path):
if os.path.exists(path):
if not os.path.isdir(path) or os.listdir(path):
raise CreationError("Path is not a directory or is not empty")
exists = True
else:
exists = False
try:
if not exists:
os.mkdir(path)
os.mkdir(os.path.join(path, 'objects'))
except OSError as e: # pragma: no cover
raise CreationError("Could not create directories: %s: %s" %
(e.__class__.__name__, e.message))
MetadataStore.create_db(os.path.join(path, 'database'))
def __init__(self, path):
self.store = os.path.join(path, 'objects')
if not os.path.isdir(self.store):
raise InvalidStore("objects is not a directory")
db = os.path.join(path, 'database')
if not os.path.isfile(db):
raise InvalidStore("database is not a file")
self.metadata = MetadataStore(db)
class FileStore(object):
"""Represents a file store.
"""
def __init__(self, path):
self.store = os.path.join(path, 'objects')
if not os.path.isdir(self.store):
raise InvalidStore("objects is not a directory")
db = os.path.join(path, 'database')
if not os.path.isfile(db):
raise InvalidStore("database is not a file")
self.metadata = MetadataStore(db)
@staticmethod
def create_store(path):
if os.path.exists(path):
if not os.path.isdir(path) or os.listdir(path):
raise CreationError("Path is not a directory or is not empty")
exists = True
else:
exists = False
try:
if not exists:
os.mkdir(path)
os.mkdir(os.path.join(path, 'objects'))
except OSError as e: # pragma: no cover
raise CreationError("Could not create directories: %s: %s" % (
e.__class__.__name__, e.message))
MetadataStore.create_db(os.path.join(path, 'database'))
def close(self):
self.metadata.close()
self.metadata = None
self.store = None
def open_file(self, objectid, path=None, binary=True):
"""Returns a file object for a given objectid.
"""
filepath = self.get_filename(objectid)
if os.path.isdir(filepath):
if path:
return open(os.path.join(filepath, path),
'rb' if binary else 'r')
else:
raise ValueError("Object is a directory, not a file")
else: # os.path.isfile(filepath):
if path is not None:
raise ValueError("Object is a file, not a directory")
else:
return open(filepath, 'rb' if binary else 'r')
def get_filename(self, objectid):
"""Returns the file path for a given objectid.
"""
if not isinstance(objectid, string_types):
raise TypeError("hash should be a string, not %s" % type(objectid))
metadata = self.metadata.get(objectid)
return self._make_filename(metadata['hash'])
def _make_filename(self, filehash, make_dir=False):
"""Gets or makes the path for the given filehash.
"""
dirname = os.path.join(self.store, filehash[:2])
if make_dir and not os.path.isdir(dirname):
os.mkdir(dirname)
return os.path.join(dirname, filehash[2:])
def add_file(self, newfile, metadata):
"""Adds a file given a file object or path and dict of metadata.
The file will be copied/written in the store, and an entry will be
added to the database.
Note that, if you pass a file object, it needs to support
newfile.seek(0, os.SEEK_SET) as it will be read twice: once to compute
its SHA1 hash, and a second time to write it to disk.
"""
if isinstance(newfile, string_types):
if os.path.islink(newfile):
warnings.warn("%s is a symbolic link, using target file "
"instead" % newfile,
UsageWarning)
with open(newfile, 'rb') as fp:
return self.add_file(fp, metadata)
newfile.seek(0, os.SEEK_SET)
filehash = hash_file(newfile)
newfile.seek(0, os.SEEK_SET)
metadata = dict(metadata)
metadata['hash'] = filehash
objectid = hash_metadata(metadata)
storedfile = self._make_filename(filehash, make_dir=True)
if not os.path.exists(storedfile):
copy_file(newfile, storedfile)
try:
self.metadata.add(objectid, metadata)
except: # pragma: no cover
os.remove(storedfile)
raise
return Entry(self, objectid, metadata)
def add_directory(self, newdir, metadata):
"""Adds a directory given a path and dict of metadata.
The directory will be recursively copied to the store, and an entry
will be added to the database.
"""
if not isinstance(newdir, string_types):
raise TypeError("newdir should be a string, not %s" % type(newdir))
try:
dirhash = hash_directory(newdir)
except (IOError, OSError):
raise ValueError("Can't access directory")
metadata = dict(metadata)
metadata['hash'] = dirhash
objectid = hash_metadata(metadata)
storeddir = self._make_filename(dirhash, make_dir=True)
if not os.path.exists(storeddir):
copy_directory(newdir, storeddir)
try:
self.metadata.add(objectid, metadata)
except: # pragma: no cover
shutil.rmtree(storeddir)
raise
return Entry(self, objectid, metadata)
def add(self, newpath, metadata):
"""Adds a file or directory with a dict of metadata.
This simply calls either add_file() or add_directory() with the given
arguments.
"""
if not isinstance(newpath, string_types):
raise TypeError("newpath should be a string, not %s" %
type(newpath))
if os.path.isdir(newpath):
return self.add_directory(newpath, metadata)
else:
return self.add_file(newpath, metadata)
def remove(self, objectid):
"""Removes a file or directory given its objectid.
It is deleted from the store and removed from the database.
"""
if isinstance(objectid, Entry):
entry = objectid
else:
entry = self.get(objectid)
self.metadata.remove(entry.objectid)
if not self.metadata.has_filehash(entry['hash']):
# Garbage collection
if os.path.isdir(entry.filename):
shutil.rmtree(entry.filename)
else:
os.remove(entry.filename)
def get(self, objectid):
"""Gets an Entry from a hash.
"""
metadata = self.metadata.get(objectid) # Might raise KeyError
return Entry(self, objectid, metadata)
def query_one(self, conditions):
"""Returns at most one Entry matching the conditions.
Returns one of the Entry object matching the conditions or None.
"""
objectid, metadata = self.metadata.query_one(conditions)
if objectid is None:
return None
else:
return Entry(self, objectid, metadata)
def query(self, conditions, limit=None):
"""Returns all the Entries matching the conditions.
An EntryIterator is returned, with which you can access the different
results.
"""
infos = self.metadata.query_all(conditions, limit)
return EntryIterator(self, infos)
def verify(self):
"""Checks the integrity of the store.
class FileStore(object):
"""Represents a file store.
"""
def __init__(self, path):
self.store = os.path.join(path, 'objects')
if not os.path.isdir(self.store):
raise InvalidStore("objects is not a directory")
db = os.path.join(path, 'database')
if not os.path.isfile(db):
raise InvalidStore("database is not a file")
self.metadata = MetadataStore(db)
@staticmethod
def create_store(path):
if os.path.exists(path):
if not os.path.isdir(path) or os.listdir(path):
raise CreationError("Path is not a directory or is not empty")
exists = True
else:
exists = False
try:
if not exists:
os.mkdir(path)
os.mkdir(os.path.join(path, 'objects'))
except OSError as e: # pragma: no cover
raise CreationError("Could not create directories: %s: %s" %
(e.__class__.__name__, e.message))
MetadataStore.create_db(os.path.join(path, 'database'))
def close(self):
self.metadata.close()
self.metadata = None
self.store = None
def open_file(self, objectid, path=None, binary=True):
"""Returns a file object for a given objectid.
"""
filepath = self.get_filename(objectid)
if os.path.isdir(filepath):
if path:
return open(os.path.join(filepath, path),
'rb' if binary else 'r')
else:
raise ValueError("Object is a directory, not a file")
else: # os.path.isfile(filepath):
if path is not None:
raise ValueError("Object is a file, not a directory")
else:
return open(filepath, 'rb' if binary else 'r')
def get_filename(self, objectid):
"""Returns the file path for a given objectid.
"""
if not isinstance(objectid, string_types):
raise TypeError("hash should be a string, not %s" % type(objectid))
metadata = self.metadata.get(objectid)
return self._make_filename(metadata['hash'])
def _make_filename(self, filehash, make_dir=False):
"""Gets or makes the path for the given filehash.
"""
dirname = os.path.join(self.store, filehash[:2])
if make_dir and not os.path.isdir(dirname):
os.mkdir(dirname)
return os.path.join(dirname, filehash[2:])
def add_file(self, newfile, metadata):
"""Adds a file given a file object or path and dict of metadata.
The file will be copied/written in the store, and an entry will be
added to the database.
Note that, if you pass a file object, it needs to support
newfile.seek(0, os.SEEK_SET) as it will be read twice: once to compute
its SHA1 hash, and a second time to write it to disk.
"""
if isinstance(newfile, string_types):
if os.path.islink(newfile):
warnings.warn(
"%s is a symbolic link, using target file "
"instead" % newfile, UsageWarning)
with open(newfile, 'rb') as fp:
return self.add_file(fp, metadata)
newfile.seek(0, os.SEEK_SET)
filehash = hash_file(newfile)
newfile.seek(0, os.SEEK_SET)
metadata = dict(metadata)
metadata['hash'] = filehash
objectid = hash_metadata(metadata)
storedfile = self._make_filename(filehash, make_dir=True)
if not os.path.exists(storedfile):
copy_file(newfile, storedfile)
try:
self.metadata.add(objectid, metadata)
except BaseException: # pragma: no cover
os.remove(storedfile)
raise
return Entry(self, objectid, metadata)
def add_directory(self, newdir, metadata):
"""Adds a directory given a path and dict of metadata.
The directory will be recursively copied to the store, and an entry
will be added to the database.
"""
if not isinstance(newdir, string_types):
raise TypeError("newdir should be a string, not %s" % type(newdir))
try:
dirhash = hash_directory(newdir)
except (IOError, OSError):
raise ValueError("Can't access directory")
metadata = dict(metadata)
metadata['hash'] = dirhash
objectid = hash_metadata(metadata)
storeddir = self._make_filename(dirhash, make_dir=True)
if not os.path.exists(storeddir):
copy_directory(newdir, storeddir)
try:
self.metadata.add(objectid, metadata)
except BaseException: # pragma: no cover
shutil.rmtree(storeddir)
raise
return Entry(self, objectid, metadata)
def add(self, newpath, metadata):
"""Adds a file or directory with a dict of metadata.
This simply calls either add_file() or add_directory() with the given
arguments.
"""
if not isinstance(newpath, string_types):
raise TypeError("newpath should be a string, not %s" %
type(newpath))
if os.path.isdir(newpath):
return self.add_directory(newpath, metadata)
else:
return self.add_file(newpath, metadata)
def remove(self, objectid):
"""Removes a file or directory given its objectid.
It is deleted from the store and removed from the database.
"""
if isinstance(objectid, Entry):
entry = objectid
else:
entry = self.get(objectid)
self.metadata.remove(entry.objectid)
if not self.metadata.has_filehash(entry['hash']):
# Garbage collection
if os.path.isdir(entry.filename):
shutil.rmtree(entry.filename)
else:
os.remove(entry.filename)
def get(self, objectid):
"""Gets an Entry from a hash.
"""
metadata = self.metadata.get(objectid) # Might raise KeyError
return Entry(self, objectid, metadata)
def query_one(self, conditions):
"""Returns at most one Entry matching the conditions.
Returns one of the Entry object matching the conditions or None.
"""
objectid, metadata = self.metadata.query_one(conditions)
if objectid is None:
return None
else:
return Entry(self, objectid, metadata)
def query(self, conditions, limit=None):
"""Returns all the Entries matching the conditions.
An EntryIterator is returned, with which you can access the different
results.
"""
infos = self.metadata.query_all(conditions, limit)
return EntryIterator(self, infos)
def verify(self):
"""Checks the integrity of the store.
