summary.py

import argparse
import re
from arxiv2bib import arxiv2bib
from datetime import date

def retrieve_arxiv_bib(id):
    return arxiv2bib([id])[0]

def clean_title(title):
    title = title.strip()
    title = title.replace("\n", "")
    title = re.sub(" +", " ", title)
    return title

# title can contain spaces, caps
def filename(title):
    today = date.today()
    title = title.lower()
    title = title.replace(" ", "-")
    title = title.replace(",", "")
    title = title.replace(":", "")
    title = title.strip()
    title = re.sub("-+", "-", title)
    title = ''.join(e for e in title if e.isalnum() or e == "-")
    print("Filename", title)
    return f"_summaries/{today}-{title}.markdown"

def eprint_without_version(eprint):
    return re.sub("v\d+", "", eprint)

def template(title, bib_id):
    return \
f"""---
layout: summary
title: "{title}"
giscus_comments: true
bib_id: {bib_id}
---

### Three Important Things

#### 1. Foo

#### 2. Bar

#### 3. Baz

### Most Glaring Deficiency

### Conclusions for Future Work
"""

def create_summary_template(title, bib_id):
    filepath = filename(title)

    with open(filepath, "w") as f:
        f.write(template(title, bib_id))

    print(f"New summary template created at {filepath}")

def update_summary_bib(bibtex, eprint):
    bibtex = bibtex[:-2]
    bibtex += f""",\nEprintNoVer   = {{{eprint_without_version(eprint)}}}\n}}"""
    with open("_bibliography/summaries.bib", "a") as f:
        f.write("\n\n")
        f.write(bibtex)
    
    print("Updated _bibliography/summaries.bib with")
    print(bibtex)

def main():
    parser = argparse.ArgumentParser(description='Create summary template from arxiv ID')
    parser.add_argument('--arxiv_id', type=str,
                        help='arXiv article ID', required=False)
    parser.add_argument('--title', required=False)
    parser.add_argument('--bib-id', required=False)
    args = parser.parse_args()

    if args.title:
        title = args.title
        bib_id = args.bib_id
        create_summary_template(title, bib_id)
    else:
        bib = retrieve_arxiv_bib(args.arxiv_id)
        title = clean_title(bib.title)
        bib_id = bib.id

        create_summary_template(title, bib_id)
        update_summary_bib(bib.bibtex(), bib_id)


if __name__ == "__main__":
    main()