Changeset 37:abe2ef05ce80 in stamper for stamper

Timestamp:

Aug 12, 2014, 6:15:59 PM (10 years ago)

Author:

Borja Lopez <borja@…>

Branch:

default

Phase:

public

Message:

Added feature to import stamps from a file:

stamps -i /path/to/stamps.json

The stamps file is read, stamps are loaded and then merged into the
main stamps stream, keeping proper order (by start, end dates)
and removing duplicated entries.

File:

• stamper/stamper.py (modified) (4 diffs)

stamper/stamper.py

@@ -4 +4 @@
 import pygal
 from datetime import datetime, date, timedelta
-from os.path import expanduser, exists, islink
+from os.path import expanduser, exists, islink, isdir
 from os import symlink, remove
 from collections import OrderedDict
+from operator import itemgetter
 
 
@@ -24 +25 @@
         self.stamps = []
 
+    def __json_load(self, filename):
+        """
+        Load the stamps from a file in json format, returning
+        the parsed list.
+        """
+        with open(filename, 'r') as stamps_file:
+            try:
+                stamps = json.load(stamps_file)
+            except ValueError:
+                stamps = []
+        return stamps
+
+    def remove_duplicates(self):
+        """
+        Remove duplicated stamps from the stamps list
+        """
+        stamps = [dict(t) for t in set(
+            [tuple(d.items()) for d in self.stamps])]
+        self.stamps = stamps
+
     def ensure_stamps_file(self):
         if not exists(self.stamps_file):
@@ -30 +51 @@
 
     def load_stamps(self):
-        with open(self.stamps_file, 'r') as stamps_file:
-            try:
-                self.stamps = json.load(stamps_file)
-            except ValueError:
-                self.stamps = []
+        self.stamps = self.__json_load(self.stamps_file)
+
+    def sort_stamps(self):
+        """
+        Sort all the stamps by start and end dates
+        """
+        self.stamps = sorted(self.stamps, key=itemgetter('start', 'end'))
 
     def save_stamps(self):
@@ -346 +369 @@
                 break
         self.save_stamps()
+
+    def import_stamps(self, filename):
+        """
+        Import the stamps from the given file into the main stamps list,
+        merging them into the list (removing duplicated entries)
+        """
+        if not exists(filename):
+            print('[error] ' + filename + 'does not exist')
+            return
+        if isdir(filename):
+            print('[error] ' + filename + 'is a directory')
+            return
+        stamps = self.__json_load(filename)
+        if not stamps:
+            print('[warning] no stamps can be imported from ' + filename)
+            return
+        self.stamps.extend(stamps)
+        self.remove_duplicates()
+        self.sort_stamps()
+        self.save_stamps()
+        print('[warning] ' + str(len(stamps)) + ' stamps merged')
+        print('[warning] remember to review the resulting stamps file')