Add timing info to console

shenril · Aug 20, 2019 · 4d88eae · 4d88eae
1 parent 16c7e01
commit 4d88eae
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/lib/utils/banner.py b/lib/utils/banner.py
@@ -4,7 +4,6 @@
 
 from lib import __version__ as version
 
-
 class Banner:
     r = Fore.RED
     y = Fore.YELLOW
@@ -23,11 +22,14 @@ def banner(self):
         print(self.g + "~/#" + self.e + " Sitadel - Web Application Security Scanner" + self.g + " #\\~" + self.e)
         print(self.g + "~/#" + self.e + " Shenril (@shenril)" + self.g + " #\\~" + self.e)
         print(self.g + "~/#" + self.e + " https://github.com/shenril/Sitadel" + self.g + " #\\~" + self.e)
-        print("\n\n")
+        print("\n")
 
     def preamble(self, url):
         print('URL: %s' % url)
-        print('Started: %s' % (time.strftime('%d/%m/%Y %H:%M:%S')))
+        print('---------  Scan Started: %s ---------' % (time.strftime('%d/%m/%Y %H:%M:%S')))
+
+    def postscript(self):
+        print('---------  Scan Finished: %s ---------' % (time.strftime('%d/%m/%Y %H:%M:%S')))
 
     def version(self):
         return self.g + "~/#" + self.e + " Sitadel (" + version + ")\n"
diff --git a/sitadel.py b/sitadel.py
@@ -169,6 +169,8 @@ def main(self):
             self.ma.attacks(args.attack, self.url, discovered_urls)
         except KeyboardInterrupt:
             raise
+        finally:
+            self.bn.postscript()
 
 
 if __name__ == "__main__":