Added correct HOME path into win.
This commit is contained in:
18
dm.py
18
dm.py
@@ -1,20 +1,21 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import json, os, configparser, shutil
|
||||
|
||||
print(os.getenv("HOME"))
|
||||
|
||||
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")
|
||||
# определим каталог для конфигруции и создадим если его нет
|
||||
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')
|
||||
print(cfg_file)
|
||||
# создадим файл конфигурации
|
||||
config.add_section('DataBase')
|
||||
config.set('DataBase', 'db_type', 'mysql')
|
||||
@@ -44,8 +45,8 @@ def firstInit():
|
||||
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')
|
||||
db_user = config.get('DataBase', 'db_user')
|
||||
db_password = config.get('DataBase', 'db_password')
|
||||
|
||||
# Создаём нужные каталоги
|
||||
if os.path.isdir(work_dir):
|
||||
@@ -66,8 +67,6 @@ def firstInit():
|
||||
|
||||
def dbConnect():
|
||||
global c, db_type, db_hostname, db_user, db_password, db_name
|
||||
print(db_name)
|
||||
print(db_type)
|
||||
if db_type == "mysql":
|
||||
import pymysql
|
||||
conn = pymysql.connect(
|
||||
@@ -78,7 +77,6 @@ def dbConnect():
|
||||
charset='utf8')
|
||||
elif db_type == "sqlite":
|
||||
import sqlite3
|
||||
print(db_name)
|
||||
conn = sqlite3.connect(db_name)
|
||||
else:
|
||||
print(db_type + " database type does`t support")
|
||||
@@ -137,7 +135,7 @@ def createTables(tbl_list):
|
||||
tbl_descr_list.append(one_Table_descr)
|
||||
tbl_struct_list.append(one_Table_struct)
|
||||
print(qwery_create)
|
||||
#c.execute(qwery_create)
|
||||
c.execute(qwery_create)
|
||||
return tbl_names_list
|
||||
|
||||
def initDBstructure():
|
||||
|
Reference in New Issue
Block a user