ext.papermanager: implemented writing to .paperinfo

2 files changed, 76 insertions, 4 deletions

diff --git a/ranger/ext/papermanager.py b/ranger/ext/papermanager.py
index 6c422659..932505c9 100644
--- a/ranger/ext/papermanager.py
+++ b/ranger/ext/papermanager.py
@@ -52,15 +52,87 @@ class PaperManager(object):
             return result
 
     def set_paper_info(self, filename, update_dict):
-        pass
+        result = None
+        found = False
+        valid = (filename, basename(filename))
+        first_metafile = None
+
+        if not self.deep_search:
+            metafile = next(self._get_metafile_names(filename))
+            return self._set_pager_info_raw(filename, update_dict, metafile)
+
+        for i, metafile in enumerate(self._get_metafile_names(filename)):
+            if i == 0:
+                first_metafile = metafile
+
+            csvfile = None
+            try:
+                csvfile = open(metafile, "r")
+            except:
+                # .paperinfo file doesn't exist... look for another one.
+                pass
+            else:
+                reader = csv.reader(csvfile, skipinitialspace=True)
+                for row in reader:
+                    name, year, title, authors, url = row
+                    if name in valid:
+                        return self._set_pager_info_raw(filename, update_dict,
+                                metafile)
+            finally:
+                if csvfile:
+                    csvfile.close()
+
+        # No .paperinfo file found, so let's create a new one in the same path
+        # as the given file.
+        if first_metafile:
+            return self._set_pager_info_raw(filename, update_dict, first_metafile)
+
+    def _set_pager_info_raw(self, filename, update_dict, metafile):
+        valid = (filename, basename(filename))
+
+        try:
+            with open(metafile, "r") as infile:
+                reader = csv.reader(infile, skipinitialspace=True)
+                rows = list(reader)
+        except IOError:
+            rows = []
+
+        with open(metafile, "w") as outfile:
+            writer = csv.writer(outfile)
+            found = False
+
+            # Iterate through all rows and write them back to the file.
+            for row in rows:
+                if not found and row[0] in valid:
+                    # When finding the row that corresponds to the given filename,
+                    # update the items with the information from update_dict.
+                    self._fill_row_with_ostruct(row, update_dict)
+                    found = True
+                writer.writerow(row)
+
+            # If the row was never found, create a new one.
+            if not found:
+                row = [basename(filename), None, None, None, None]
+                self._fill_row_with_ostruct(row, update_dict)
+                writer.writerow(row)
+
+    def _fill_row_with_ostruct(self, row, update_dict):
+        for key, value in update_dict.items():
+            if key == "year":
+                row[1] = value
+            elif key == "title":
+                row[2] = value
+            elif key == "authors":
+                row[3] = value
+            elif key == "url":
+                row[4] = value
 
     def _get_metafile_content(self, metafile):
         if metafile in self.metafile_cache:
             return self.metafile_cache[metafile]
         else:
             if exists(metafile):
-                reader = csv.reader(open(metafile, "r"),
-                        skipinitialspace=True)
+                reader = csv.reader(open(metafile, "r"), skipinitialspace=True)
 
                 entries = list(entry for entry in reader if len(entry) == 5)
                 self.metafile_cache[metafile] = entries
diff --git a/ranger/gui/widgets/browsercolumn.py b/ranger/gui/widgets/browsercolumn.py
index 3757af4b..12523648 100644
--- a/ranger/gui/widgets/browsercolumn.py
+++ b/ranger/gui/widgets/browsercolumn.py
@@ -310,7 +310,7 @@ class BrowserColumn(Pager):
             if use_linemode == "filename":
                 infostring = self._draw_infostring_display(drawn, space)
             elif use_linemode == "papertitle":
-                if paperinfo:
+                if paperinfo and paperinfo.authors:
                     authorstring = paperinfo.authors
                     if ',' in authorstring:
                         authorstring = authorstring[0:authorstring.find(",")]