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
No related branches found
No related tags found
1 merge request!13Support for configuration files
Pipeline #
...@@ -33,7 +33,7 @@ class TestFile(PadFile): ...@@ -33,7 +33,7 @@ class TestFile(PadFile):
attack_type = None attack_type = None
if "attack" in path: if "attack" in path:
attack_type = "attack" 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'): def load(self, directory=None, extension='.hdf5'):
"""Loads the data at the specified location and using the given extension. """Loads the data at the specified location and using the given extension.
...@@ -96,11 +96,13 @@ class Interface(BaseInterface): ...@@ -96,11 +96,13 @@ class Interface(BaseInterface):
dumpparser = subparsers.add_parser('dumplist', help="") dumpparser = subparsers.add_parser('dumplist', help="")
dumpparser.add_argument('-d', '--directory', dest="directory", default='', 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='', 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, 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 dumpparser.set_defaults(func=dumplist) # action
...@@ -110,9 +112,9 @@ class TestDatabase(PadDatabase): ...@@ -110,9 +112,9 @@ class TestDatabase(PadDatabase):
def __init__(self, protocol='Default', original_directory=data_dir, original_extension='', **kwargs): def __init__(self, protocol='Default', original_directory=data_dir, original_extension='', **kwargs):
# call base class constructors to open a session to the database # call base class constructors to open a session to the database
PadDatabase.__init__(self, name='testspoof', protocol=protocol, super(TestDatabase, self).__init__(name='testspoof', protocol=protocol,
original_directory=original_directory, original_directory=original_directory,
original_extension=original_extension, **kwargs) original_extension=original_extension, **kwargs)
################################################ ################################################
# Low level support methods for the database # # Low level support methods for the database #
...@@ -194,4 +196,4 @@ class TestDatabase(PadDatabase): ...@@ -194,4 +196,4 @@ class TestDatabase(PadDatabase):
return None return None
database = TestDatabase(original_directory=data_dir, original_extension='') database = TestDatabase()
...@@ -45,10 +45,12 @@ def create_database(): ...@@ -45,10 +45,12 @@ def create_database():
del engine 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): def __init__(self):
bob.pad.base.database.PadDatabase.__init__(self, 'pad_test', original_directory="original/directory", original_extension=".orig") bob.pad.base.database.PadDatabase.__init__(self, 'pad_test',
bob.db.base.SQLiteDatabase.__init__(self, dbfile, TestFileSql) original_directory="original/directory", original_extension=".orig")
bob.db.base.SQLiteBaseDatabase.__init__(self, dbfile, TestFileSql)
def groups(self, protocol=None): def groups(self, protocol=None):
return ['group'] return ['group']
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment