shiyue/stock-scanner
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/Cassianvale/stock-scanner

Browse Source
This commit is contained in:
Cassianvale
2025-03-04 11:55:47 +08:00
parent 40787aa85f 3195e579d1
commit dabca88a75
4 changed files with 24 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
README.md
View File

@@ -18,6 +18,21 @@ docker run -d \
-e API_URL=替换为你的api地址 \
-e API_MODEL=替换为你的模型 \
lanzhihong/stock-scanner:latest
注意⚠️： 环境变量名变更，更新版本后需要调整！！！
针对API_URL处理兼容更多的api地址
API_URL 处理逻辑说明：
1. 当 API_URL 以 / 结尾时直接追加 chat/completions，保留原有版本号：
示例：
输入: https://ark.cn-beijing.volces.com/api/v3/
输出: https://ark.cn-beijing.volces.com/api/v3/chat/completions
2. 当 API_URL 不以 / 结尾时使用默认版本号 v1：
示例：
输入: https://ark.cn-beijing.volces.com/api
输出: https://ark.cn-beijing.volces.com/api/v1/chat/completions
```
默认8888端口，部署完成后访问 http://127.0.0.1:8888 即可使用。

18
stock_analyzer.py
View File

@@ -5,14 +5,9 @@ import os
import requests
from typing import Dict, List, Optional, Tuple
from dotenv import load_dotenv
import logging
class StockAnalyzer:
def __init__(self, initial_cash=1000000):
# 设置日志
logging.basicConfig(level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s')
self.logger = logging.getLogger(__name__)
# 加载环境变量
load_dotenv()
@@ -93,7 +88,6 @@ class StockAnalyzer:
return df.sort_values('date')
except Exception as e:
self.logger.error(f"获取股票数据失败: {str(e)}")
raise Exception(f"获取股票数据失败: {str(e)}")
def calculate_ema(self, series, period):
@@ -173,7 +167,7 @@ class StockAnalyzer:
return df
except Exception as e:
self.logger.error(f"计算技术指标时出错: {str(e)}")
print(f"计算技术指标时出错: {str(e)}")
raise
def calculate_score(self, df):
@@ -207,7 +201,7 @@ class StockAnalyzer:
return score
except Exception as e:
self.logger.error(f"计算评分时出错: {str(e)}")
print(f"计算评分时出错: {str(e)}")
raise
def get_ai_analysis(self, df, stock_code):
@@ -259,11 +253,11 @@ class StockAnalyzer:
if response.status_code == 200:
return response.json()['choices'][0]['message']['content']
else:
self.logger.error(f"API 错误: {response.status_code} - {response.text}")
print(f"API 错误: {response.status_code} - {response.text}")
return f"AI 分析暂时无法使用 (HTTP {response.status_code})"
except Exception as e:
self.logger.error(f"AI 分析发生错误: {str(e)}")
print(f"AI 分析发生错误: {str(e)}")
return f"AI 分析过程中发生错误: {str(e)}"
def get_recommendation(self, score):
@@ -313,7 +307,7 @@ class StockAnalyzer:
return report
except Exception as e:
self.logger.error(f"分析股票时出错: {str(e)}")
print(f"分析股票时出错: {str(e)}")
raise
def scan_market(self, stock_list, min_score=60, market_type='A'):
@@ -326,7 +320,7 @@ class StockAnalyzer:
if report['score'] >= min_score:
recommendations.append(report)
except Exception as e:
self.logger.error(f"分析股票 {stock_code} 时出错: {str(e)}")
print(f"分析股票 {stock_code} 时出错: {str(e)}")
continue
# 按得分排序

6
us_stock_service.py
View File

@@ -1,12 +1,7 @@
import akshare as ak
import pandas as pd
import logging
class USStockService:
def __init__(self):
logging.basicConfig(level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s')
self.logger = logging.getLogger(__name__)
def search_us_stocks(self, keyword):
"""
@@ -55,5 +50,4 @@ class USStockService:
return formatted_results
except Exception as e:
self.logger.error(f"搜索美股代码时出错: {str(e)}")
raise Exception(f"搜索美股代码失败: {str(e)}")

19
web_server.py
View File

@@ -2,26 +2,14 @@ from flask import Flask, render_template, request, jsonify
from stock_analyzer import StockAnalyzer
from us_stock_service import USStockService
import threading
import logging
from logging.handlers import RotatingFileHandler
import traceback
import os
app = Flask(__name__)
analyzer = StockAnalyzer()
us_stock_service = USStockService()
# 配置日志
logging.basicConfig(level=logging.INFO)
handler = RotatingFileHandler('flask_app.log', maxBytes=10000000, backupCount=5)
handler.setFormatter(logging.Formatter(
'[%(asctime)s] %(levelname)s in %(module)s: %(message)s'
))
app.logger.addHandler(handler)
@app.route('/')
def index():
# 如果环境变量不存在或为空，返回 None
announcement = os.getenv('ANNOUNCEMENT_TEXT') or None
return render_template('index.html', announcement=announcement)
@@ -50,8 +38,7 @@ def analyze():
return jsonify({'results': results})
except Exception as e:
app.logger.error(f"处理请求失败: {str(e)}")
app.logger.error(f"详细错误: {traceback.format_exc()}")
print(f"分析股票时出错: {str(e)}")
return jsonify({'error': str(e)}), 500
@app.route('/search_us_stocks', methods=['GET'])
@@ -65,11 +52,11 @@ def search_us_stocks():
return jsonify({'results': results})
except Exception as e:
app.logger.error(f"搜索美股代码时出错: {str(e)}")
print(f"搜索美股代码时出错: {str(e)}")
return jsonify({'error': str(e)}), 500
if __name__ == '__main__':
# 将 host 设置为 '0.0.0.0' 使其支持所有网络接口访问
app.run(host='0.0.0.0', port=8888, debug=True)