You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
2.0 KiB
64 lines
2.0 KiB
# -*- coding: utf-8 -*-
|
|
# tzquant
|
|
|
|
import os
|
|
import shutil
|
|
import numpy as np
|
|
|
|
# 定义要移动的目录列表
|
|
directories_to_move = ['tz_ctastrategy', 'tzquant', 'tzqtdatahub', 'tz_riskmanager', 'gate_api', 'clients']
|
|
|
|
# 获取numpy模块的__file__属性，它包含了numpy模块文件的路径
|
|
numpy_file_path = np.__file__
|
|
|
|
# 使用os.path模块来获取numpy所在的目录
|
|
numpy_dir = os.path.dirname(numpy_file_path)
|
|
|
|
# 获取numpy所在目录的上层目录
|
|
parent_dir = os.path.dirname(numpy_dir)
|
|
|
|
# 打印上层目录，用于验证
|
|
print("NumPy所在目录的上层目录是: ", parent_dir)
|
|
|
|
# 确保parent_dir已经定义，并准备移动目录
|
|
for directory in directories_to_move:
|
|
# 构建源目录和目标目录的路径
|
|
source_path = os.path.join(os.getcwd(), directory)
|
|
target_path = os.path.join(parent_dir, directory)
|
|
|
|
# 检查源目录是否存在
|
|
if os.path.isdir(source_path):
|
|
# 使用shutil.move来移动目录
|
|
if os.path.exists(target_path):
|
|
if os.path.isdir(target_path):
|
|
shutil.rmtree(target_path) # 删除整个目录及其内容
|
|
else:
|
|
os.remove(target_path) # 删除文件
|
|
shutil.move(source_path, target_path)
|
|
print(f"Moved {directory} to {target_path}")
|
|
else:
|
|
print(f"Directory {directory} does not exist in the current directory.")
|
|
|
|
# 移动完成后，可以打印确认信息
|
|
print("All directories have been moved to the parent directory of NumPy installation.")
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import pandas as pd
|
|
from datetime import datetime
|
|
|
|
from tz_ctastrategy import (
|
|
ArrayManager,
|
|
BarData,
|
|
BarGenerator,
|
|
CtaTemplate,
|
|
OrderData,
|
|
StopOrder,
|
|
TickData,
|
|
TradeData,
|
|
)
|
|
from tz_ctastrategy.backtesting_mul import BacktestingEngine
|
|
from tz_ctastrategy.template import CtaTemplate
|
|
from tzquant.trader.constant import OrderType
|
|
from tzquant.trader.object import *
|