Initial commit
This commit is contained in:
33
skills/qingyan-research/generate_html.py
Executable file
33
skills/qingyan-research/generate_html.py
Executable file
@@ -0,0 +1,33 @@
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import argparse
|
||||
|
||||
def safe_filename(title):
|
||||
# 移除非法文件名字符
|
||||
return re.sub(r'[\\/*?:"<>|]', "", title).strip()
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Save HTML content to a local file.")
|
||||
parser.add_argument("--title", required=True, help="Report title (used as filename)")
|
||||
args = parser.parse_args()
|
||||
|
||||
# 从标准输入 (STDIN) 读取所有内容,这不会受到 Shell 参数长度限制
|
||||
try:
|
||||
content = sys.stdin.read()
|
||||
if not content:
|
||||
print("Error: No content received from STDIN.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
filename = f"{safe_filename(args.title)}.html"
|
||||
|
||||
with open(filename, "w", encoding="utf-8") as f:
|
||||
f.write(content)
|
||||
|
||||
print(f"Successfully generated: {os.path.abspath(filename)}")
|
||||
except Exception as e:
|
||||
print(f"Error: {str(e)}", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user