Add project name variable

config.py

@@ -1,6 +1,8 @@
 #!/usr/bin/venv python
 # -*- coding: utf-8 -*-
 
+PROJECT_NAME = "baoSOC"
+
 VT_ATTRIBUTES_MAPPING = {
     'asn': 'str',
     'as_owner': 'int', 

main.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 
 from argparse import ArgumentParser
-from config import VT_ATTRIBUTES_MAPPING
+from config import VT_ATTRIBUTES_MAPPING, PROJECT_NAME
 from vt import VT
 from dnschecker import DNSChecker as DNS
 import ipaddress
@@ -12,7 +12,7 @@ from os.path import exists
 
 
 def checkArguments():
-    parser = ArgumentParser(description="baoSOC")
+    parser = ArgumentParser(description=PROJECT_NAME)
     parser.add_argument('-c', '--config', help='Config file')
     parser.add_argument('--dns', help='Get domain name information', action="store_true")
     # For dns command
@@ -33,7 +33,7 @@ def checkArguments():
 
 def usage():
     print("------------------------------")
-    print("|           baoSOC           |")
+    print(f"|           {PROJECT_NAME}           |")
     print("------------------------------\n")
     print("A tool for SOC analyst\n")
     print("Usage: main.py [COMMAND]")
@@ -56,7 +56,7 @@ def usage():
     print("\t --sha512 FILE\t\tGet the SHA512 of the file")
 
 def mainMenu():
-    print("\n           baoSOC           ")
+    print(f"\n           {PROJECT_NAME}           ")
     print(" What would you like to do? ")
     print("\n OPTION 1: Sanitise URL For emails ")
     print(" OPTION 2: Decoders (PP, URL, SafeLinks) ")