Watch Star Fork

fasiondog / hikyuuPythonMIT

Open
#IE2SV linux下hku_config.py配置参数出错

pchaos  Created at

linux下hku_config.py 配置参数为环境变量,或者类似: ~/stock 情况时,会发生目录错乱的问题。 下面是我修改的版本

#!/usr/bin/python
# -*- coding: utf8 -*-
# cp936

#===============================================================================
# 作者:fasiondog
# 历史:1)20100220, Added by fasiondog
#===============================================================================

import os
import sys
import shutil

def assure_path_exists(path):
	dir = os.path.dirname(path)
	if not os.path.exists(dir):
		os.makedirs(dir)

usr_dir = os.path.expanduser('~')
hku_dir = usr_dir + '/.hikyuu'

if not os.path.lexists(hku_dir):
    os.mkdir(hku_dir)

def ask_question(question, default_anwser='N'):
    flag = False
    x = ''
    while not flag:
	x = input(question)
	if x == '':
	    x = default_anwser
	flag = True if (x=='' or x=='Y' or x=='y' or x=='N' or x=='n') else False
    return False if (x=='' or x=='N' or x=='n') else True

data_config_file = hku_dir + '/data_dir.ini'
if os.path.exists(data_config_file):
    os.remove(data_config_file);

data_directory = input("\n请输入股票数据存放路径(默认c:\stock):")
if data_directory == '':
    data_directory = 'c:\stock'

data_directory=os.path.expanduser(data_directory)

with open(data_config_file, 'w') as conf:
    conf.writelines(['[data_dir]\n', 'data_dir=' + data_directory])


import_config = """
;db - 指定基础信息数据库
;sql - 当指定的基础信息数据库不存在时,使用该指定的脚本创建数据库
[database]
db = {dir}\stock.db
sql = {dir}\createdb.sql

; ☆☆☆ 注:权息信息目前指支持钱龙数据格式
;sh、sz分别指定上证、深圳的权息数据所在目录
[weight]
;sh = c:/stock/weight/WEIGHT_sh/WEIGHT
;sz = c:/stock/weight/WEIGHT_sz/WEIGHT
sh = D:\LON\QLDATA\history\SHASE\weight
sz = D:\LON\QLDATA\history\SZNSE\weight

;指定K线数据源,格式为 “类型, 根目录”(类型仅能为 dzh 或 tdx,即大智慧、通达信)
;可多个数据源同时导入,防止信息遗漏
[from]
1 = tdx, {tdx}
;2 = dzh, d:\DZH5

;指定K线导入的目标文件
;选项“all”代表不管基础信息库中是否存在该股票,都将其K线导入
;如果选择all选择为true,将会导入很多不需要的K线数据,一般仅用于备份数据或将来使用
;all默认为false,如目前数据库仅配置导入A股、B股、基金,而不导入国债
[dest]
;all = True
sh_day = {dir}\sh_day.h5
sz_day = {dir}\sz_day.h5
sh_5min = {dir}\sh_5min.h5
sz_5min = {dir}\sz_5min.h5
sh_1min = {dir}\sh_1min.h5
sz_1min = {dir}\sz_1min.h5
"""

hikyuu_config = """
[tmpdir]
tmpdir = {dir}\\tmp

[logger]
properties = {dir}\\logger.properties

[block]
type  = qianlong
dir = {dir}\\block
指数板块 = zsbk.ini
行业板块 = hybk.ini
地域板块 = dybk.ini
概念板块 = gnbk.ini
self = self.ini

[preload]
day = 1
week = 0
month = 0
quarter = 0
halfyear = 0
year = 0

[baseinfo]
type = sqlite3
db = {dir}\\stock.db
;type = mysql
;host = 127.0.0.1
;port = 3306
;usr = root
;db = stk_base

[sh_day]
;type = tdx
;file = D:\\TdxW_HuaTai\\vipdoc
type = hdf5
file = c:\\stock\\sh_day.h5
;type = mysql
;host = 127.0.0.1
;port = 3306
;usr = root
;database = stk_sh_day

[sh_week]
type = hdf5
file = {dir}\\sh_day.h5

[sh_month]
type = hdf5
file = {dir}\\sh_day.h5

[sh_quarter]
type = hdf5
file = {dir}\\sh_day.h5

[sh_halfyear]
type = hdf5
file = {dir}\\sh_day.h5

[sh_year]
type = hdf5
file = {dir}\\sh_day.h5

[sh_1min]
type = hdf5
file = {dir}\\sh_1min.h5

[sh_5min]
type = hdf5
file = {dir}\\sh_5min.h5

[sh_15min]
type = hdf5
file = {dir}\\sh_5min.h5

[sh_30min]
type = hdf5
file = {dir}\\sh_5min.h5

[sh_60min]
type = hdf5
file = {dir}\\sh_5min.h5


;深圳证券
[sz_day]
type = hdf5
file = {dir}\\sz_day.h5
;type = mysql
;host = 127.0.0.1
;port = 3306
;usr = root
;database = stk_sz_day

[sz_week]
type = hdf5
file = {dir}\\sz_day.h5

[sz_month]
type = hdf5
file = {dir}\\sz_day.h5

[sz_quarter]
type = hdf5
file = {dir}\\sz_day.h5

[sz_halfyear]
type = hdf5
file = {dir}\\sz_day.h5

[sz_year]
type = hdf5
file = {dir}\\sz_day.h5

[sz_1min]
type = hdf5
file = {dir}\\sz_1min.h5

[sz_5min]
type = hdf5
file = {dir}\\sz_5min.h5

[sz_15min]
type = hdf5
file = {dir}\\sz_5min.h5

[sz_30min]
type = hdf5
file = {dir}\\sz_5min.h5

[sz_60min]
type = hdf5
file = {dir}\\sz_5min.h5
"""


if not os.path.lexists(data_directory):
    shutil.copytree('config', data_directory)
    os.remove(data_directory + '/__init__.py')

    if sys.platform == 'win32':
	filename = data_directory + '/hikyuu_win.ini'
    else:
	filename = data_directory + '/hikyuu_linux.ini'

    with open(filename, 'w') as f:
	f.write(hikyuu_config.format(dir=data_directory))

    tdx_dir = input("请输入通达信安装路径(默认从c:\TdxW_HuaTai):")
    if tdx_dir == '':
	tdx_dir = 'c:\TdxW_HuaTai'
    else:
	tdx_dir=os.path.expanduser(tdx_dir)

    with open(data_directory + '/importdata.ini', 'w') as f:
	f.write(import_config.format(dir=data_directory, tdx=tdx_dir))

    os.mkdir(data_directory + '/tmp')


else:
    if os.path.lexists(data_directory + '/block'):
	pass
	x = ask_question("block目录已存在是否覆盖?(Y/[N]):")
	if x:
	    shutil.rmtree(data_directory + '/block')
	    shutil.copytree('config/block', data_directory + '/block')
	    os.remove(data_directory + '/block/__init__.py')

    if os.path.exists(data_directory + '/logger.properties'):
	pass
	x = ask_question("logger.properties文件已存在是否覆盖?(Y/[N]):")
	if x:
	    os.remove(data_directory + '/logger.properties')
	    shutil.copy('config/logger.properties', data_directory + '/logger.properties')

    if os.path.exists(data_directory + '/importdata.ini'):
	x = ask_question("importdata.ini文件已存在是否覆盖?(Y/[N]):")
	if x:
	    os.remove(data_directory + '/importdata.ini')
	    tdx_dir = input("请输入通达信安装路径(默认从c:\TdxW_HuaTai):")
	    if tdx_dir == '':
		tdx_dir = 'c:\TdxW_HuaTai'
	    else:
		    tdx_dir=os.path.expanduser(tdx_dir)

	    with open(data_directory + '/importdata.ini', 'w') as f:
		f.write(import_config.format(dir=data_directory, tdx=tdx_dir))
    else:
	tdx_dir = input("请输入通达信安装路径(默认从c:\TdxW_HuaTai):")
	if tdx_dir == '':
	    tdx_dir = 'c:\TdxW_HuaTai'
	else:
		tdx_dir=os.path.expanduser(tdx_dir)

	with open(data_directory + '/importdata.ini', 'w') as f:
	    f.write(import_config.format(dir=data_directory, tdx=tdx_dir))

    if os.path.exists(data_directory + '/createdb.sql'):
	os.remove(data_directory + '/createdb.sql')
	shutil.copy('config/createdb.sql', data_directory + '/createdb.sql')

    if sys.platform == 'win32':
	hikyuu_ini = '/hikyuu_win.ini'
    else:
	hikyuu_ini = '/hikyuu_linux.ini'

    filename = data_directory + hikyuu_ini

    if os.path.exists(data_directory + hikyuu_ini):
	x = ask_question(hikyuu_ini[1:] + "文件已存在是否覆盖?(Y/[N]):")
	if x:
	    os.remove(filename)
	with open(data_directory + hikyuu_ini, 'w') as f:
	    f.write(hikyuu_config.format(dir=data_directory))
    else:
	with open(data_directory + hikyuu_ini, 'w') as f:
	    f.write(hikyuu_config.format(dir=data_directory))

#print(import_config.format(dir=data_directory))

3859_fasiondog total 2 participants
Show action logs Hide action logs
pchaos created 任务 2017-07-20 14:18

Comments (5)

3859_fasiondog
fasiondog 2017-07-21 00:46 owner

告诉我 linux 下,默认的tdx路径,比如"/home/usr/.wine/driver/c"(这个对不对?)

pchaos 2017-07-21 09:28

默认放在 /tmp/TDX

pchaos 2017-07-21 09:29

修改后的版本还是没有处理当输入路径为 : ~/stock 情况时,会发生目录错乱的问题 可以试试 设置stock路径为 $HOME/stock, 目录就错乱了。测试一下就知道我说的什么了。

输入的路径要用下面的函数 def assure_path_exists(path): dir = os.path.dirname(path) if not os.path.exists(dir): os.makedirs(dir) 处理一下,返回真实的路径

3859_fasiondog
fasiondog 2017-07-21 11:32 owner

你的意思是输入用~代表用户路径输入,明白了,我改的是不需要输入用户路径,自动加上用户路径,晚上再处理下

pchaos 2017-07-21 12:21

还是没完全理解,不光是用户路径,也可能是环境变量,例如:$HOME/yourDir , $tmp/yourdir。 你用函数 assure_path_exists(path)处理下就返回真实的路径了。 例如 tdxDir= $HOME/yourDir ,assure_path_exists(tdxDir)

Sign in and comment

Assignee
Labels
Not set
Milestone
Branch
Scheduled start
Not set
Scheduled end
Not set
Top level
Priority
5th_float_left_gitee 5th_float_left_close