Skip to content
Snippets Groups Projects
Commit 2e5bdc4c authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

Use new readonly/no-lock session try constructions

parent d664cb27
No related branches found
No related tags found
No related merge requests found
...@@ -155,11 +155,9 @@ def define_protocols(session, protodir, verbose): ...@@ -155,11 +155,9 @@ def define_protocols(session, protodir, verbose):
def create_tables(args): def create_tables(args):
"""Creates all necessary tables (only to be used at the first time)""" """Creates all necessary tables (only to be used at the first time)"""
from bob.db.utils import connection_string from bob.db.utils import create_engine_try_nolock
from sqlalchemy import create_engine engine = create_engine_try_nolock(args.type, args.files[0], echo=(args.verbose >= 2))
engine = create_engine(connection_string(args.type, args.files[0]),
echo=(args.verbose >= 2))
Client.metadata.create_all(engine) Client.metadata.create_all(engine)
RealAccess.metadata.create_all(engine) RealAccess.metadata.create_all(engine)
Attack.metadata.create_all(engine) Attack.metadata.create_all(engine)
...@@ -171,7 +169,7 @@ def create_tables(args): ...@@ -171,7 +169,7 @@ def create_tables(args):
def create(args): def create(args):
"""Creates or re-creates this database""" """Creates or re-creates this database"""
from bob.db.utils import session from bob.db.utils import session_try_nolock
dbfile = args.files[0] dbfile = args.files[0]
...@@ -185,7 +183,7 @@ def create(args): ...@@ -185,7 +183,7 @@ def create(args):
# the real work... # the real work...
create_tables(args) create_tables(args)
s = session(args.type, args.files[0], echo=(args.verbose >= 2)) s = session_try_nolock(args.type, args.files[0], echo=(args.verbose >= 2))
add_clients(s, args.protodir, args.verbose) add_clients(s, args.protodir, args.verbose)
add_real_lists(s, args.protodir, args.verbose) add_real_lists(s, args.protodir, args.verbose)
add_attack_lists(s, args.protodir, args.verbose) add_attack_lists(s, args.protodir, args.verbose)
......
...@@ -34,7 +34,7 @@ class Database(object): ...@@ -34,7 +34,7 @@ class Database(object):
self.session = None self.session = None
else: else:
self.session = utils.session(INFO.type(), INFO.files()[0]) self.session = utils.session_try_readonly(INFO.type(), SQLITE_FILE)
def is_valid(self): def is_valid(self):
"""Returns if a valid session has been opened for reading the database""" """Returns if a valid session has been opened for reading the database"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment