做通达信自定义数据时，必用的6种Python代码

    def get_stock_type(code):        if code.startswith(('60', '68')):            return '1'        elif code.startswith(('30', '00')):            return '0'        elif code.startswith('92'):            return '2'        else:            return '3' # 3为其他的，不一定真存在先着

    def add_prefix(code):        code = code.strip()        if not code.isdigit():


    
            return None        if code.startswith(('60', '68')):            return f"sh{code}"        elif code.startswith(('30', '00')):            return f"sz{code}"        elif code.startswith('92'):            return f"bj{code}"        else:            return None

def extract_code(s):    """从 'sh600519' 或 '600519.sh' 中提取6位数字代码"""    if s[0].isalpha():          # 前缀格式，如 sh600519        return s[-6:]           # 取右边6个字符    elif s[-1].isalpha():       # 后缀格式，如 600519.sh        return s[:6]            # 取左边6个字符    else:        return s                # 已经是纯数字
# 测试codes = ["sh600519", "600519.sh"]for c in codes:    print(f"{c} -> {extract_code(c)}")  # 输出600519

# -*- coding: utf-8 -*-# @Author : 小红牛# 微信公众号：gxzfp888from datetime import datetime
def format_to_yyyymmdd(date_str, original_format):    """    将指定格式的日期字符串转换为 YYYYMMDD 格式。
    参数:        date_str: 日期字符串，如 '2026-05-06'        original_format: 原字符串的格式，如 '%Y-%m-%d'
    返回:        YYYYMMDD 格式的字符串，如 '20260506'    """    dt = datetime.strptime(date_str, original_format)    return dt.strftime('%Y%m%d')
# 示例：不同格式的日期转换examples = [    ("2026-5-6", "%Y-%m-%d"),          # 连字符分隔    ("2026/05/06", "%Y/%m/%d"),          # 斜线分隔    ("06 May 2026", "%d %b %Y"),         # 英文缩写月份    ("06 May 2026", "%d %B %Y"),         # 英文完整月份（这里May是缩写，实际用%h或%b）    ("20260506", "%Y%m%d"),              # 已经是目标格式    ("06-05-2026", "


    
%d-%m-%Y"),          # 日-月-年    ("2026年05月06日", "%Y年%m月%d日"),   # 中文格式]
for date_str, fmt in examples:    try:        result = format_to_yyyymmdd(date_str, fmt)        print(f"{date_str:20} -> {result}")    except ValueError as e:        print(f"{date_str:20} 解析失败：{e}")

2026-5-6           -> 202605062026/05/06           -> 2026050606 May 2026          -> 2026050606 May 2026          -> 2026050620260506             -> 2026050606-05-2026           -> 202605062026年05月06日        -> 20260506

value = 123456789  # 单位：元
# 转换为万元value_wan = value / 10000  # 12345.6789
# 四舍五入保留 2 位小数result = round(value_wan, 2)  # 12345.68
print(result)  # 输出 12345.68

 f"{float(value) / 10000:.2f}"

f"{stock_type}|{code}|{date_compact}|{amount:.2f}\n")

line1 = f"{market}|{code}|{qoq} 净利润：{profit + '万' if profit else ''} 营业收入{revenue + 


    
'万' if revenue else ''} {yoy} {date}"line1 = ' '.join(line1.split())  # 合并多余空格f"{line1}\n|0.000"

with open(sell_filename, 'w', encoding='utf-8') as f_sell:    for code, dt, amount in sell_results:        stock_type = self.get_stock_type(code)        f_sell.write(f"{stock_type}|{code}|{date_compact}|{amount:.0f}\n")