使用pyserpZotero与SERPAPI高效获取Google学术引用教程

pip install pyserpZotero box pyyaml

API_KEY: "您的SERPAPI_KEY"
ZOT_ID: "您的ZOTERO_USER_ID"
ZOT_KEY: "您的ZOTERO_API_KEY"

from box import Box
import pyserpZotero
import yaml

# 导入凭证
with open("config.yaml", "r") as ymlfile:
    cfg = Box(yaml.safe_load(ymlfile), default_box=True, default_box_attr=None)

API_KEY = cfg.API_KEY
ZOT_ID   = cfg.ZOT_ID
ZOT_KEY = cfg.ZOT_KEY

# 实例化 serpZot 对象
citeObj = pyserpZotero.serpZot(API_KEY = API_KEY,
                               ZOT_ID  = ZOT_ID,
                               ZOT_KEY = ZOT_KEY)

TERMS = ['reinforcement learning', 'traveling salesman', 'nowcasting', 'propensity score'] # 搜索关键词列表
MIN_YEAR = "2010" # 搜索结果的最早年份
SAVE_BIB = False  # 是否将结果保存为Bibtex文件
USE_ZOT  = True   # 是否将结果上传到Zotero
CLEAN    = False  # 是否尝试清理破损的LaTeX和格式

for i in range(len(TERMS)):
    print(f"正在搜索: {TERMS[i]}")
    # 执行搜索，并将结果存储在 citeObj 内部
    citeObj.searchScholar(TERM     = TERMS[i],
                          MIN_YEAR = MIN_YEAR,
                          SAVE_BIB = SAVE_BIB)
    print("搜索完成，准备上传到Zotero...")
    # 上传解析后的结果到Zotero
    citeObj.search2Zotero()
    print("已成功上传到Zotero（如果USE_ZOT为True）")

if CLEAN:
    print("正在尝试清理Zotero中的LaTeX格式...")
    citeObj.cleanZot(ZOT_ID      = ZOT_ID,
                     ZOT_KEY     = ZOT_KEY,
                     SEARCH_TERM = "\") # 可选参数，默认为所有条目，这里搜索包含反斜杠的条目
    print("LaTeX格式清理完成。")

print("正在检查arXiv并下载免费PDF...")
citeObj.arxivDownload()
print("arXiv PDF下载及上传完成。")

#### 配置搜索参数
TERMS = ['reinforcement learning', 'traveling salesman', 'nowcasting', 'propensity score'] # 搜索关键词
MIN_YEAR = "2010" # 最早年份
SAVE_BIB = False  # 是否保存Bibtex文件
USE_ZOT  = True   # 是否上传到Zotero
CLEAN    = False  # 是否清理LaTeX格式

#### 导入必要的库
from box import Box
import pyserpZotero
import yaml

#### 从YAML文件导入API凭证
with open("config.yaml", "r") as ymlfile:
    cfg = Box(yaml.safe_load(ymlfile), default_box=True, default_box_attr=None)

API_KEY = cfg.API_KEY
ZOT_ID  = cfg.ZOT_ID
ZOT_KEY = cfg.ZOT_KEY

#### 实例化 serpZot 对象
citeObj = pyserpZotero.serpZot(API_KEY  = API_KEY,
                               ZOT_ID   = ZOT_ID,
                               ZOT_KEY  = ZOT_KEY)

#### 执行搜索并上传结果到Zotero
for i in range(len(TERMS)):
    print(f"--- 开始处理关键词: {TERMS[i]} ---")
    # 调用 searchScholar 方法执行 Google 学术搜索
    # 结果会被内部存储，并根据 SAVE_BIB 决定是否保存为 .bib 文件
    citeObj.searchScholar(TERM     = TERMS[i],
                                MIN_YEAR = MIN_YEAR,
                                SAVE_BIB = SAVE_BIB)
    print(f"关键词 '{TERMS[i]}' 搜索完成。")

    # 如果 USE_ZOT 为 True，将搜索结果上传到 Zotero
    if USE_ZOT:
        citeObj.search2Zotero()
        print(f"关键词 '{TERMS[i]}' 的结果已上传到 Zotero。")
    else:
        print(f"关键词 '{TERMS[i]}' 的结果未上传到 Zotero (USE_ZOT=False)。")
    print("--- 关键词处理结束 ---")


#### 清理Zotero中的LaTeX格式 (如果 CLEAN 为 True)
if CLEAN:
    print("
--- 正在尝试清理Zotero中的LaTeX格式 ---")
    citeObj.cleanZot(ZOT_ID      = ZOT_ID,
                     ZOT_KEY     = ZOT_KEY,
                     SEARCH_TERM = "\") # 可选参数，用于指定要清理的条目，这里清理所有包含反斜杠的条目
    print("LaTeX格式清理完成。")
else:
    print("
未执行LaTeX格式清理 (CLEAN=False)。")

#### 检查arXiv并下载免费PDF (如果可用)
print("
--- 正在检查arXiv并下载免费PDF ---")
citeObj.arxivDownload()
print("arXiv PDF下载及上传完成。")

print("
所有操作执行完毕。")