Added saving presets to JSON files.

src/main.py

@@ -5,9 +5,9 @@
 
 # Import basic libraries
 import argparse as ap
-import configparser as cp
 import sys
 import json
+from datetime import datetime
 
 # Libraries for image processing
 import numpy as np
@@ -22,24 +22,39 @@ class apply_filters:
     def __init__(self):
 
         if len(sys.argv) < 4:
-            # Parse configuration from file
-            print("Reading " + sys.argv[1] + 
-                " for configuration" + sys.argv[2], file=sys.stderr)
-            self.conf_file = open(sys.argv[1])
-            self.preset = sys.argv[2]
+            # Parse configuration from json file
+            self.conf_fn = sys.argv[1]
+            self.preset_name = sys.argv[2]
+            try:
+                self.conf_file = open(self.conf_fn)
+            except(FileNotFoundError):
+                print("Creating new file")
+                self.conf_file = open(self.conf_fn, "x")
+            self.conf = json.load(self.conf_file)
+            
             self.parse_conf()
         else:
-            # Parse and save arguments.
+            # Parse arguments from command line
             self.parse_arguments()
             self.input_file = self.args.input_file
             self.output_file = self.args.output_file
             self.dpi = self.args.dpi
             self.filters = self.args.filters
             self.flip = self.args.flip
-            self.conf_file = self.args.config_file if self.args.config_file else None
-            self.preset = self.args.preset if self.args.preset else None
             self.width = self.args.width if self.args.width else 0
 
+            # Save preset to config file
+            if self.args.config_file:
+
+                self.conf_file = self.args.config_file
+
+                if self.args.preset:
+                    self.preset_name = self.args.preset
+                else:
+                    # If no preset name given, create one from time
+                    self.preset_name = "preset_" + datetime.now().strftime("%d_%m_%Y_%H_%M_%S")
+            self.write_conf()
+
         # Convert dimensions
         self.img = Image.open(self.input_file)
         self.convert_dpi()
@@ -53,9 +68,8 @@ class apply_filters:
     def parse_conf(self):
 
         # Parse configuration file if given.
-        
-        self.conf = json.load(self.conf_file)
-        preset = self.conf[self.preset]
+        preset = self.conf[self.preset_name]
+
         self.input_file = preset['inputFile']
         self.output_file = preset['outputFile']
         self.dpi = preset['dpi']
@@ -69,6 +83,32 @@ class apply_filters:
         except(KeyError):
             self.width = 0
 
+
+    def write_conf(self):
+
+        temp = {}
+        temp_fp = open(self.conf_file, 'a+')
+        # If config file new or empty, loading fails
+        try:
+            temp = json.load(temp_fp)
+        except(json.decoder.JSONDecodeError):
+            temp = {}
+        
+        # Create new preset
+        temp[self.preset_name] = ({
+            "inputFile": self.input_file,
+            "outputFile": self.output_file,
+            "dpi": self.dpi,
+            "flip": self.flip,
+            "width": self.width,
+            "filters": self.filters
+            })
+        
+        # Update json file with new preset
+        with open(self.conf_file, 'w') as temp_fp:
+            json.dump(temp, temp_fp, indent=4)
+
+
     def parse_arguments(self):
 
         # Parse arguments