voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add PhotoDB init option create. Disable to require existing.

Browse source
This commit is contained in:
voussoir 2018-03-13 02:24:11 -07:00
parent 101f6d9c47
commit d85d911b22
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
etiquette/photodb.py
View file

@ -1075,6 +1075,8 @@ class PhotoDB(
def __init__(
self,
data_directory=None,
*,
create=True,
ephemeral=False,
):
super().__init__()
@ -1098,7 +1100,6 @@ class PhotoDB(
if self.data_directory.exists and not self.data_directory.is_dir:
raise exceptions.BadDataDirectory(self.data_directory.absolute_path)
os.makedirs(self.data_directory.absolute_path, exist_ok=True)
self.log = logging.getLogger('etiquette:%s' % self.data_directory.absolute_path)
self.log.setLevel(logging.DEBUG)
@ -1110,6 +1111,12 @@ class PhotoDB(
else:
self.database_filepath = self.data_directory.with_child(constants.DEFAULT_DBNAME)
existing_database = self.database_filepath.exists
if not create and not self.ephemeral and not existing_database:
raise FileNotFoundError('"%s" does not exist and create is off.' % self.data_directory)
if not self.ephemeral:
os.makedirs(self.data_directory.absolute_path, exist_ok=True)
self.sql = sqlite3.connect(self.database_filepath.absolute_path)
if existing_database: