From 890d2b5e3806e58f21dd1a7168979949a558d684 Mon Sep 17 00:00:00 2001
From: Terrtia <or1994@hotmail.fr>
Date: Tue, 26 Mar 2019 15:21:36 +0100
Subject: [PATCH] chg: [server + workers] add config file + add option to
 specify save directory

---
 server/.gitignore                 |  1 +
 server/configs/server.conf.sample |  5 ++
 server/configs/update_conf.py     | 77 +++++++++++++++++++++++++++++++
 3 files changed, 83 insertions(+)
 create mode 100644 server/configs/server.conf.sample
 create mode 100755 server/configs/update_conf.py

diff --git a/server/.gitignore b/server/.gitignore
index 7d17db7..aef579e 100644
--- a/server/.gitignore
+++ b/server/.gitignore
@@ -2,6 +2,7 @@
 *.csr
 *.pem
 *.key
+configs/server.conf
 data/
 logs/
 redis/
diff --git a/server/configs/server.conf.sample b/server/configs/server.conf.sample
new file mode 100644
index 0000000..979ceb1
--- /dev/null
+++ b/server/configs/server.conf.sample
@@ -0,0 +1,5 @@
+[Save_Directories]
+# By default all datas are saved in $D4_HOME/data/
+use_default_save_directory = yes
+save_directory = None
+
diff --git a/server/configs/update_conf.py b/server/configs/update_conf.py
new file mode 100755
index 0000000..9ee2c3a
--- /dev/null
+++ b/server/configs/update_conf.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+
+import os
+import argparse
+import configparser
+
+def print_message(message_to_print, verbose):
+    if verbose:
+        print(message_to_print)
+
+if __name__ == "__main__":
+
+    # parse parameters
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-v', '--verbose',help='Display Info Messages', type=int, default=1, choices=[0, 1])
+    parser.add_argument('-b', '--backup',help='Create Config Backup', type=int, default=1, choices=[0, 1])
+    args = parser.parse_args()
+    if args.verbose == 1:
+        verbose = True
+    else:
+        verbose = False
+    if args.backup == 1:
+        backup = True
+    else:
+        backup = False
+
+    config_file_server = os.path.join(os.environ['D4_HOME'], 'configs/server.conf')
+    config_file_sample = os.path.join(os.environ['D4_HOME'], 'configs/server.conf.sample')
+    config_file_backup = os.path.join(os.environ['D4_HOME'], 'configs/server.conf.backup')
+
+    # Check if confile file exist
+    if not os.path.isfile(config_file_server):
+        # create config file
+        with open(config_file_server, 'w') as configfile:
+            with open(config_file_sample, 'r') as config_file_sample:
+                configfile.write(config_file_sample.read())
+        print_message('Config File Created', verbose)
+    else:
+        config_server = configparser.ConfigParser()
+        config_server.read(config_file_server)
+        config_sections = config_server.sections()
+
+        config_sample = configparser.ConfigParser()
+        config_sample.read(config_file_sample)
+        sample_sections = config_sample.sections()
+
+        mew_content_added = False
+        for section in sample_sections:
+            new_key_added = False
+            if section not in config_sections:
+                # add new section
+                config_server.add_section(section)
+                mew_content_added = True
+            for key in config_sample[section]:
+                if key not in config_server[section]:
+                    # add new section key
+                    config_server.set(section, key, config_sample[section][key])
+                    if not new_key_added:
+                        print_message('[{}]'.format(section), verbose)
+                        new_key_added = True
+                        mew_content_added = True
+                    print_message('    {} = {}'.format(key, config_sample[section][key]), verbose)
+
+        # new keys have been added to config file
+        if mew_content_added:
+            # backup config file
+            if backup:
+                with open(config_file_backup, 'w') as configfile:
+                    with open(config_file_server, 'r') as configfile_origin:
+                        configfile.write(configfile_origin.read())
+                print_message('New Backup Created', verbose)
+            # create new config file
+            with open(config_file_server, 'w') as configfile:
+                config_server.write(configfile)
+            print_message('Config file updated', verbose)
+        else:
+            print_message('Nothing to update', verbose)