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.

75 lines
2.0 KiB

AUTOFILL_TPL = '\\autofill{FIELD}{INFO}'
def autofill(text, paper):
for field, repl in get_autofill_info(paper):
text = text.replace(find_autofill_pattern(text, field),
autofill_repl(repl, field))
return text
def get_autofill_info(paper):
fields = paper.bibentry.fields
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)))
return info
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 autofill_template(field):
return AUTOFILL_TPL.replace('FIELD', field)
def autofill_repl(repl, field):
return autofill_template(field).replace('INFO', repl)
def find_autofill_pattern(text, field):
pattern = autofill_template(field).replace('INFO}', '')
start = text.find(pattern)
after = start + len(pattern)
info_length = text[after:].find('}')
end = start + len(pattern) + info_length + 1
return text[start:end]
##### ugly replace by proper #####
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
#####