Added correct HOME path into win.

This commit is contained in:
Sergey Kalinin
2017-03-20 12:38:14 +03:00
parent f9da7147d2
commit 07bb927271
2 changed files with 52 additions and 17 deletions

18
dm.py
View File

@@ -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():