From d1e553734b7c6840366c19bd91a61283c7ee1578 Mon Sep 17 00:00:00 2001
From: Pavan Mandava <mspavan04@gmail.com>
Date: Wed, 29 Jan 2020 23:52:07 +0100
Subject: [PATCH] Args Parser Added to run.py, font-size changed

---
 index.html |  4 ++--
 run.py     | 29 +++++++++++++++++++++++++----
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/index.html b/index.html
index 3d03f94..00f6b61 100644
--- a/index.html
+++ b/index.html
@@ -18,7 +18,7 @@
               color: white;
               padding: 4px 4px;
               text-align: center;
-              font-size: 16px;
+              font-size: 18px;
               font-weight: bold;
               cursor: pointer;
         }
@@ -26,7 +26,7 @@
               background-color: #E87131;
         }
         table tr td {
-              font-size: 17px;
+              font-size: 18px;
               font-weight: bold;
         }
     </style>
diff --git a/run.py b/run.py
index ea1c99f..b5a7d2d 100644
--- a/run.py
+++ b/run.py
@@ -9,6 +9,7 @@ import utils.csv_utils as csv_utils
 import fb2_parser.read_fb2 as read_fb2
 import aligner.bitext_align as aligner
 import time
+import argparse
 
 
 def validate_all_xml_files():
@@ -33,6 +34,8 @@ def save_validated_files_to_db():
                 if result:
                     w_str = const.BLUE
                 print(w_str, 'Result :: ', result, const.END, '\n')
+            else:
+                print(const.BLUE, "Book Data already added to the Database", const.END)
 
     json_data['books'] = books_json
     json_utils.write_json_file(const.JSON_PATH, json_data)
@@ -91,7 +94,25 @@ def create_xml_file(book_content, book_metadata_dict):
     create_xml.create_xml_file(book_content, book_metadata_dict)
 
 
-if env.check_env_variables():
-    read_data_files_and_align_sentences('dost_under_enru')
-    validate_all_xml_files()
-    # save_validated_files_to_db()
+if __name__ == '__main__':
+
+    parser = argparse.ArgumentParser(
+        description='Choose a run type',
+        formatter_class=argparse.RawDescriptionHelpFormatter)
+
+    group = parser.add_mutually_exclusive_group()
+    group.add_argument('-rB', dest='read_book', help='Enter a Book Code from books_data.csv')
+    group.add_argument('-v', dest='validate', action='store_true', help='Validate all the pending XML Files against book.xsd')
+    group.add_argument('-s', dest='save_to_db', action='store_true', help='Save Pending XML Files Content to the Database')
+
+    args = parser.parse_args()
+
+    if env.check_env_variables():
+        if args.read_book and len(args.read_book) > 0:
+            read_data_files_and_align_sentences(args.book_code)
+        elif args.validate:
+            validate_all_xml_files()
+        elif args.save_to_db:
+            save_validated_files_to_db()
+        else:
+            print(const.WARNING, "Invalid Option!", const.END)