建议提交信息:
docs: 修订 AA 论文文字与格式,更新 .gitignore 并整理项目结构 - 修正 AA54562-25.tex 中的拼写错误和零宽字符 - 统一数学符号格式,去除段落多余前导空格和换行 - 更新 .gitignore 忽略 ADS 输出文件和 AI 工具目录
This commit is contained in:
@@ -0,0 +1,66 @@
|
||||
import ads
|
||||
import json
|
||||
import argparse
|
||||
import sys
|
||||
|
||||
# 如果你没有在环境变量里设置 ADS_DEV_KEY,将使用以下的硬编码 Token
|
||||
ads.config.token = "dpJWki7eHJ48TwlKz2AUyhXAxBgZrKo6AjE8hZwp"
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Search ADS and return metadata")
|
||||
parser.add_argument("--query", required=True, help="ADS Search Query")
|
||||
parser.add_argument("--output", required=True, help="Output JSON file path")
|
||||
parser.add_argument("--rows", type=int, default=10, help="Number of rows to return")
|
||||
parser.add_argument("--year_range", help="Year range to filter, e.g. 2018-2023 or 2020")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
print(f"Searching ADS for query: {args.query}")
|
||||
query_params = {
|
||||
"q": args.query,
|
||||
"rows": args.rows,
|
||||
"fl": ["bibcode", "title", "author", "year", "abstract", "citation_count", "reference_count", "pub", "doi"]
|
||||
}
|
||||
|
||||
if args.year_range:
|
||||
if '-' in args.year_range:
|
||||
start_year, end_year = args.year_range.split('-')
|
||||
query_params["fq"] = f"year:[{start_year} TO {end_year}]"
|
||||
else:
|
||||
query_params["fq"] = f"year:{args.year_range}"
|
||||
|
||||
try:
|
||||
papers = list(ads.SearchQuery(**query_params))
|
||||
results = []
|
||||
for p in papers:
|
||||
record = {
|
||||
"bibcode": getattr(p, "bibcode", "") or "",
|
||||
"title": getattr(p, "title", [""])[0] if getattr(p, "title", None) else "",
|
||||
"author": getattr(p, "author", []),
|
||||
"year": getattr(p, "year", "") or "",
|
||||
"abstract": getattr(p, "abstract", "") or "",
|
||||
"citation_count": getattr(p, "citation_count", 0) or 0,
|
||||
"reference_count": getattr(p, "reference_count", 0) or 0,
|
||||
"pub": getattr(p, "pub", "") or "",
|
||||
"doi": getattr(p, "doi", [""])[0] if getattr(p, "doi", None) else ""
|
||||
}
|
||||
results.append(record)
|
||||
|
||||
with open(args.output, "w", encoding="utf-8") as f:
|
||||
json.dump(results, f, ensure_ascii=False, indent=2)
|
||||
|
||||
print(f"Found {len(results)} papers. Saved metadata to {args.output}.")
|
||||
|
||||
# 打印简单摘要到终端
|
||||
for i, r in enumerate(results[:5]):
|
||||
print(f"\n[{i+1}] {r['title']} ({r['year']})")
|
||||
print(f" Bibcode: {r['bibcode']} | Citations: {r['citation_count']}")
|
||||
authors = ", ".join(r['author'][:3]) + (" et al." if len(r['author']) > 3 else "")
|
||||
print(f" Authors: {authors}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"Query Failed: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user