diff --git a/bob/db/replaymobile/query.py b/bob/db/replaymobile/query.py index 2e50437827f24782eb7ea2b6aba7d62e5d4293f7..751f72f81f4705ab97379050a37ff4b5f414ef28 100644 --- a/bob/db/replaymobile/query.py +++ b/bob/db/replaymobile/query.py @@ -22,10 +22,10 @@ class Database(Database): and for the data itself inside the database. """ - def __init__(self): + def __init__(self, original_directory=None, original_extension=None): # opens a session to the database - keep it open until the end self.connect() - super(Database, self).__init__() + super(Database, self).__init__(original_directory, original_extension) def __del__(self): """Releases the opened file descriptor""" diff --git a/bob/db/replaymobile/verificationprotocol.py b/bob/db/replaymobile/verificationprotocol.py index 80b8d603f871137f69c6ab9479155191975dc78b..979c8ebbc92f1d300439cb7e58c810d54fce96d8 100644 --- a/bob/db/replaymobile/verificationprotocol.py +++ b/bob/db/replaymobile/verificationprotocol.py @@ -8,7 +8,7 @@ framework). It also implements a kind of hack so that you can run vulnerability analysis with it. """ from bob.db.base import File as BaseFile -from bob.db.base import Database as BaseDatabase +from bob.db.base import FileDatabase as BaseDatabase from .query import Database as LDatabase @@ -66,8 +66,8 @@ class Database(BaseDatabase): """ __doc__ = __doc__ - def __init__(self, max_number_of_frames=None): - super(Database, self).__init__() + def __init__(self, max_number_of_frames=None, original_directory=None, original_extension=None): + super(Database, self).__init__(original_directory, original_extension) # call base class constructors to open a session to the database self._db = LDatabase()