CNBP / DICOMTransit

Python/LocalDB_create.py

#This file create a SQLite local database for CNBP requirement.
#Creation: 2018-07-11T160228EST
#Author: Yang Ding

import sys
import os
import sqlite3
import logging
from pathlib import Path
from LocalDB_schema import *
import argparse
import getpass

logging.basicConfig(stream=sys.stdout, level=logging.INFO)
logger = logging.getLogger('LocalDBCreate')

def LocalDBCreate(PathString, TableName, KeyFieldString, ColumnsNameTypeList):
    """
    Create the local database based on sceham.
    :param PathString:
    :param TableName:
    :param KeyFieldString:
    :param ColumnsNameTypeList:
    :return:
    """


    # if SQL already exist, quit script.
    SQLPath = Path(PathString)

    # check if path is a fiela nd exist.
    if SQLPath.is_file():
        logger.info('SQLite database file already exist. Not gonna mess with it!')
        return False
        '''Delete current database! During testing only'''
        '''os.remove(sqliteFile)
        logger.info('DEBUG: database file already exist. Deleted it!')'''

    #Create the PRIMARY KEY column.
    KeyFieldType = CNBP_schema_keyfield_type  # column data type

The variable CNBP_schema_keyfield_type does not seem to be defined.

    #Try to connect the database to start the process:

    try:
        # Create on Connecting to the database file
        ConnectedDatabase = sqlite3.connect(PathString)

        c = ConnectedDatabase.cursor()

        logger.info('Creating PRIMARY KEY DBKEY column in database.')

        # Creating a new SQLite table with DBKey column (inspired by: https://sebastianraschka.com/Articles/2014_sqlite_in_python_tutorial.html)
        c.execute('CREATE TABLE {tn} ({nf} {ft} PRIMARY KEY)'.format(tn=TableName, nf=KeyFieldString, ft=KeyFieldType))

        logger.info('PRIMARY KEY DBKEY column successfully created in database.')

        logger.info('Creating secondary columns in database.')

        # Adding accessory columns via a loop
        for column in ColumnsNameTypeList:
            if (column[1] != "TEXT" and
                column[1] != "REAL" and
                column[1] != "BLOB" and
                #column[1] != "NULL" and
                column[1] != "INTEGER"):
                continue  # skip iteration is the data type is not specified properly.
            else:
                c.execute("ALTER TABLE {tn} ADD COLUMN '{cn}' {ct}".format(tn=TableName, cn=column[0], ct=column[1]))

        logger.info('Secondary columns created in database.')

        # Committing changes and closing the connection to the database file
        ConnectedDatabase.commit()
        ConnectedDatabase.close()
    except:
        logger.info('SQLite database creation/update issue, suspect schema non-compliant SQLite database. Did you corrupt this SQLite database somehow?')
        raise IOError
    return True