104 lines
2.0 KiB
Python
104 lines
2.0 KiB
Python
"""
|
||
验证工具模块
|
||
提供各种数据验证功能
|
||
"""
|
||
import re
|
||
from typing import Optional
|
||
|
||
|
||
def validate_stock_code(code: str) -> bool:
|
||
"""
|
||
验证股票代码格式
|
||
|
||
A股代码格式：
|
||
- 上海：6开头，6位数字
|
||
- 深圳：0/3开头，6位数字
|
||
- 创业板：3开头，6位数字
|
||
- 科创板：688开头，6位数字
|
||
|
||
Args:
|
||
code: 股票代码
|
||
|
||
Returns:
|
||
是否有效
|
||
"""
|
||
if not code:
|
||
return False
|
||
|
||
# 移除可能的后缀（如.SH, .SZ）
|
||
code = code.split('.')[0]
|
||
|
||
# 检查是否为6位数字
|
||
if not re.match(r'^\d{6}$', code):
|
||
return False
|
||
|
||
# 检查首位数字
|
||
first_digit = code[0]
|
||
if first_digit in ['0', '3', '6']:
|
||
return True
|
||
|
||
# 检查科创板
|
||
if code.startswith('688'):
|
||
return True
|
||
|
||
return False
|
||
|
||
|
||
def normalize_stock_code(code: str) -> Optional[str]:
|
||
"""
|
||
标准化股票代码，添加市场后缀
|
||
|
||
Args:
|
||
code: 股票代码
|
||
|
||
Returns:
|
||
标准化后的代码（如600000.SH）或None
|
||
"""
|
||
if not validate_stock_code(code):
|
||
return None
|
||
|
||
# 移除已有后缀
|
||
code = code.split('.')[0]
|
||
|
||
# 添加市场后缀
|
||
if code.startswith('6'):
|
||
return f"{code}.SH" # 上海
|
||
elif code.startswith(('0', '3')):
|
||
return f"{code}.SZ" # 深圳
|
||
elif code.startswith('688'):
|
||
return f"{code}.SH" # 科创板
|
||
|
||
return None
|
||
|
||
|
||
def validate_date_format(date_str: str) -> bool:
|
||
"""
|
||
验证日期格式（YYYYMMDD）
|
||
|
||
Args:
|
||
date_str: 日期字符串
|
||
|
||
Returns:
|
||
是否有效
|
||
"""
|
||
if not date_str:
|
||
return False
|
||
|
||
# 检查格式
|
||
if not re.match(r'^\d{8}$', date_str):
|
||
return False
|
||
|
||
# 简单的日期范围检查
|
||
year = int(date_str[:4])
|
||
month = int(date_str[4:6])
|
||
day = int(date_str[6:8])
|
||
|
||
if year < 1990 or year > 2100:
|
||
return False
|
||
if month < 1 or month > 12:
|
||
return False
|
||
if day < 1 or day > 31:
|
||
return False
|
||
|
||
return True
|