| """ |
| Build BibTeX string from fetched metadata (ground truth). |
| Single central builder: dispatches by source and formats one consistent style. |
| """ |
| from typing import Any, List, Optional |
| import re |
|
|
|
|
| def _escape(s: str) -> str: |
| """Escape BibTeX special chars: \\ { }""" |
| if not s: |
| return "" |
| return s.replace("\\", "\\\\").replace("{", "\\{").replace("}", "\\}") |
|
|
|
|
| def _author_list_to_bibtex(authors: Any) -> str: |
| """Convert authors (list or str) to BibTeX author field (Name1 and Name2).""" |
| if isinstance(authors, str): |
| return _escape(authors.strip()) |
| if isinstance(authors, list): |
| return " and ".join(_escape(str(a).strip()) for a in authors if a) |
| return "" |
|
|
|
|
| def _first_author_last_name(authors: Any) -> str: |
| """Get last name (last word) of first author for key generation.""" |
| if isinstance(authors, str): |
| parts = authors.strip().split() |
| return parts[-1] if parts else "unknown" |
| if isinstance(authors, list) and authors: |
| first = str(authors[0]).strip() |
| parts = first.split() |
| return parts[-1] if parts else "unknown" |
| return "unknown" |
|
|
|
|
| def _bibtex_key(authors: Any, year: str) -> str: |
| """Generate a safe BibTeX key: LastNameYear.""" |
| last = _first_author_last_name(authors) |
| |
| last = re.sub(r"[^a-zA-Z0-9]", "", last) |
| y = (year or "nodate").strip() |
| y = re.sub(r"[^0-9]", "", y)[:4] if y else "nodate" |
| return f"{last}{y}" if last else f"ref{y}" |
|
|
|
|
| def build_fetched_bibtex(source: str, result: Any) -> str: |
| """ |
| Build a BibTeX entry string from fetched metadata. |
| source: 'arxiv' | 'crossref' | 'scholar' | 'semantic_scholar' | 'openalex' | 'dblp' |
| result: the fetcher result object (ArxivMetadata, CrossRefResult, etc.) |
| """ |
| title = "" |
| authors: Any = [] |
| year = "" |
| doi = "" |
| url = "" |
| venue = "" |
| entry_type = "misc" |
|
|
| if source == "arxiv": |
| title = getattr(result, "title", "") or "" |
| authors = getattr(result, "authors", []) or [] |
| year = getattr(result, "year", "") or "" |
| doi = getattr(result, "doi", "") or "" |
| url = getattr(result, "abs_url", "") or "" |
| venue = getattr(result, "journal_ref", "") or "" |
| entry_type = "article" if venue else "misc" |
| elif source == "crossref": |
| title = getattr(result, "title", "") or "" |
| authors = getattr(result, "authors", []) or [] |
| year = getattr(result, "year", "") or "" |
| doi = getattr(result, "doi", "") or "" |
| url = getattr(result, "url", "") or "" |
| venue = getattr(result, "container_title", "") or "" |
| entry_type = "article" |
| elif source == "scholar": |
| title = getattr(result, "title", "") or "" |
| authors = getattr(result, "authors", "") or "" |
| year = getattr(result, "year", "") or "" |
| url = getattr(result, "url", "") or "" |
| entry_type = "misc" |
| elif source == "semantic_scholar": |
| title = getattr(result, "title", "") or "" |
| authors = getattr(result, "authors", []) or [] |
| year = getattr(result, "year", "") or "" |
| url = getattr(result, "url", "") or "" |
| entry_type = "misc" |
| elif source == "openalex": |
| title = getattr(result, "title", "") or "" |
| authors = getattr(result, "authors", []) or [] |
| year = getattr(result, "year", "") or "" |
| doi = getattr(result, "doi", "") or "" |
| url = getattr(result, "url", "") or "" |
| entry_type = "misc" |
| elif source == "dblp": |
| title = getattr(result, "title", "") or "" |
| authors = getattr(result, "authors", []) or [] |
| year = getattr(result, "year", "") or "" |
| doi = getattr(result, "doi", "") or "" |
| url = getattr(result, "url", "") or "" |
| entry_type = "misc" |
| else: |
| return "" |
|
|
| key = _bibtex_key(authors, year) |
| author_str = _author_list_to_bibtex(authors) |
|
|
| lines = [f" author = {{{author_str}}}", f" title = {{{_escape(title)}}}", f" year = {{{year or '?'}}}"] |
| if venue: |
| lines.append(f" journal = {{{_escape(venue)}}}") |
| if doi: |
| lines.append(f" doi = {{{_escape(doi)}}}") |
| if url: |
| lines.append(f" url = {{{_escape(url)}}}") |
| lines.append(f" note = {{Fetched from {source}}}") |
|
|
| return f"@{entry_type}{{{key},\n" + ",\n".join(lines) + "\n}" |
|
|
