用python实现,从数据库表code_list查出符合条件的code,组成code数组,如果code在文件目录stock_data的以日期和code为命名的json文件,就保留,否则删除其他文件
时间: 2024-10-18 16:15:51 浏览: 24
tushare.rar_python 抓取股票_tushare实时数据_tushare抓取股票_股票 python_获取美股数据
在Python中,可以使用`sqlite3`库连接SQLite数据库,`os`库遍历文件夹以及`json`库解析和读取JSON文件。首先,你需要连接到数据库,查询代码列表,然后检查每个代码对应的文件是否存在。以下是简化的示例代码:
```python
import sqlite3
import os
import json
# 连接到SQLite数据库
conn = sqlite3.connect('your_database.db')
cursor = conn.cursor()
# 查询code_list表中的数据
cursor.execute("SELECT code FROM code_list WHERE condition_column = 'your_condition'")
codes_to_check = [row[0] for row in cursor.fetchall()]
# 定义股票数据文件夹路径
data_folder = 'stock_data/'
# 构建结果代码数组
kept_codes = []
for code in codes_to_check:
# 文件名格式:date_code.json
file_name = f"{date_pattern}_{code}.json"
file_path = os.path.join(data_folder, file_name)
# 检查文件是否存在
if os.path.isfile(file_path):
kept_codes.append(code)
else:
# 如果文件不存在,则可以选择删除
# 删除操作(这里仅做打印确认,实际删除请注释掉 print)
print(f"File {file_name} does not exist, would delete.")
conn.close()
# 保存保留的代码数组
kept_codes_array = kept_codes
阅读全文