Initial commit
This commit is contained in:
Sebastian Rust
84
make/cookiecutter-update.py
Executable file
84
make/cookiecutter-update.py
Executable file
@@ -0,0 +1,84 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
#
|
||||
# Copyright (c) 2023, Sebastian Rust <harlequix@gmail.com>
|
||||
# All rights reserved.
|
||||
#
|
||||
# This file is part of 'quic-message'.
|
||||
|
||||
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import shutil
|
||||
import subprocess
|
||||
import json
|
||||
from cookiecutter.main import cookiecutter
|
||||
|
||||
|
||||
class TemporaryWorkdir():
|
||||
"""Context Manager for a temporary working directory of a branch in a git repo"""
|
||||
|
||||
def __init__(self, path, repo, branch='master'):
|
||||
self.repo = repo
|
||||
self.path = path
|
||||
self.branch = branch
|
||||
|
||||
def __enter__(self):
|
||||
if not os.path.exists(os.path.join(self.repo, ".git")):
|
||||
raise Exception("Not a git repository: %s" % self.repo)
|
||||
|
||||
if os.path.exists(self.path):
|
||||
raise Exception("Temporary directory already exists: %s" % self.path)
|
||||
|
||||
os.makedirs(self.path)
|
||||
subprocess.run(["git", "worktree", "add", "--no-checkout", self.path, self.branch],
|
||||
cwd=self.repo)
|
||||
|
||||
def __exit__(self, type, value, traceback):
|
||||
shutil.rmtree(self.path)
|
||||
subprocess.run(["git", "worktree", "prune"], cwd=self.repo)
|
||||
|
||||
|
||||
def update_template(template_url, root, branch):
|
||||
"""Update template branch from a template url"""
|
||||
tmpdir = os.path.join(root, ".git", "cookiecutter")
|
||||
project_slug = os.path.basename(root)
|
||||
config_file = os.path.join(root, ".cookiecutter.json")
|
||||
tmp_workdir = os.path.join(tmpdir, project_slug)
|
||||
|
||||
# read context from file.
|
||||
context = None
|
||||
if os.path.exists(config_file):
|
||||
with open(config_file, 'r') as fd:
|
||||
context = json.loads(fd.read())
|
||||
context['project_slug'] = project_slug
|
||||
|
||||
# create a template branch if necessary
|
||||
if subprocess.run(["git", "rev-parse", "-q", "--verify", branch], cwd=root).returncode != 0:
|
||||
firstref = subprocess.run(["git", "rev-list", "--max-parents=0", "HEAD"],
|
||||
cwd=root,
|
||||
stdout=subprocess.PIPE,
|
||||
universal_newlines=True).stdout.strip()
|
||||
subprocess.run(["git", "branch", branch, firstref])
|
||||
|
||||
with TemporaryWorkdir(tmp_workdir, repo=root, branch=branch):
|
||||
# update the template
|
||||
cookiecutter(template_url,
|
||||
no_input=(context != None),
|
||||
extra_context=context,
|
||||
overwrite_if_exists=True,
|
||||
output_dir=tmpdir)
|
||||
|
||||
# commit to template branch
|
||||
subprocess.run(["git", "add", "-A", "."], cwd=tmp_workdir)
|
||||
subprocess.run(["git", "commit", "-m", "Update template"],
|
||||
cwd=tmp_workdir)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
with open(sys.argv[1], 'r') as fd:
|
||||
context = json.load(fd)
|
||||
|
||||
update_template(context['_template'], os.getcwd(), branch=sys.argv[2])
|
||||
|
||||
163
make/mkpreamble.py
Executable file
163
make/mkpreamble.py
Executable file
@@ -0,0 +1,163 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
#
|
||||
# Copyright (c) 2023, Sebastian Rust <harlequix@gmail.com>
|
||||
# All rights reserved.
|
||||
|
||||
import sys
|
||||
import argparse
|
||||
import json
|
||||
import re
|
||||
|
||||
import bibtexparser
|
||||
from bibtexparser.bparser import BibTexParser
|
||||
from bibtexparser.bwriter import BibTexWriter
|
||||
from bibtexparser.customization import convert_to_unicode
|
||||
|
||||
|
||||
def load_bibtex(f):
|
||||
parser = BibTexParser()
|
||||
parser.alt_dict = {}
|
||||
|
||||
with open(f, 'r') as fd:
|
||||
txt = re.sub('^\s*%.*$', '', fd.read(), flags=re.MULTILINE)
|
||||
bib = bibtexparser.loads(txt, parser=parser)
|
||||
|
||||
return bib.entries
|
||||
|
||||
|
||||
def getstr(d, key, default=None):
|
||||
val = d.get(key, None)
|
||||
if val != None and len(val.strip()) > 0: return val.strip()
|
||||
else: return default
|
||||
|
||||
|
||||
def getlist(d, key, sep=',\s*'):
|
||||
val = getstr(d, key)
|
||||
if val: return [a.strip() for a in re.split(sep, val)]
|
||||
else: return []
|
||||
|
||||
|
||||
def reindent(s, delta):
|
||||
if delta >= 0:
|
||||
return re.sub('^', delta * ' ', s, flags=re.MULTILINE)
|
||||
else:
|
||||
return re.sub('^' + (-delta) * ' ', '', s, flags=re.MULTILINE)
|
||||
|
||||
|
||||
def format_entry(pattern, *values):
|
||||
if all(values):
|
||||
return pattern % values
|
||||
|
||||
|
||||
def format_command(name, content):
|
||||
if content:
|
||||
if '\n' in content: return '\\%s{%%\n%s%%\n}' % (name, reindent(content, 2))
|
||||
else: return '\\%s{%s}' % (name, content)
|
||||
|
||||
|
||||
def format_environment(name, content):
|
||||
if content:
|
||||
return '\\begin{%s}\n%s\n\\end{%s}' % (name, reindent(content, 2), name)
|
||||
|
||||
|
||||
def format_title(metadata):
|
||||
title = getstr(metadata, 'title', '')
|
||||
thanks = getstr(metadata, 'thanks')
|
||||
mrclass = getlist(metadata, 'mrclass')
|
||||
# TODO: keywords?
|
||||
|
||||
L = [title]
|
||||
|
||||
if thanks:
|
||||
L.append(r'\thanks{%s}' % thanks)
|
||||
|
||||
if mrclass:
|
||||
L.append(r'\unfootnote{\textbf{AMS CLassification:} %s.}' % (', '.join(mrclass)))
|
||||
|
||||
return '%\n'.join(L)
|
||||
|
||||
|
||||
def format_authors(authors):
|
||||
L = []
|
||||
for a in authors:
|
||||
name = getstr(a, 'name')
|
||||
thanks = getstr(a, 'thanks')
|
||||
if thanks: L.append(name + format_command('thanks', thanks))
|
||||
else: L.append(name)
|
||||
return r' \and '.join(L)
|
||||
|
||||
|
||||
def format_date(metadata):
|
||||
date = getstr(metadata, 'date')
|
||||
if date:
|
||||
y, m, d = date.split('-')
|
||||
return r'\formatdate{%s}{%s}{%s}' % (d.strip(), m.strip(), y.strip())
|
||||
|
||||
|
||||
def format_abstract(metadata):
|
||||
abstract = getstr(metadata, 'abstract')
|
||||
return format_environment("abstract", abstract)
|
||||
|
||||
|
||||
def format_authorlist(authors):
|
||||
L = []
|
||||
for a in authors:
|
||||
L.append(r' \vspace{1em}')
|
||||
L.append(r' \parbox[t]{0.5\textwidth}{')
|
||||
L.append(format_entry(r' \textbf{%s}\\', a.get('name',None)))
|
||||
L.append(format_entry(r' \texttt{%s}\\', a.get('email',None)))
|
||||
L.append(format_entry(r' \texttt{\url{%s}}\\', a.get('url',None)))
|
||||
L.append(r' \\')
|
||||
L.append(format_entry(r' %s\\', a.get('university',None)))
|
||||
L.append(format_entry(r' %s\\', a.get('department',None)))
|
||||
L.append(format_entry(r' %s\\', a.get('address',None)))
|
||||
L.append(format_entry(r' %s\\', a.get('city',None)))
|
||||
L.append(' }')
|
||||
|
||||
return '%\n'.join([it for it in L if it != None])
|
||||
|
||||
|
||||
def format_preamble(metadata, authors):
|
||||
L = []
|
||||
L.append('% Auto-generated preamble')
|
||||
|
||||
L.append(format_command('title', format_title(metadata)))
|
||||
L.append(format_command('author', format_authors(authors)))
|
||||
L.append(format_command('date', format_date(metadata)))
|
||||
|
||||
# add institute entry for beamer slides
|
||||
if metadata['entrysubtype'] == 'slides':
|
||||
for a in authors:
|
||||
if a['name'] in metadata['author']:
|
||||
L.append(format_command('institute', a['university']))
|
||||
|
||||
L.append(format_command(r'newcommand{\makeabstract}', format_abstract(metadata)))
|
||||
L.append(format_command(r'newcommand{\makeauthorlist}', format_authorlist(authors)))
|
||||
|
||||
return '\n\n'.join([it for it in L if it != None])
|
||||
|
||||
|
||||
# ----------------------------------------------------------------------- #
|
||||
# Argument parsing #
|
||||
# ----------------------------------------------------------------------- #
|
||||
|
||||
parser = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter,
|
||||
description='Generate preamble')
|
||||
|
||||
parser.add_argument("metadata", help="Metadata file",)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
entries = load_bibtex(args.metadata)
|
||||
metadata = None
|
||||
authors = []
|
||||
|
||||
for e in entries:
|
||||
if e['ENTRYTYPE'] == 'article':
|
||||
metadata = e
|
||||
|
||||
elif e['ENTRYTYPE'] == 'misc':
|
||||
authors.append(e)
|
||||
|
||||
print(format_preamble(metadata, authors))
|
||||
Reference in New Issue
Block a user