Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Create _schema_exists method

Browse Source
This commit is contained in:
Nick Hall 2018-03-07 15:52:12 +00:00
parent cf42e5a4b8
commit 75bf103429
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gramps/gen/db/generic.py
View File

@ -625,9 +625,7 @@ class DbGeneric(DbWriteBase, DbReadBase, UpdateCallback, Callback):
# run backend-specific code: # run backend-specific code:
self._initialize(directory, username, password) self._initialize(directory, username, password)
# We use the existence of the person table as a proxy for the database if not self._schema_exists():
# being new
if not self.dbapi.table_exists("person"):
self._create_schema() self._create_schema()
self._set_metadata('version', str(self.VERSION[0])) self._set_metadata('version', str(self.VERSION[0]))

9
gramps/plugins/db/dbapi/dbapi.py
View File

@ -56,6 +56,15 @@ class DBAPI(DbGeneric):
def _initialize(self, directory, username, password): def _initialize(self, directory, username, password):
raise NotImplementedError raise NotImplementedError
def _schema_exists(self):
"""
Check to see if the schema exists.
We use the existence of the person table as a proxy for the database
being new.
"""
return self.dbapi.table_exists("person")
def _create_schema(self): def _create_schema(self):
""" """
Create and update schema. Create and update schema.