import_joinquant.py

# -*- coding: utf-8 -*-

import os
import platform
import migrate

base_path = os.path.dirname(os.path.abspath(__file__))
data_path = os.path.join(base_path, "data", "joinquant")
backup_path = os.path.join(base_path, "backup", "joinquant")

print(data_path)

is_windows = platform.system() == 'Windows'

command_pre = '''mysql -uquant -p123456 -h127.0.0.1 quant --local-infile=1 -e 'load data local infile "'''
command_suffix = '''" into table joinquant_stock_hs(
    `code`,
    `biz_date`,
    `pe`,
    `pb`,
    `ps`,
    `pcf`)' '''

if is_windows:
    command_pre = '''mysql -uquant -p123456 -h127.0.0.1 quant --local-infile=1 -e "load data local infile \''''
    command_suffix = '''\' into table joinquant_stock_hs(`code`, `biz_date`, `pe`, `pb`, `ps`, `pcf`)" '''

for file_name in os.listdir(data_path):
    file = os.path.join(data_path, file_name)
    if is_windows:
        file = file.replace('\\', '\\\\')
    command = command_pre + file + command_suffix
    os.system(command)
    dest = os.path.join(backup_path, file_name)
    migrate.move(file, dest)