From d5c1fa29e5315246c0dca95bfa84b2ac037d214f Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Thu, 11 May 2017 16:38:58 +0200
Subject: [PATCH] The SQLiteDatabase now accepts original_directory and
 original_extension

---
 bob/db/replaymobile/query.py                | 4 ++--
 bob/db/replaymobile/verificationprotocol.py | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/bob/db/replaymobile/query.py b/bob/db/replaymobile/query.py
index 2e50437..751f72f 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 80b8d60..979c8eb 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()
-- 
GitLab