镜像自地址
https://github.com/binary-husky/gpt_academic.git
已同步 2025-12-06 14:36:48 +00:00
68 行
2.2 KiB
Python
68 行
2.2 KiB
Python
from typing import List
|
|
from crazy_functions.review_fns.data_sources.base_source import PaperMetadata
|
|
|
|
class EndNoteFormatter:
|
|
"""EndNote参考文献格式生成器"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def create_document(self, papers: List[PaperMetadata]) -> str:
|
|
"""生成EndNote格式的参考文献文本
|
|
|
|
Args:
|
|
papers: 论文列表
|
|
|
|
Returns:
|
|
str: EndNote格式的参考文献文本
|
|
"""
|
|
endnote_text = ""
|
|
|
|
for paper in papers:
|
|
# 开始一个新条目
|
|
endnote_text += "%0 Journal Article\n" # 默认类型为期刊文章
|
|
|
|
# 根据venue_type调整条目类型
|
|
if hasattr(paper, 'venue_type') and paper.venue_type:
|
|
if paper.venue_type.lower() == 'conference':
|
|
endnote_text = endnote_text.replace("Journal Article", "Conference Paper")
|
|
elif paper.venue_type.lower() == 'preprint':
|
|
endnote_text = endnote_text.replace("Journal Article", "Electronic Article")
|
|
|
|
# 添加标题
|
|
endnote_text += f"%T {paper.title}\n"
|
|
|
|
# 添加作者
|
|
for author in paper.authors:
|
|
endnote_text += f"%A {author}\n"
|
|
|
|
# 添加年份
|
|
if paper.year:
|
|
endnote_text += f"%D {paper.year}\n"
|
|
|
|
# 添加期刊/会议名称
|
|
if hasattr(paper, 'venue_name') and paper.venue_name:
|
|
endnote_text += f"%J {paper.venue_name}\n"
|
|
elif paper.venue:
|
|
endnote_text += f"%J {paper.venue}\n"
|
|
|
|
# 添加DOI
|
|
if paper.doi:
|
|
endnote_text += f"%R {paper.doi}\n"
|
|
endnote_text += f"%U https://doi.org/{paper.doi}\n"
|
|
elif paper.url:
|
|
endnote_text += f"%U {paper.url}\n"
|
|
|
|
# 添加摘要
|
|
if paper.abstract:
|
|
endnote_text += f"%X {paper.abstract}\n"
|
|
|
|
# 添加机构
|
|
if hasattr(paper, 'institutions'):
|
|
for institution in paper.institutions:
|
|
endnote_text += f"%I {institution}\n"
|
|
|
|
# 条目之间添加空行
|
|
endnote_text += "\n"
|
|
|
|
return endnote_text |