You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
2.7 KiB
104 lines
2.7 KiB
from .latex_tools import format_for_latex
|
|
|
|
AUTOFILL_TPL = '\\autofill{FIELD}{INFO}'
|
|
|
|
|
|
def get_autofill_pattern(field):
|
|
return AUTOFILL_TPL.replace('FIELD', field)
|
|
|
|
|
|
def autofill(text, paper):
|
|
for field, info in get_autofill_info(paper):
|
|
text = replace_pattern(text,
|
|
get_autofill_pattern(field),
|
|
format_for_latex(info))
|
|
return text
|
|
|
|
|
|
def get_autofill_info(paper):
|
|
fields = paper.bibentry.fields
|
|
tags = paper.tags
|
|
info = []
|
|
if 'year' in fields:
|
|
info.append(('YEAR', fields['year']))
|
|
if 'title' in fields:
|
|
info.append(('TITLE', fields['title']))
|
|
if 'abstract' in fields:
|
|
info.append(('ABSTRACT', fields['abstract']))
|
|
info.append(('AUTHOR', get_author_as_str(paper)))
|
|
info.append(('TAG', ', '.join(tags)))
|
|
return info
|
|
|
|
|
|
def find_first_level_delimiter(text, opening='{', closing='}'):
|
|
if opening in text:
|
|
match = text.split(opening, 1)[1]
|
|
cnt = 1
|
|
for index in xrange(len(match)):
|
|
if match[index] in (opening + closing):
|
|
cnt = (cnt + 1) if match[index] == opening else (cnt - 1)
|
|
if not cnt:
|
|
return match[:index]
|
|
|
|
|
|
def fill_pattern(pattern, info):
|
|
return pattern.replace('INFO', info)
|
|
|
|
|
|
def find_pattern(text, pattern):
|
|
look_at = pattern.replace('INFO}', '')
|
|
found = []
|
|
start = -1
|
|
while True:
|
|
start = text.find(look_at, start + 1)
|
|
if start < 0:
|
|
break
|
|
delim_start = start + len(look_at) - 1
|
|
repl = find_first_level_delimiter(text[delim_start:])
|
|
found.append(pattern.replace('INFO', repl))
|
|
return found
|
|
|
|
|
|
def replace_pattern(text, pattern, info):
|
|
repl = fill_pattern(pattern, info)
|
|
for found in find_pattern(text, pattern):
|
|
text = text.replace(found, repl)
|
|
return text
|
|
|
|
|
|
##### ugly replace by proper #####
|
|
def get_author_as_str(paper):
|
|
persons = paper.bibentry.persons
|
|
authors = []
|
|
if 'author' in persons:
|
|
for author in persons['author']:
|
|
authors.append(format_author(author))
|
|
return concatenate_authors(authors)
|
|
|
|
|
|
def format_author(author):
|
|
first = author.first()
|
|
middle = author.middle()
|
|
last = author.last()
|
|
formatted = ''
|
|
if first:
|
|
formatted += first[0]
|
|
if middle:
|
|
formatted += ' ' + middle[0] + '.'
|
|
if last:
|
|
formatted += ' ' + last[0]
|
|
return formatted
|
|
|
|
|
|
def concatenate_authors(authors):
|
|
concatenated = ''
|
|
for a in range(len(authors)):
|
|
if len(authors) > 1 and a > 0:
|
|
if a == len(authors) - 1:
|
|
concatenated += ' and '
|
|
else:
|
|
concatenated += ', '
|
|
concatenated += authors[a]
|
|
return concatenated
|
|
#####
|