From a6472586831460d13c147921b388ed49dfa932dd Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Wed, 3 Oct 2018 10:18:09 +0200
Subject: [PATCH] added minimal create.py
---
bob/db/fargo/create.py | 81 ++++++++++++++++++++++++++++++++++++++++++
1 file changed, 81 insertions(+)
create mode 100644 bob/db/fargo/create.py
diff --git a/bob/db/fargo/create.py b/bob/db/fargo/create.py
new file mode 100644
index 0000000..f5a6130
--- /dev/null
+++ b/bob/db/fargo/create.py
@@ -0,0 +1,81 @@
+#!/usr/bin/env python
+# vim: set fileencoding=utf-8 :
+
+import os
+from .models import *
+
+import bob.core
+logger = bob.core.log.setup('bob.db.fargo')
+
+
+def add_clients(session, imagesdir, verbose=True):
+ """Add clients
+
+ This function add clients to the database.
+
+ Parameters
+ ----------
+ imagesdir : :py:obj:str
+ The directory where the images have been extracted
+ verbose : bool
+ Print some information
+
+ """
+ if verbose:
+ logger.info("Adding clients ...")
+ for d in os.listdir(imagesdir):
+ print(d)
+
+
+def create_tables(args):
+ """Creates all necessary tables (only to be used at the first time)"""
+
+ from bob.db.base.utils import create_engine_try_nolock
+ engine = create_engine_try_nolock(args.type, args.files[0], echo=(args.verbose > 2))
+ Base.metadata.create_all(engine)
+
+
+# Driver API
+# ==========
+
+def create(args):
+ """Creates or re-creates this database"""
+
+ from bob.db.base.utils import session_try_nolock
+
+ dbfile = args.files[0]
+
+ if args.recreate:
+ if args.verbose and os.path.exists(dbfile):
+ print(('unlinking %s...' % dbfile))
+ if os.path.exists(dbfile):
+ os.unlink(dbfile)
+
+ if not os.path.exists(os.path.dirname(dbfile)):
+ os.makedirs(os.path.dirname(dbfile))
+
+ # the real work...
+ create_tables(args)
+ s = session_try_nolock(args.type, args.files[0], echo=(args.verbose >= 2))
+ add_clients(s, args.imagesdir, args.verbose)
+ s.commit()
+ s.close()
+
+ return 0
+
+
+def add_command(subparsers):
+ """Add specific subcommands that the action "create" can use"""
+
+ parser = subparsers.add_parser('create', help=create.__doc__)
+
+ parser.add_argument('-R', '--recreate', action='store_true', default=False,
+ help="If set, I'll first erase the current database")
+ parser.add_argument('-v', '--verbose', action='count', default=0,
+ help="Do SQL operations in a verbose way")
+ parser.add_argument('-i', '--imagesdir', action='store',
+ default='/idiap/temp/heusch/bob.project.fargo/images/'
+ metavar='DIR',
+ help="Change the path to the extracted images of the FARGO database (defaults to %(default)s)")
+
+ parser.set_defaults(func=create) # action
--
GitLab