gbucchino/baoSOC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add project name variable

Browse Source
This commit is contained in:
geoffrey 2024-06-22 15:13:39 +02:00
parent df8aa9868b
commit 78d9b221cc
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

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

8
main.py
View File

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