From 70e805f58ffa4a68b589291635cfabd7d160e88b Mon Sep 17 00:00:00 2001
From: Phyks <webmaster@phyks.me>
Date: Fri, 25 Apr 2014 01:13:19 +0200
Subject: [PATCH] Manual check of generated bibtex

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

TODO :
* Add new fields
---
 main.py | 46 ++++++++++++++++++++++++++++++++++++----------
 1 file changed, 36 insertions(+), 10 deletions(-)

diff --git a/main.py b/main.py
index 32d0e4e..abdaa57 100755
--- a/main.py
+++ b/main.py
@@ -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: