papers.py

import datetime
from pathlib import Path
import re

from things import Thing
from templater import apply_template


class Paper(Thing):
    page_prefix = "pub_"
    def validate(self):
        self.year = str(self.year)
        if not hasattr(self, 'last_updated') and self.year!="Preprints":
            self.last_updated = datetime.datetime(int(self.year), 1, 1)
        self.generate_link_icons()
    
    def generate_link_icons(self):
        # generate icons for links in paper
        new_urls = []
        icons = {}
        for (name, url) in self.urls:
            if re.search(r'\bvideo\b', name, flags=re.IGNORECASE):
                name = '<i class="fa-solid fa-video"></i> '+name
                if 'video' not in icons: # only use the first video link
                    icons['video'] = f'''
                        <a href="{url}" target="_blank">
                            <i class="fa-solid fa-video"></i>
                        </a>
                        '''
            if re.search(r'\bpdf\b', name, flags=re.IGNORECASE):
                name = '<i class="fa-regular fa-file-pdf"></i> '+name
                if 'pdf' not in icons or 'preprint' in name.lower(): # use first pdf link or preprint version
                    icons['pdf'] = f'''
                        <a href="{url}" target="_blank">
                            <i class="fa-regular fa-file-pdf"></i>
                        </a>
                        '''
            if re.search(r'\b(twitter|tweeprint)\b', name, flags=re.IGNORECASE):
                name = '<i class="fa-brands fa-twitter"></i> '+name
                if 'twitter' not in icons: # use first twitter link
                    icons['twitter'] = f'''
                        <a href="{url}" target="_blank">
                            <i class="fa-brands fa-twitter"></i>
                        </a>
                        '''
            if re.search(r'\b(mastodon)\b', name, flags=re.IGNORECASE):
                name = '<i class="fa-brands fa-mastodon"></i> '+name
                if 'mastodon' not in icons: # use first mastodon link
                    icons['mastodon'] = f'''
                        <a href="{url}" target="_blank">
                            <i class="fa-brands fa-mastodon"></i>
                        </a>
                        '''
            if re.search(r'\b(bluesky)\b', name, flags=re.IGNORECASE):
                name = '<i class="fa-brands fa-bluesky"></i> '+name
                if 'bluesky' not in icons: # use first bluesky link
                    icons['bluesky'] = f'''
                        <a href="{url}" target="_blank">
                            <i class="fa-brands fa-bluesky"></i>
                        </a>
                        '''
            if re.search(r'\bhtml\b', name, flags=re.IGNORECASE):
                name = '<i class="fa-regular fa-file-lines"></i> '+name
                if 'html' not in icons: # use first html link
                    icons['html'] = f'''
                        <a href="{url}" target="_blank">
                            <i class="fa-regular fa-file-lines"></i>
                        </a>
                        '''
            new_urls.append((name, url))
        self.urls = new_urls
        self.icons_dict = icons
        self.icons = ''.join([icon_html for icon_type, icon_html in sorted(icons.items(), reverse=True)])

    @property
    def sort_date(self):
        if hasattr(self, 'last_updated'):
            return self.last_updated
        if hasattr(self, 'year') and self.year!='Preprints':
            return datetime.datetime(int(self.year), 1, 1)
        return datetime.datetime.now()

    @property
    def publication_list_year(self):
        if hasattr(self, 'year') and self.year!='Preprints':
            return int(self.year)
        elif hasattr(self, 'last_updated'):
            return int(self.last_updated.year)
        else:
            raise ValueError("Cannot assign a publication list year to "+self.name)


def get_papers():
    papers = {}
    fnames = Path('papers').rglob('*.yaml')
    for fname in fnames:
        if 'paper_template' in str(fname):
            continue
        paper = Paper(fname)
        papers[paper.key] = paper
    return papers


def write_papers(papers):
    for key, paper in papers.items():
        filename = f'pub_{key}.html'
        if not hasattr(paper, 'socialcard') and hasattr(paper, 'abstract'):
            shortabs = paper.abstract
            if len(shortabs) > 160:
                shortabs = shortabs[:160] + '...'
            paper.socialcard = {'description': shortabs}
        if hasattr(paper.author_objects[0], 'links'):
            linkvaldict = dict((k.lower(), v) for k, v in paper.author_objects[0].links)
            if 'mastodon' in linkvaldict:
                if not hasattr(paper, 'socialcard'):
                    paper.socialcard = {}
                paper.socialcard['fediverse_creator'] = linkvaldict['mastodon']
        apply_template('paper.html', filename, keys_from=paper)