#!/usr/bin/env python3
import os

file_path = '/app/qlib/source/qlib/__init__.py'

# 读取文件
with open(file_path, 'r') as f:
    lines = f.readlines()

# 修改内容
new_lines = []
skip_next = False
for i, line in enumerate(lines):
    if 'from setuptools_scm import get_version' in line:
        new_lines.append('try:\n')
        new_lines.append('    from setuptools_scm import get_version\n')
        skip_next = True
    elif skip_next and '__version__ = get_version' in line:
        new_lines.append('    ' + line)
        new_lines.append('except LookupError:\n')
        new_lines.append('    __version__ = "0.9.0"\n')
        skip_next = False
    else:
        new_lines.append(line)

# 写回文件
with open(file_path, 'w') as f:
    f.writelines(new_lines)

print("Fixed qlib version detection")