data-manipulation/dm.py

# -*- coding: utf-8 -*-
import json, os, configparser, shutil, re
from datetime import datetime, date, time

def firstInit():
    # Инициализация переменныхб создание конфигруационного файла, копирование шаблонов
    global db_type, db_hostname, db_user, db_password, db_name, template_file, db_type
    config = configparser.RawConfigParser()
    # проверяем тип ОС
    if os.name == "nt":
        cfg_dir = os.path.join(os.getenv("USERPROFILE"), ".dm")
    elif os.name == "posix":
        cfg_dir = os.path.join(os.getenv("HOME"), ".dm")
    # определим каталог для конфигруции и создадим если его нет
    if os.path.isdir(cfg_dir):
        print(cfg_dir + " already exists")
    else:
        os.mkdir(cfg_dir)
    cfg_file = os.path.join(cfg_dir, 'dm.cfg')
    # создадим файл конфигурации
    config.add_section('DataBase')
    config.set('DataBase', 'db_type', 'mysql')
    config.set('DataBase', 'db_hostname', 'localhost')
    config.set('DataBase', 'db_name', 'dm')
    config.set('DataBase', 'db_user', 'dm')
    config.set('DataBase', 'db_password', 'password')
    config.add_section('Directory')
    config.set('Directory', 'work_dir', cfg_dir)
    config.set('Directory', 'template_dir', os.path.join(cfg_dir, 'db_template'))

    # Если конфиг уже есть читаем его если нет, создаем и потом читаем
    if os.path.isfile(cfg_file):
        config.read(cfg_file)
    else:
        # Запись конфигурации в файл 'example.cfg'
        with open(cfg_file, 'w') as configfile:
            config.write(configfile)
            configfile.close()
        config.read(cfg_file)

    work_dir = config.get('Directory', 'work_dir')
    template_dir = config.get('Directory', 'template_dir')
    db_type = config.get('DataBase', 'db_type')
    db_hostname =  config.get('DataBase', 'db_hostname')
    if db_type == 'sqlite':
        db_name =  os.path.join(work_dir, config.get('DataBase', 'db_name'))
    else:
        db_name = config.get('DataBase', 'db_name')
    db_user = config.get('DataBase', 'db_user')
    db_password = config.get('DataBase', 'db_password')

    # Создаём нужные каталоги
    if os.path.isdir(work_dir):
        print(work_dir + " already exists")
    else:
        os.mkdir(work_dir)
    if os.path.isdir(template_dir):
        print(template_dir + " already exists")
    else:
        os.mkdir(template_dir)
    # копируем и читаем файл шаблон БД
    template_file = os.path.join(template_dir, 'tables.json')
    if os.path.isfile(template_file):
        print("Template file already exists")
    else:
        shutil.copy('tables.json', template_file)


def dbConnect():
    global c, db_type, db_hostname, db_user, db_password, db_name, conn
    if db_type == "mysql":
        import pymysql
        conn = pymysql.connect(
            db=db_name,
            user=db_user,
            passwd=db_password,
            host=db_hostname,
            charset='utf8')
        conn.commit()
    elif db_type == "sqlite":
        import sqlite3
        conn = sqlite3.connect(db_name)
    else:
        print(db_type + " database type does`t support")
        exit()
    #print(conn)
    c = conn.cursor()
    return c

def createTables(tbl_list):
    global dbTablesDescriptionList, dbTablesStructList, c, db_type
    i = 0
    dbTablesNamesList = []
    dbTablesDescriptionList = []
    dbTablesStructList = []

    while i < len(tbl_list):
        one_Table_descr = []
        one_Table_struct = []
        one_Table_relation = []
        tbl_descr = tbl_list[i]["tableDescription"]
        tbl_name = tbl_list[i]["tableName"]
        #field_list = tbl_list[i]["fieldList"][i].keys()
        dbTablesNamesList.append([tbl_name, tbl_descr])
        one_Table_descr.append(tbl_name)
        one_Table_struct.append(tbl_name)

        x = 0
        # список всех полей таблицы
        qwery_create = "CREATE TABLE IF NOT EXISTS " + tbl_name + " ("
        index = ""
        field_names_list = []
        struct_fields_list = []
        # формируем запрос в зависимости от типа БД
        if db_type == 'mysql':
            while x < len(tbl_list[i]["fieldList"]):
                if tbl_list[i]["fieldList"][x]["autoIncrement"] == "yes":
                    auto_increment = " AUTO_INCREMENT, "
                else:
                    auto_increment = ", "
                qwery_create = qwery_create + tbl_list[i]["fieldList"][x]["fName"] + " " + \
                               tbl_list[i]["fieldList"][x]["fType"] + auto_increment
                if tbl_list[i]["fieldList"][x]["index"] == "PRIMARY KEY":
                    index = ", PRIMARY KEY (" + tbl_list[i]["fieldList"][x]["fName"] + ")"
                field_names_list.append(
                    [tbl_list[i]["fieldList"][x]["fName"], tbl_list[i]["fieldList"][x]["fDescription"]])
                # struct_fields_list.append(tbl_list[i]["fieldList"][x]["fName"])
                struct_fields_list.append(
                    [tbl_list[i]["fieldList"][x]["fName"], fieldTypeConvert(tbl_list[i]["fieldList"][x]["fType"]), auto_increment.strip(', ')])
                if tbl_list[i]["fieldList"][x]["relation"]:
                    relation_list = [tbl_list[i]["fieldList"][x]["fName"]]
                    relation_list.append(tbl_list[i]["fieldList"][x]["relation"])

                    one_Table_relation.append(relation_list)
                x = x + 1
            qwery_create = qwery_create.strip(', ') + index + ');'
            #qwery_create = qwery_create + index + ');'
            #print(qwery_create)
        elif db_type == 'sqlite':
            while x < len(tbl_list[i]["fieldList"]):
                if tbl_list[i]["fieldList"][x]["autoIncrement"] == "yes":
                    auto_increment = " AUTOINCREMENT,"
                else:
                    auto_increment = ", "
                if tbl_list[i]["fieldList"][x]["index"] == "PRIMARY KEY":
                   index = " PRIMARY KEY"
                else:
                    index = ''
                qwery_create = qwery_create + tbl_list[i]["fieldList"][x]["fName"] + ' ' + \
                               tbl_list[i]["fieldList"][x]["fType"] + index + auto_increment

                field_names_list.append([tbl_list[i]["fieldList"][x]["fName"], tbl_list[i]["fieldList"][x]["fDescription"]])
                # struct_fields_list.append(tbl_list[i]["fieldList"][x]["fName"])
                struct_fields_list.append(
                    [tbl_list[i]["fieldList"][x]["fName"], fieldTypeConvert(tbl_list[i]["fieldList"][x]["fType"]), auto_increment.strip(', ')])
                if tbl_list[i]["fieldList"][x]["relation"]:
                    relation_list = [tbl_list[i]["fieldList"][x]["fName"]]
                    relation_list.append(tbl_list[i]["fieldList"][x]["relation"])

                    one_Table_relation.append(relation_list)
                x = x + 1

            qwery_create = qwery_create.strip(', ') + ");"
        one_Table_struct.append(struct_fields_list)
        one_Table_struct.append(one_Table_relation)
        one_Table_descr.append(field_names_list)
        i = i + 1
        dbTablesDescriptionList.append(one_Table_descr)
        dbTablesStructList.append(one_Table_struct)
        #print(qwery_create)
        c.execute(qwery_create)
    return dbTablesNamesList

def initDBstructure():
    global dbTablesDescriptionList, template_file, tblNamesList
    table_list = open(template_file, "r", encoding="utf-8")
    data = json.load(table_list, encoding="utf-8")
    tbl_list = data["tables"]
    tblNamesList = createTables(tbl_list)

    return tblNamesList

# выборка данных из заданной таблицы
def selectData(tbl):
    global dbTablesStructList, c, db_type
    # если юольше 1 поля добавить CONCAT
    qwery = "SELECT "
    subqwery = ""
    for item in dbTablesStructList:
        if item[0] == tbl:
            for field in item[1]:
                field = field[0]
                qwery = qwery + field + ","
            for rel in item[2]:
                field = rel[0]
                field_rel = rel[1][0]
                field_replace = rel[1][1]
                # определяем название таблицы для вложенного запроса
                table1 = field_rel.split('.')[0]
                field1 = field_rel.split('.')[1]
                if table1 == tbl:
                    table1 = table1 + "_1"
                    subqwery = tbl + " AS " +table1
                else:
                    subqwery = table1
                # составляем подзапрос и подменяем им поле в запросе
                if db_type == "mysql":
                    field_replace = field_replace.replace(",", ",' ',")
                    #subqwery = "(SELECT CONCAT(" + field_replace + ") FROM " + subqwery + " WHERE " + table1 + "." + field1 + "=" + tbl + "." + field + ") AS " + field
                    # subqwery = "(SELECT CONCAT({}) FROM {} WHERE  {}.{}={}.{}) AS {}"\
                    #     .format(field_replace,subqwery,table1,field1,tbl,field,field)
                    subqwery = "(SELECT CONCAT('>', {}, '<', {}) FROM {} WHERE  {}.{}={}.{}) AS {}"\
                        .format(field,field_replace,subqwery,table1,field1,tbl,field,field)
                elif db_type == "sqlite":
                    field_replace = field_replace.replace(",", " || ' ' ||")
                    #subqwery = "(SELECT (" + field_replace + ") FROM " + subqwery +" WHERE "+ table1 + "." + field1 +"="+ tbl +"."+ field +") AS " + field
                    subqwery = "(SELECT ({}) FROM {} WHERE {}.{}={}.{}) AS {}" \
                        .format(field_replace, subqwery, table1, field1, tbl, field, field)
                qwery = qwery.replace(field, subqwery)
    #qwery = qwery.rstrip(',') + " FROM " + tbl + " LIMIT 10000"
    qwery = '{} FROM {} LIMIT 10000'.format(qwery.rstrip(','), tbl)
    print(qwery)
    c.execute(qwery)
    return c.fetchall()

# получаем на вход имя таблицы и возвращаем список заголовков полей
def getTableStructure(tbl):
    global dbTablesDescriptionList, dbTablesStructList
    #print(dbTablesDescriptionList)
    #print(dbTablesStructList)
    for item in dbTablesDescriptionList:
        if item[0] == tbl:
            return item[1]

# Получаем список названий полей и типов для заданной таблицы
def getFields(tbl):
    global dbTablesDescriptionList, dbTablesStructList
    #print(dbTablesDescriptionList)
    #print(dbTablesStructList)
    for item in dbTablesStructList:
        if item[0] == tbl:
            #print(item[1])
            return item[1]
# Ищем описание поля по его названию и возвращаем.
def getFieldDescription(tbl, field):
    global dbTablesDescriptionList, dbTablesStructList
    for item in dbTablesDescriptionList:
        if item[0] == tbl:
            for i in item[1]:
                if i[0] == field:
                    fName = i[1]
                    return fName
    return "null"

def fieldTypeConvert(ftype):
    # разбираем строку на тип и длину
    ftype = str.lower(ftype)
    line = re.search("(.+?)\(([0-9]+)\)", ftype)
    if  line:
        fType = line.groups()[0]
        fLength = line.groups()[1]
        if fType == 'int' or fType == 'integer':
            fType = 'integer'
        elif fType == 'char' or fType == 'varchar':
            fType = 'character'
    else:
        fType = ftype
    return fType

def getFieldType(tbl, field):
    global dbTablesDescriptionList, dbTablesStructList
    #print(dbTablesDescriptionList)
    #print(dbTablesStructList)
    for item in dbTablesStructList:
        if item[0] == tbl:
            for i in item[1]:
                if i[0] == field:
                    fType = i[1]
                    return fType
            #print(item[1])
            #return item[1]

def getRelationsForField(tblSearch, fieldName):
    global dbTablesStructList, c, db_type
    #print("Ищем связи для:" +tblSearch +','+ fieldName)
    #print(dbTablesStructList)
    #searchField = tblSearch + '.' + fieldName
    dataList = []
    for item in dbTablesStructList:
        #print(item)
        tblName = item[0]
        for field in item[1]:
            field = field[0]
        for rel in item[2]:
            if tblName == tblSearch:
                # выдергиваем из списка название и поле нужной нам таблицы
                if rel[0] == fieldName:
                    #print('table -  ' + tblName)
                    ##print('field - ' + fieldName)
                    #print('relation - ' + str(rel))
                    l = rel[1][0].split('.')
                    relTable = l[0]
                    relField = l[1]
                    replaceFields = rel[1][1]
                   # print('reltable -  ' + relTable)
                    #print('relfield - ' + relField)
                    #print('relrelation - ' + replaceFields)
                    return [relTable, relField, replaceFields]
    return "q"

# добавление записи в БД на вход принимает список вида:
# [table [['field_1', 'значение'], ['field_2', 'значение'],....., ['field_n', 'значение']]]
def insertDataIntoBD(dataList):
    global c, conn
    dbConnect()
    tableName = dataList[0]
    #print(dataList)
    #datetime.strptime("21/11/06 16:30", "%d/%m/%y %H:%M")

    #qwery = 'INSERT INTO ' + tableName + ' ('
    qwery = 'INSERT INTO {} ('.format(tableName)
    qweryData = ''
    qweryField = ''
    for item in dataList[1]:
        fType = getFieldType(tableName, item[0])
        #print(fType)
        # проверяем если значение поля пустое то в запрос оно не включается
        if item[1] != '':
            #qweryField = qweryField + item[0] + ','
            qweryField = '{}{},'.format(qweryField,item[0])
            if fType == 'integer':
                #qweryData = qweryData + '' + item[1] + ','
                qweryData = '{}{},'.format(qweryData,item[1])
            elif fType == 'datetime':
                # преобразуем дату всяко разно
                dt = datetime.strptime(item[1], "%d.%m.%y %H:%M")
                item[1] = str(dt)
                #qweryData = qweryData + '\'' + item[1] + '\','
                qweryData = "{}'{}',".format(qweryData,item[1])
            elif fType == 'date':
                d = item[1].split('.')
                #myDate = d[2] + '-' + d[1] + '-' + d[1]
                myDate = '{}-{}-{}'.format(d[2], d[1], d[1])
                item[1] = str(myDate)
                #qweryData = qweryData + '\'' + item[1] + '\','
                qweryData = "{}'{}',".format(qweryData, item[1])
            else:
                #qweryData = qweryData + '\'' + item[1] + '\','
                qweryData = "{}'{}',".format(qweryData, item[1])


    #qwery = qwery + qweryField.rstrip(',') + ')' + ' VALUES (' + qweryData.rstrip(',') + ');'
    qwery = '{}{}) VALUES ({});'.format(qwery, qweryField.rstrip(','), qweryData.rstrip(','))

    print(qwery)
    c.execute(qwery)
    conn.commit()
    #c.close()
    return

# Выборка данных по значению
# принимает на вход название таблицы, назване поля, значение этого поля
def selectDataFromDB(tblName, fieldName, fieldValue):
    global dbTablesStructList, c, db_type
    qwery = "SELECT "
    subqwery = ""

    for item in dbTablesStructList:
        if item[0] == tblName:
            for field in item[1]:
                field = field[0]
                qwery = qwery + field + ","
            for rel in item[2]:
                field = rel[0]
                fieldRel = rel[1][0]
                fieldReplace = rel[1][1]
                # определяем название таблицы для вложенного запроса
                table1 = fieldRel.split('.')[0]
                field1 = fieldRel.split('.')[1]
                if table1 == tblName:
                    table1 = table1 + "_1"
                    subqwery = tblName + " AS " +table1
                else:
                    subqwery = table1
                # составляем подзапрос и подменяем им поле в запросе
                if db_type == "mysql":
                    fieldReplace = fieldReplace.replace(",", ",' ',")
                    #subqwery = "(SELECT CONCAT(" + fieldReplace + ") FROM " + subqwery + " WHERE " + table1 + "." + field1 + "=" + tblName + "." + field + ") AS " + field
                    subqwery = '(SELECT CONCAT({}) FROM {} WHERE {}.{}={}.{}) AS {}'\
                        .format(fieldReplace, subqwery, table1, field1, tblName, field, field)
                elif db_type == "sqlite":
                    fieldReplace = fieldReplace.replace(",", " || ' ' ||")
                    #subqwery = "(SELECT (" + fieldReplace + ") FROM " + subqwery +" WHERE "+ table1 + "." + field1 +"="+ tblName +"."+ field +") AS " + field
                    subqwery = '(SELECT ({}) FROM {} WHERE {}.{}={}.{}) AS {}'\
                        .format(fieldReplace, subqwery, table1, field1, tblName, field, field)
                #print("---" + subqwery)
                qwery = qwery.replace(field, subqwery)
    #qwery = qwery.rstrip(',') + " FROM " + tblName + " LIMIT 10000"
    #qwery = qwery.rstrip(',') + " FROM " + tblName + " WHERE " + fieldName + '=' + fieldValue
    qwery = '{} FROM {} WHERE {}={}'.format(qwery.rstrip(','), tblName, fieldName, fieldValue)

    print(qwery)
    c.execute(qwery)
    return c.fetchall()

def selectRelationsDataFromDB(tblSearch, fieldName, fieldValue):
    global dbTablesStructList, c, db_type
    searchField = tblSearch + '.' + fieldName
    dataList = []
    for item in dbTablesStructList:
        tblName = item[0]
        for field in item[1]:
            field = field[0]
        for rel in item[2]:
            if rel[1][0] == searchField:
                data = selectDataFromDB(tblName, rel[0], fieldValue)
                dataList.append([tblName, data])
    return dataList

# Ищем название таблицы по её описанию
def getTablesNameOfDescription(tblDescr):
    global tblNamesList
    for i in tblNamesList:
        if i[1] == tblDescr:
            return(i[0])

# Ищем описание таблицы по её названию
def getTablesDescriptionOfName(tblName):
    global tblNamesList
    for i in tblNamesList:
        if i[0] == tblName:
            return(i[1])

# Удаление записи из БД
def deleteRecordsFromDB(tableName, valueList):
    global c, db_type, conn
    qwery = 'DELETE FROM {} WHERE '.format(tableName)
    #print(qwery)
    subQwery=''
    i = 0
    for item in getFields(tableName):
        print(item[0], item[1])
        if item[1] == 'integer':
            #subQwery = subQwery + item[0] + '=' + valueList[i] + ' AND '
            subQwery = '{}{}={} AND '.format(subQwery, item[0], valueList[i])
        else:
            #subQwery = subQwery + item[0] + '=\'' + valueList[i] + '\' AND '
            subQwery = "{}{}='{}' AND ".format(subQwery, item[0], valueList[i])
        i += 1
    qwery += subQwery
    qwery = qwery.rstrip('AND ').replace("='None'", " IS NULL")
    print(qwery)
    c.execute(qwery)
    conn.commit()
    return

#initDBstructure()
firstInit()
dbConnect()