Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
bob.pad.base
Commits
ee0abff6
Commit
ee0abff6
authored
May 12, 2017
by
Pavel KORSHUNOV
Browse files
Merge branch 'database' into 'master'
Fix issue
bob.db.base#16
Closes
bob.db.base#16
See merge request
!11
parents
4b9f2304
c49ad044
Pipeline
#9223
passed with stages
in 22 minutes and 7 seconds
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
bob/pad/base/test/dummy/database.py
View file @
ee0abff6
...
...
@@ -29,11 +29,13 @@ dummy_data = {'train_real': 1.0, 'train_attack': 2.0,
class
TestFile
(
PadFile
):
def
__init__
(
self
,
path
,
id
):
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
)
PadFile
.
__init__
(
self
,
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.
...
...
@@ -57,6 +59,7 @@ class TestFile(PadFile):
path
=
self
.
make_path
(
directory
or
''
,
extension
or
''
)
return
dummy_data
[
os
.
path
.
basename
(
path
)]
def
dumplist
(
args
):
"""Dumps lists of files based on your criteria"""
...
...
@@ -76,6 +79,7 @@ def dumplist(args):
class
Interface
(
BaseInterface
):
def
name
(
self
):
return
dummy_name
...
...
@@ -96,11 +100,11 @@ 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 +114,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
Pad
Database
.
__init__
(
self
,
name
=
'testspoof'
,
protocol
=
protocol
,
original_directory
=
original_directory
,
original_extension
=
original_extension
,
**
kwargs
)
super
(
Test
Database
,
self
)
.
__init__
(
name
=
'testspoof'
,
protocol
=
protocol
,
original_directory
=
original_directory
,
original_extension
=
original_extension
,
**
kwargs
)
################################################
# Low level support methods for the database #
...
...
@@ -168,6 +172,7 @@ class TestDatabase(PadDatabase):
# does not implement the given access protocol
return
False
def
get_all_data
(
self
):
return
self
.
all_files
()
...
...
bob/pad/base/test/dummy/database_sql.py
View file @
ee0abff6
...
...
@@ -45,10 +45,10 @@ 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
.
SQLite
Base
Database
):
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
.
db
.
base
.
SQLite
Base
Database
.
__init__
(
self
,
dbfile
,
TestFileSql
)
def
groups
(
self
,
protocol
=
None
):
return
[
'group'
]
...
...
@@ -59,4 +59,4 @@ class TestDatabaseSql (bob.pad.base.database.PadDatabase, bob.db.base.SQLiteData
def
annotations
(
self
,
file
):
return
None
database
=
TestDatabaseSql
()
\ No newline at end of file
database
=
TestDatabaseSql
()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment