127 lines
3.7 KiB
Python
127 lines
3.7 KiB
Python
|
import json
|
|||
|
|
import sqlite3
|
|||
|
|
import os
|
|||
|
|
from datetime import datetime, timedelta
|
|||
|
|
from SendBot import SendToFeishu
|
|||
|
|
|
|||
|
|
def create_database():
|
|||
|
|
conn = sqlite3.connect('./db/qianxin.db')
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS articles (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
title TEXT,
|
|||
|
|
guid TEXT,
|
|||
|
|
source TEXT,
|
|||
|
|
description TEXT,
|
|||
|
|
pubDate DATETIME
|
|||
|
|
)''')
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
def insert_data(data):
|
|||
|
|
conn = sqlite3.connect('./db/qianxin.db')
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
for entry in data:
|
|||
|
|
cursor.execute('''
|
|||
|
|
INSERT INTO articles (title, guid, source, description, pubDate)
|
|||
|
|
VALUES (?, ?, ?, ?, ?)
|
|||
|
|
''', (entry['title'], entry['guid'], entry['source'], entry['description'], entry['pubDate']))
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
def get_qianxin_json():
|
|||
|
|
# 检查文件是否存在
|
|||
|
|
if not os.path.exists('./JSON/qianxin.json'):
|
|||
|
|
raise FileNotFoundError(f"qianxin.json文件不存在，请检查程序是否运行正常！")
|
|||
|
|
|
|||
|
|
# 打开并读取JSON文件
|
|||
|
|
with open('./JSON/qianxin.json', 'r', encoding='utf-8') as file:
|
|||
|
|
data = json.load(file)
|
|||
|
|
|
|||
|
|
# 假设data是一个包含多个JSON对象的列表
|
|||
|
|
if not isinstance(data, list):
|
|||
|
|
raise ValueError("JSON文件格式错误，请检查common.py是否异常！")
|
|||
|
|
|
|||
|
|
# 提取所需字段并编号
|
|||
|
|
total_data = []
|
|||
|
|
for index, item in enumerate(data, start=1):
|
|||
|
|
entry = {
|
|||
|
|
"id": index,
|
|||
|
|
"title": item.get("title", ""),
|
|||
|
|
"guid": item.get("guid", ""),
|
|||
|
|
"description": item.get("description", ""),
|
|||
|
|
"pubDate": item.get("pubDate", ""),
|
|||
|
|
"source": item.get("source", "")
|
|||
|
|
}
|
|||
|
|
total_data.append(entry)
|
|||
|
|
|
|||
|
|
return total_data
|
|||
|
|
|
|||
|
|
def select_articles(e_hour):
|
|||
|
|
conn = sqlite3.connect('./db/qianxin.db')
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 获取当前日期和时间
|
|||
|
|
now = datetime.now()
|
|||
|
|
start_time = now - timedelta(hours=e_hour)
|
|||
|
|
end_time = now
|
|||
|
|
|
|||
|
|
# 查询指定时间段内的数据
|
|||
|
|
cursor.execute('''
|
|||
|
|
SELECT * FROM articles
|
|||
|
|
WHERE pubDate BETWEEN ? AND ?
|
|||
|
|
''', (start_time.strftime('%Y-%m-%d %H:%M:%S'), end_time.strftime('%Y-%m-%d %H:%M:%S')))
|
|||
|
|
|
|||
|
|
results = cursor.fetchall()
|
|||
|
|
conn.close()
|
|||
|
|
return results
|
|||
|
|
|
|||
|
|
def clear_table():
|
|||
|
|
conn = sqlite3.connect('./db/qianxin.db')
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
cursor.execute('DELETE FROM articles')
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
def get_filtered_articles(entries):
|
|||
|
|
result = ""
|
|||
|
|
for entry in entries:
|
|||
|
|
result += f"来源：{entry[3]}\t文章：{entry[1]}\n"
|
|||
|
|
result += f"链接：{entry[2]}\t上传时间：{entry[5]}\n"
|
|||
|
|
result += f"描述：{entry[4]}\n"
|
|||
|
|
result += "-" * 40 + "\n" # 添加分隔线以便区分不同文章
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
def Src_qianxin(e_hour):
|
|||
|
|
if not os.path.exists('./db/qianxin.db'):
|
|||
|
|
# 创建数据库和表
|
|||
|
|
create_database()
|
|||
|
|
|
|||
|
|
# 清空表
|
|||
|
|
clear_table()
|
|||
|
|
|
|||
|
|
# 获取 JSON 数据
|
|||
|
|
M_qianxin_data = get_qianxin_json()
|
|||
|
|
|
|||
|
|
# 插入数据到数据库
|
|||
|
|
insert_data(M_qianxin_data)
|
|||
|
|
|
|||
|
|
# 查询指定时间段内的数据
|
|||
|
|
filtered_articles = select_articles(e_hour)
|
|||
|
|
# print(filtered_articles)
|
|||
|
|
|
|||
|
|
if filtered_articles:
|
|||
|
|
results = get_filtered_articles(filtered_articles)
|
|||
|
|
print("奇安信攻防社区资讯递送中：")
|
|||
|
|
SendToFeishu(results, "奇安信攻防社区资讯递送")
|
|||
|
|
print("-" * 40 + "\n")
|
|||
|
|
# print(results)
|
|||
|
|
else:
|
|||
|
|
# 如果为空，则跳过执行
|
|||
|
|
print("奇安信攻防社区数据为空，跳过执行。")
|
|||
|
|
# print(results)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
Src_qianxin(4)
|