Mendeley文献管理工具上手指南

import jsonfrom collections import defaultdictfrom datetime import datetime  class Reference:    def __init__(self, title, authors, year, journal, doi):        self.title = title        self.authors = authors        self.year = year        self.journal = journal        self.doi = doi        self.tags = []        self.notes = []        self.highlights = []     def add_tag(self, tag):        if tag not in self.tags:            self.tags.append(tag)     def add_note(self, note_text, page_num):        self.notes.append({"text": note_text, "page": page_num, "date": datetime.now().isoformat()})     def add_highlight(self, text, color, page_num):        self.highlights.append({"text": text, "color": color, "page": page_num})     def to_bibtex(self):        authors_str = " and ".join(self.authors)        key = self.authors[0].split()[-1].lower() + str(self.year)        bibtex = f"@article{{{key},\n"        bibtex += f"  title = {{{self.title}}},\n"        bibtex += f"  author = {{{authors_str}}},\n"        bibtex += f"  journal = {{{self.journal}}},\n"        bibtex += f"  year = {{{self.year}}},\n"        bibtex += f"  doi = {{{self.doi}}},\n"        bibtex += "}"        return bibtex  class MendeleyLibrary:    def __init__(self):        self.references = []        self.folders = defaultdict(list)     def add_reference(self, ref, folder="Unsorted"):        if not self._is_duplicate(ref):            self.references.append(ref)            self.folders[folder].append(ref)            return True        return False     def _is_duplicate(self, new_ref):        for ref in self.references:            if ref.doi and ref.doi == new_ref.doi:                return True            if ref.title.lower() == new_ref.title.lower():                return True        return False     def search(self, keyword=None, tag=None, author=None, year=None):        results = []        for ref in self.references:            match = True            if keyword and keyword.lower() not in ref.title.lower():                match = False            if tag and tag not in ref.tags:                match = False            if author and not any(author.lower() in a.lower() for a in ref.authors):                match = False            if year and ref.year != year:                match = False            if match:                results.append(ref)        return results     def export_bibtex(self, folder=None):        refs = self.folders.get(folder, []) if folder else self.references        return "\n\n".join(ref.to_bibtex() for ref in refs)     def get_statistics(self):        return {            "total": len(self.references),            "by_year": dict(defaultdict(int, {r.year: sum(1 for x in self.references if x.year == r.year) for r in self.references})),            "by_folder": {k: len(v) for k, v in self.folders.items()}        }  lib = MendeleyLibrary()ref1 = Reference("Attention Is All You Need", ["Ashish Vaswani", "Noam Shazeer"], 2017, "NeurIPS", "10.5555/3295222")ref1.add_tag("transformer")ref1.add_tag("attention")ref1.add_note("Key insight: self-attention replaces recurrence", 4)lib.add_reference(ref1, "Deep Learning") ref2 = Reference("BERT: Pre-training of Deep Bidirectional Transformers", ["Jacob Devlin", "Ming-Wei Chang"], 2019, "NAACL", "10.18653/v1/N19-1423")ref2.add_tag("bert")ref2.add_tag("pre-training")lib.add_reference(ref2, "Deep Learning") ref3 = Reference("Attention Is All You Need", ["A. Vaswani"], 2017, "NeurIPS", "10.5555/3295222")print(f"Duplicate blocked: {not lib.add_reference(ref3)}") results = lib.search(tag="transformer")print(f"Found {len(results)} papers with tag 'transformer'")print(f"\nBibTeX export:\n{lib.export_bibtex('Deep Learning')}")print(f"\nStatistics: {json.dumps(lib.get_statistics(), indent=2)}")