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

Manual check of generated bibtex 

* Display the bibtex before adding it, and provide interface to edit it.

TODO :
* Add new fields
This commit is contained in:
Phyks 2014-04-25 01:13:19 +02:00
parent 0f17eaafa6
commit 70e805f58f
1 changed files with 36 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
main.py
View File

@ -21,6 +21,11 @@ from bibtexparser.bparser import BibTexParser
import params
def rawInput(string):
sys.stdout.flush()
return raw_input(string)
def warning(*objs):
"""
Write to stderr
@ -164,6 +169,30 @@ def getExtension(filename):
return filename[filename.rfind('.'):]
def checkBibtex(filename, bibtex):
print("The bibtex entry found for "+filename+" is :")
print(bibtex)
check = rawInput("Is it correct ? [Y/n] ")
bibtex = StringIO(bibtex)
bibtex = BibTexParser(bibtex).get_entry_dict()
bibtex_name = bibtex.keys()[0]
bibtex = bibtex[bibtex_name]
if check.lower() == 'n':
fields = ['type', 'id'] + [i for i in sorted(bibtex)
if i not in ['id', 'type']]
for field in fields:
new_value = rawInput(field.capitalize()+" ? ["+bibtex[field]+"] ")
if new_value != '':
bibtex[field] = new_value
Exception("TODO : Add new fields")
return bibtex
def addFile(src, filetype):
"""
Add a file to the library
@ -179,19 +208,19 @@ def addFile(src, filetype):
warning("Could not determine the DOI or the ISBN for "+src+"." +
"Switching to manual entry.")
while doi_isbn not in ['doi', 'isbn']:
doi_isbn = raw_input("DOI / ISBN ? ").lower()
doi_isbn = rawInput("DOI / ISBN ? ").lower()
if doi_isbn == 'doi':
doi = raw_input('DOI ? ')
doi = rawInput('DOI ? ')
else:
isbn = raw_input('ISBN ? ')
isbn = rawInput('ISBN ? ')
elif filetype == 'article':
warning("Could not determine the DOI for "+src+", switching to manual " +
"entry.")
doi = raw_input('DOI ? ')
doi = rawInput('DOI ? ')
elif filetype == 'book':
warning("Could not determine the ISBN for "+src+", switching to manual " +
"entry.")
isbn = raw_input('ISBN ? ')
isbn = rawInput('ISBN ? ')
elif doi is not False:
print("DOI for "+src+" is "+doi+".")
elif isbn is not False:
@ -201,10 +230,7 @@ def addFile(src, filetype):
bibtex = doi2Bib(doi).strip().replace(',', ",\n")
else:
bibtex = isbn2Bib(isbn).strip()
bibtex = StringIO(bibtex)
bibtex = BibTexParser(bibtex).get_entry_dict()
bibtex_name = bibtex.keys()[0]
bibtex = bibtex[bibtex_name]
bibtex = checkBibtex(src, bibtex)
authors = re.split(' and ', bibtex['author'])
@ -228,7 +254,7 @@ def addFile(src, filetype):
warning("Error, file "+new_name+" already exists.")
default_rename = new_name.replace(getExtension(new_name),
" (2)"+getExtension(new_name))
rename = raw_input("New name ["+default_rename+"] ? ")
rename = rawInput("New name ["+default_rename+"] ? ")
if rename == '':
new_name = default_rename
else: