AtlTriggerDBCopy Namespace Reference

Classes
class	DBConnection
class	OracleExporter
class	SQLiteInserter

Functions
def	parseCmdline ()
def	main ()

Function Documentation

main()

def AtlTriggerDBCopy.main

(

)

Definition at line 274 of file AtlTriggerDBCopy.py.

def main():
    args = parseCmdline()
    if args is None:
        return 1
 
    # instantiate the sqlite inserter with the sqlite db filename
    sqliteInserter = SQLiteInserter(args.createfile)
 
    # connect to oracle and do the work
    with DBConnection(args.dbalias) as oraConn:
        oraExp = OracleExporter(oraConn)
 
        # extract the schema into a temporary file and create it on the sqlite side
        tempTableCreationFileName = "tmpCreateTablesSQLite.sql"
        oraExp.extractSchema(tempTableCreationFileName)
        sqliteInserter.createSchema(tempTableCreationFileName)
 
        # collect some info about the size, to not run completely blind
        entries = {}
        tablesToCreate = oraExp.getTables(exceptTables = oraExp.ignoreTablesCreate)
        tablesToFill = oraExp.getTables(exceptTables = oraExp.ignoreTablesFill)
        print("Tables to create: %i" % len(tablesToCreate))
        print("Tables to fill: %i" % len(tablesToFill))
 
 
        for tableName in tablesToCreate:
            entries[tableName] = oraExp.tableSize(tableName)
            doNotCopy = tableName not in tablesToFill
            print("  table %s has %i entries %s" % (tableName, entries[tableName], ("(will not copy)" if doNotCopy else "") ) )
        totalEntries = sum(entries.values())
        print("\nTotal number of entries: %i" %totalEntries)
        
        # copy the data one table at the time
        print("Start copying data")
        copiedEntries = 0
        for tableName in tablesToFill:            
            print("Copying table %s" % tableName, end = '', flush=True)
            oraExp.copyTable(tableName, sqliteInserter, entries[tableName])
            copiedEntries += entries[tableName]
            print(" => done %i / %i (%f%%)" % (copiedEntries, totalEntries, 100 * copiedEntries/totalEntries))
 
    sqliteInserter.connection.close()
 
 

parseCmdline()

def AtlTriggerDBCopy.parseCmdline

(

)

Definition at line 11 of file AtlTriggerDBCopy.py.

def parseCmdline():
    import argparse
    parser = argparse.ArgumentParser(prog = "AtlTriggerDBCopy.py", description="Example: %(prog)s -d TRIGGERDBMC -c triggerDBMC_Run2.db")
    parser.add_argument("-d", "--dbalias", dest="dbalias", help="TriggerDB connection alias for the source DB")
    parser.add_argument("-c", "--create", dest="createfile", help="create sqlite db file")
    parser.add_argument("-v", help="increase output verbosity", action="count", default=0)
    args = parser.parse_args()
    if not args.dbalias:
        print("No source Trigger DB specified")
        parser.print_help()
        return None
    if not args.createfile:
        print("No sqlite target file specified")
        parser.print_help(parser)
        return None
    return args
 
# class to handle the oracle db connection
# it uses the CORAL authentication method implemented in the TrigConfIO.TriggerConfigAccessBase (ConfigDBLoader)