Skip to content
Snippets Groups Projects
Commit b28bffe9 authored by Pavel KORSHUNOV's avatar Pavel KORSHUNOV
Browse files

putting back the fix from commit c49ad044

parent 02fe6e88
Branches
Tags
1 merge request!13Support for configuration files
Pipeline #
......@@ -33,7 +33,7 @@ class TestFile(PadFile):
attack_type = None
if "attack" in path:
attack_type = "attack"
PadFile.__init__(self, client_id=1, path=path, file_id=id, attack_type=attack_type)
super(TestFile, self).__init__(client_id=1, path=path, file_id=id, attack_type=attack_type)
def load(self, directory=None, extension='.hdf5'):
"""Loads the data at the specified location and using the given extension.
......@@ -96,11 +96,13 @@ class Interface(BaseInterface):
dumpparser = subparsers.add_parser('dumplist', help="")
dumpparser.add_argument('-d', '--directory', dest="directory", default='',
help="if given, this path will be prepended to every entry returned (defaults to '%(default)s')")
help="if given, this path will be prepended to every entry returned "
"(defaults to '%(default)s')")
dumpparser.add_argument('-e', '--extension', dest="extension", default='',
help="if given, this extension will be appended to every entry returned (defaults to '%(default)s')")
help="if given, this extension will be appended to every entry returned "
"(defaults to '%(default)s')")
dumpparser.add_argument('--self-test', dest="selftest", default=False,
action='store_true', help=SUPPRESS)
action='store_true', help=SUPPRESS)
dumpparser.set_defaults(func=dumplist) # action
......@@ -110,9 +112,9 @@ class TestDatabase(PadDatabase):
def __init__(self, protocol='Default', original_directory=data_dir, original_extension='', **kwargs):
# call base class constructors to open a session to the database
PadDatabase.__init__(self, name='testspoof', protocol=protocol,
original_directory=original_directory,
original_extension=original_extension, **kwargs)
super(TestDatabase, self).__init__(name='testspoof', protocol=protocol,
original_directory=original_directory,
original_extension=original_extension, **kwargs)
################################################
# Low level support methods for the database #
......@@ -194,4 +196,4 @@ class TestDatabase(PadDatabase):
return None
database = TestDatabase(original_directory=data_dir, original_extension='')
database = TestDatabase()
......@@ -45,10 +45,12 @@ def create_database():
del engine
class TestDatabaseSql (bob.pad.base.database.PadDatabase, bob.db.base.SQLiteDatabase):
class TestDatabaseSql (bob.pad.base.database.PadDatabase, bob.db.base.SQLiteBaseDatabase):
def __init__(self):
bob.pad.base.database.PadDatabase.__init__(self, 'pad_test', original_directory="original/directory", original_extension=".orig")
bob.db.base.SQLiteDatabase.__init__(self, dbfile, TestFileSql)
bob.pad.base.database.PadDatabase.__init__(self, 'pad_test',
original_directory="original/directory", original_extension=".orig")
bob.db.base.SQLiteBaseDatabase.__init__(self, dbfile, TestFileSql)
def groups(self, protocol=None):
return ['group']
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment