🐛 Fix bugs and improvements

ali4heydari · Jun 11, 2019 · 3f5c301 · 3f5c301
1 parent 90db59a
commit 3f5c301
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 14 deletions.
diff --git a/TelegramChatParser/Program.cs b/TelegramChatParser/Program.cs
@@ -1,4 +1,5 @@
 using System;
+using System.IO;
 using CommandLine;
 using E2.Utils;
 
@@ -16,16 +17,24 @@ public static void Main(string[] args)
                         .ParseArguments<TgParser>(args)
                         .WithParsed<TgParser>(o =>
                         {
+                            if (!File.Exists(o.InputHtmlPath))
+                            {
+                                Console.ForegroundColor = ConsoleColor.Red;
+                                Console.WriteLine("Error: inputHtml dose not exists! aborting...");
+                                Console.ForegroundColor = ConsoleColor.White;
+                                return;
+                            }
+                            
                             TgParser tgParser = new TgParser(o.InputHtmlPath, o.CsvFilePath, o.Verbose, o.Append);
 
                             tgParser.CreateCsv();
                         });
             }
             catch (Exception e)
             {
-                Console.BackgroundColor = ConsoleColor.Red;
+                Console.ForegroundColor = ConsoleColor.Red;
                 Console.WriteLine($"Exception thrown: \nMessage -> {e.Message}\nStackTrace -> {e.StackTrace}");
-                Console.BackgroundColor = ConsoleColor.Black;
+                Console.ForegroundColor = ConsoleColor.White;
             }
         }
     }

diff --git a/TelegramChatParser/TgParser.cs b/TelegramChatParser/TgParser.cs
@@ -39,18 +39,11 @@ public TgParser()
 
         public TgParser(string inputHtmlPath, string csvFilePath, bool verbose = false, bool append = false)
         {
-            if (Append == true && !File.Exists(CsvFilePath))
+            if (append == true && !File.Exists(csvFilePath))
             {
-                Console.BackgroundColor = ConsoleColor.Yellow;
+                Console.ForegroundColor = ConsoleColor.Yellow;
                 Console.WriteLine("Warning: Csv file not exists, Creating new file...");
-                Console.BackgroundColor = ConsoleColor.Black;
-            }
-
-            if (!File.Exists(inputHtmlPath))
-            {
-                Console.BackgroundColor = ConsoleColor.Red;
-                Console.WriteLine("Error: inputHtml dose not exists! aborting...");
-                Console.BackgroundColor = ConsoleColor.Black;
+                Console.ForegroundColor = ConsoleColor.White;
             }
 
             Verbose = verbose;
@@ -107,9 +100,9 @@ private List<Message> GetMessages()
             }
             catch (Exception e)
             {
-                Console.BackgroundColor = ConsoleColor.Red;
+                Console.ForegroundColor = ConsoleColor.Red;
                 Console.WriteLine($"Exception thrown: \nMessage -> {e.Message}\nStackTrace -> {e.StackTrace}");
-                Console.BackgroundColor = ConsoleColor.Black;
+                Console.ForegroundColor = ConsoleColor.White;
             }
 
             return messagesStack.ToList();