nber-cli 0.1.1__tar.gz → 0.1.2__tar.gz

{nber_cli-0.1.1 → nber_cli-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nber-cli
-Version: 0.1.1
+Version: 0.1.2
 Summary: A command-line tool to download NBER papers.
 License-File: LICENSE
 Requires-Python: >=3.11

{nber_cli-0.1.1 → nber_cli-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nber-cli"
-version = "0.1.1"
+version = "0.1.2"
 description = "A command-line tool to download NBER papers."
 readme = "README.md"
 requires-python = ">=3.11"

{nber_cli-0.1.1 → nber_cli-0.1.2}/src/nber_cli/downloader.py

@@ -125,16 +125,20 @@ async def update_paper_state(paper_id: str, status: str):
 
 async def download_paper(paper_id: str, save_path: str):
     """下载单个NBER论文"""
-    if await get_paper_state(paper_id) == 'ok':
-        logger.info(f"Skipping {paper_id}, already downloaded.")
-        return
-
-    url = f"https://www.nber.org/papers/{paper_id}.pdf"
     filepath = os.path.join(save_path, f"{paper_id}.pdf")
 
     # 确保保存路径存在
     os.makedirs(save_path, exist_ok=True)
 
+    state = await get_paper_state(paper_id)
+    if state == 'ok' and os.path.exists(filepath):
+        logger.info(f"Skipping {paper_id}, already downloaded.")
+        return
+    if state == 'ok' and not os.path.exists(filepath):
+        logger.info(f"{paper_id} marked as downloaded but file missing, re-downloading.")
+
+    url = f"https://www.nber.org/papers/{paper_id}.pdf"
+
     ua = UserAgent()
     headers = {'User-Agent': ua.random}
     retry_options = ExponentialRetry(attempts=MAX_RETRIES)

{nber_cli-0.1.1 → nber_cli-0.1.2}/uv.lock

@@ -311,7 +311,7 @@ wheels = [
 
 [[package]]
 name = "nber-cli"
-version = "0.1.1"
+version = "0.1.2"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },