crossref-local 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

crossref_local/_server/routes_works.py

@@ -1,23 +1,49 @@
 """Work search and retrieval endpoints."""
 
 import time
-from typing import Optional
+from typing import Optional, Dict
 
 from fastapi import APIRouter, Query, HTTPException
 
 from .._core import fts
 from .._core.db import get_db
 from .._core.models import Work
-from .models import WorkResponse, SearchResponse, BatchRequest, BatchResponse
+from .models import (
+    BatchRequest,
+    BatchResponse,
+    SearchResponse,
+    WorkResponse,
+)
 
 router = APIRouter(tags=["works"])
 
+# IF cache for performance
+_if_cache: Dict[str, Optional[float]] = {}
+
+
+def _get_impact_factor(db, issn: str) -> Optional[float]:
+    """Get impact factor from journals_openalex table."""
+    if not issn:
+        return None
+    if issn in _if_cache:
+        return _if_cache[issn]
+    try:
+        row = db.fetchone(
+            "SELECT two_year_mean_citedness FROM journals_openalex WHERE issns LIKE ?",
+            (f"%{issn}%",),
+        )
+        _if_cache[issn] = row["two_year_mean_citedness"] if row else None
+    except Exception:
+        _if_cache[issn] = None
+    return _if_cache[issn]
+
 
 @router.get("/works", response_model=SearchResponse)
 def search_works(
     q: str = Query(..., description="Search query (FTS5 syntax supported)"),
-    limit: int = Query(10, ge=1, le=100, description="Max results"),
+    limit: int = Query(10, ge=1, description="Max results"),
     offset: int = Query(0, ge=0, description="Skip first N results"),
+    with_if: bool = Query(False, description="Include impact factor (OpenAlex)"),
 ):
     """
     Full-text search across works.
@@ -28,7 +54,7 @@ def search_works(
     Examples:
         /works?q=machine learning
         /works?q="neural network" AND hippocampus
-        /works?q=CRISPR&limit=20
+        /works?q=CRISPR&limit=20&with_if=true
     """
     start = time.perf_counter()
 
@@ -37,14 +63,13 @@ def search_works(
     except Exception as e:
         raise HTTPException(status_code=400, detail=f"Search error: {e}")
 
-    elapsed_ms = (time.perf_counter() - start) * 1000
+    # Get IF if requested
+    db = get_db() if with_if else None
 
-    return SearchResponse(
-        query=q,
-        total=results.total,
-        returned=len(results.works),
-        elapsed_ms=round(elapsed_ms, 2),
-        results=[
+    work_responses = []
+    for w in results.works:
+        if_val = _get_impact_factor(db, w.issn) if db and w.issn else None
+        work_responses.append(
             WorkResponse(
                 doi=w.doi,
                 title=w.title,
@@ -57,9 +82,34 @@ def search_works(
                 page=w.page,
                 abstract=w.abstract,
                 citation_count=w.citation_count,
+                impact_factor=if_val,
+                impact_factor_source="OpenAlex" if if_val else None,
             )
-            for w in results.works
-        ],
+        )
+
+    elapsed_ms = (time.perf_counter() - start) * 1000
+
+    # Build limit_info from search result
+    limit_info = None
+    if results.limit_info:
+        from .models import LimitInfoResponse
+
+        limit_info = LimitInfoResponse(
+            requested=results.limit_info.requested,
+            returned=results.limit_info.returned,
+            total_available=results.limit_info.total_available,
+            capped=results.limit_info.capped,
+            capped_reason=results.limit_info.capped_reason,
+            stage=results.limit_info.stage,
+        )
+
+    return SearchResponse(
+        query=q,
+        total=results.total,
+        returned=len(results.works),
+        elapsed_ms=round(elapsed_ms, 2),
+        results=work_responses,
+        limit_info=limit_info,
     )
 
 

{crossref_local-0.5.0.dist-info → crossref_local-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: crossref-local
-Version: 0.5.0
+Version: 0.5.1
 Summary: Local CrossRef database with 167M+ works and full-text search
 Project-URL: Homepage, https://github.com/ywatanabe1989/crossref_local
 Project-URL: Repository, https://github.com/ywatanabe1989/crossref_local

{crossref_local-0.5.0.dist-info → crossref_local-0.5.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-crossref_local/__init__.py,sha256=bOn6qJob4okwkLzzIoEGQj0I-amIBSr89iWrbkYEJns,3691
+crossref_local/__init__.py,sha256=j2HyOmjYFUjiywoLdIgYVqgXuGx9eWjcy4x7VBA4vf0,3792
 crossref_local/__main__.py,sha256=N1c1ESGgJkAwsSWXANUgmzxC1OJEIqw-cl9m4pmNP7s,110
 crossref_local/aio.py,sha256=_vDHFoE2n6GqO3j6pGeCaJVVe5I5L3M0tuMlW8IgGx8,1255
 crossref_local/cache.py,sha256=yQpnRRBsHx2pluIPPfJlrakeuCC8TysKXrfA3ohdxSg,13321
@@ -15,33 +15,35 @@ crossref_local/_cache/utils.py,sha256=D3sg7MoXZs4aC2OvagVgmOVMZjm1Xqy1ulGbs0_F91
 crossref_local/_cache/viz.py,sha256=0VAHK-i1xR5StEYrJ_IAs0HSk0MQNVFi4EjPtlZEwTY,8498
 crossref_local/_cli/__init__.py,sha256=NS07Eo93dRAuO5vxGwGwS0ctvZxMMkF_py09bzIk3Hk,175
 crossref_local/_cli/cache.py,sha256=8QykJ7-H313x7oKRw5fJ9Coay27Y8uBWGx5JR1VQ1_Y,6220
-crossref_local/_cli/cli.py,sha256=w7rJ4ZtetSr6TRz2f2vyHqZylq5vXM8O9KB61gScbFM,16903
+crossref_local/_cli/cli.py,sha256=TFhj7BjSHxs3rWyOifpi39z6-YmHIHt5Jz-cZNtVtEg,12425
 crossref_local/_cli/completion.py,sha256=yzqwMbGbbqXJQHdztuaqji-AdFTd8zDDhHeNGdw4NpU,7341
 crossref_local/_cli/main.py,sha256=NgCGB5-ThofTRPoxxwZpsfCQTo5WIw8oP7zBSKhEiDQ,444
 crossref_local/_cli/mcp.py,sha256=yjUoq3hvu2OofYf_D29_NXoTvkhxz5QbWAuWv6U-H8U,11427
-crossref_local/_cli/mcp_server.py,sha256=jZJwhiTavgTHcJctXpafPgy3aIT49-rdIZ-4i88myeU,10989
-crossref_local/_core/__init__.py,sha256=6xKO9y3VOuTF1f83uyQ48HWhszYpPfWZj94t0q0rEgk,940
-crossref_local/_core/api.py,sha256=YogfhBHDIqT-hdEI-iDe_-py13Vp6F4GW8q0psLDRyQ,7781
+crossref_local/_cli/mcp_server.py,sha256=mplCUBph9WM_Eq7b-MXgT-vvUH9hf4h_eeHUHTQ7f5c,12526
+crossref_local/_cli/search.py,sha256=bqsWn4nCCW_Glgb7dSXsntP4P5nHQJJCSgehvnV9Bbc,6327
+crossref_local/_core/__init__.py,sha256=xfun82MI9-SuaBqX0zB5liiLmj3AkY5TlksVYY6IaOg,1034
+crossref_local/_core/api.py,sha256=fYkr2VkiAQur3fdysIXP57qk0njWBf9ngg6dt5t63Do,7880
 crossref_local/_core/citations.py,sha256=4ERCTVkZu6kesZRazaWcTpUUeYz0rs97NwFZAYcLdKA,12765
 crossref_local/_core/config.py,sha256=T-2EX4EDoCLaOAmViBKGQuNvoJP8x7cvZOvmAcNrQms,5475
 crossref_local/_core/db.py,sha256=KLlcdNWLnKSexc0cuRZGsBLmJUnXeEaD_CBcXhbPXLg,3832
-crossref_local/_core/fts.py,sha256=Ti1EdSsrIQMDFTynyoiqC2bi_nHtPZ6uOyj5ERcxiQk,4515
-crossref_local/_core/models.py,sha256=4b8153r7xDBcmTQFsIeSLD1jJm4KvuUkKTLixB7lccM,5562
+crossref_local/_core/export.py,sha256=uM_Qb9WXmn7v4tDOQjBlhJNQY0Y0WbefHxNLrRhe5tI,8899
+crossref_local/_core/fts.py,sha256=Jsj3Cm9pihICk7Rvr9vIZbUsoxKzL0OligFFQxtQDVA,5030
+crossref_local/_core/models.py,sha256=tbo0hIViKM9eJQXzp2APFhRvYd1UusUdJHugpPyNAEo,8697
 crossref_local/_impact_factor/__init__.py,sha256=pcgVCPogBisANYE5Vp2PHVGPgxoMsSXr-6utqVE97-4,559
 crossref_local/_impact_factor/calculator.py,sha256=eZ13URAZzPdRyAQpS8zXe_T33e2lm_gQhtoJCXbfIGM,15977
 crossref_local/_impact_factor/journal_lookup.py,sha256=Ztx6ZeWxfmPvA3KfcW5h_yz01XPstIdk91j3nu2Q-qw,8846
 crossref_local/_remote/__init__.py,sha256=bTA_bed9XsA0HDArm1o0WagsasxN0atzXdRI6I-r7Pk,1412
-crossref_local/_remote/base.py,sha256=gvvvq7uGrG1IOQEZ5ZhWWuakg9Fl_GdkAaU1AfCmtzY,11258
+crossref_local/_remote/base.py,sha256=VlQCPdhRcUynB1IAnJAyfXhxzfsLa0tOSSINAH8lMuI,12160
 crossref_local/_remote/collections.py,sha256=840HkakS_AtZbmRMYP-GpUX4S5oJ9jaJXzjivQgyozU,5022
 crossref_local/_server/__init__.py,sha256=4fTk1r__5uWhC6XABVDV7PTPdglFIV32hr9KZ5WbKc8,3869
 crossref_local/_server/middleware.py,sha256=6tdrfFCOA06i0HIDtrHZthY8DXbKy8rMEWHf-3Ve8-Y,926
-crossref_local/_server/models.py,sha256=gjnE-MK4XtfdN_obMyqT5xjNXPa3F-BPrgMrrg908KY,2580
+crossref_local/_server/models.py,sha256=8ylDo3UhNoRGjXg5bmiUvB-BWy_8nahD-PSQSEK_8wk,2965
 crossref_local/_server/routes_citations.py,sha256=DUpxl4FdZS7NheM93vCD6ntP5I1Hgw_D8mkGBEMBV2Y,2742
 crossref_local/_server/routes_collections.py,sha256=NaWNTMcswmh87HTZYn5QpzxwtA_c18N3cqy6CrqCBNo,8135
 crossref_local/_server/routes_compat.py,sha256=kSF4_DKg7lcp67Xii95A1pKDBc3kGcqGD-IPaK6TJA8,2726
-crossref_local/_server/routes_works.py,sha256=RxQnkyzoQZtm2n5XTBpZjxjB6ti8GFs-zIzap6VHSgU,3521
+crossref_local/_server/routes_works.py,sha256=YtDKh-xxTCvbDf8K7kwxWszIYjYz-PZzqJH8q13CoFE,5080
 crossref_local/_server/server.py,sha256=D3SSXQ_cBwMqGd9tLwKK84JBDoisvQwVUuu11yr5zhE,579
-crossref_local-0.5.0.dist-info/METADATA,sha256=T2gU3EHRunTfZKRUck9aOrCZAE9HW6VoxjiDdCzyKQs,10748
-crossref_local-0.5.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-crossref_local-0.5.0.dist-info/entry_points.txt,sha256=BZbDvHLHzlKzFc-dqLAFwPrWGmGq5yFuD3vslzbmRnk,111
-crossref_local-0.5.0.dist-info/RECORD,,
+crossref_local-0.5.1.dist-info/METADATA,sha256=RXoY2iFOTFVQcDQq0vPIwBpqz1rYPIiA4Uz-kj0-YZM,10748
+crossref_local-0.5.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+crossref_local-0.5.1.dist-info/entry_points.txt,sha256=BZbDvHLHzlKzFc-dqLAFwPrWGmGq5yFuD3vslzbmRnk,111
+crossref_local-0.5.1.dist-info/RECORD,,