phyks/bmc
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Check automatically fetched bibtex

Browse Source
This commit is contained in:
Phyks 2014-04-25 14:13:37 +02:00
parent 20a06d1f41
commit 138dd92439
1 changed files with 23 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
main.py
View File

@ -1,5 +1,5 @@
#!/usr/bin/python2 -u #!/usr/bin/env python2
# coding=utf8 # -*- coding: utf8 -*-
""" """
Main app Main app
""" """
@ -18,6 +18,7 @@ try:
except: except:
from StringIO import StringIO from StringIO import StringIO
from bibtexparser.bparser import BibTexParser from bibtexparser.bparser import BibTexParser
from bibtexparser.customization import homogeneize_latex_encoding
import params import params
@ -33,6 +34,15 @@ def warning(*objs):
print("WARNING: ", *objs, file=sys.stderr) print("WARNING: ", *objs, file=sys.stderr)
def parsed2Bibtex(parsed):
bibtex = '@'+parsed['type']+'{'+parsed['id']+"\n"
for field in [i for i in sorted(parsed) if i not in ['type', 'id']]:
bibtex += "\t"+field+"={"+parsed[field]+"},\n"
bibtex += "}"
return bibtex
def bibtexAppend(data): def bibtexAppend(data):
""" """
Append data to the main bibtex file Append data to the main bibtex file
@ -175,12 +185,13 @@ def checkBibtex(filename, bibtex):
check = rawInput("Is it correct ? [Y/n] ") check = rawInput("Is it correct ? [Y/n] ")
bibtex = StringIO(bibtex) bibtex = StringIO(bibtex)
bibtex = BibTexParser(bibtex).get_entry_dict() bibtex = BibTexParser(bibtex, customization=homogeneize_latex_encoding)
bibtex = bibtex.get_entry_dict()
bibtex_name = bibtex.keys()[0] bibtex_name = bibtex.keys()[0]
bibtex = bibtex[bibtex_name] bibtex = bibtex[bibtex_name]
if check.lower() == 'n': while check.lower() == 'n':
fields = ['type', 'id'] + [i for i in sorted(bibtex) fields = [u'type', u'id'] + [i for i in sorted(bibtex)
if i not in ['id', 'type']] if i not in ['id', 'type']]
for field in fields: for field in fields:
@ -195,8 +206,10 @@ def checkBibtex(filename, bibtex):
new_value = rawInput("Value for field "+new_field+" ? ") new_value = rawInput("Value for field "+new_field+" ? ")
bibtex[new_field] = new_value bibtex[new_field] = new_value
# TODO : Reprint bibtex and check
print("\nThe bibtex entry for "+filename+" is :")
print(parsed2Bibtex(bibtex))
check = rawInput("Is it correct ? [Y/n] ")
return bibtex return bibtex
@ -234,9 +247,11 @@ def addFile(src, filetype):
print("ISBN for "+src+" is "+isbn+".") print("ISBN for "+src+" is "+isbn+".")
if doi is not False: if doi is not False:
bibtex = doi2Bib(doi).strip().replace(',', ",\n") # Add extra \n for bibtexparser
bibtex = doi2Bib(doi).strip().replace(',', ",\n")+"\n"
else: else:
bibtex = isbn2Bib(isbn).strip() # Idem
bibtex = isbn2Bib(isbn).strip()+"\n"
bibtex = checkBibtex(src, bibtex) bibtex = checkBibtex(src, bibtex)
authors = re.split(' and ', bibtex['author']) authors = re.split(' and ', bibtex['author'])