create logs folder

create logs folder

core/FileExporter.py

@@ -36,28 +36,24 @@ class FileExporter:
     def __init__(self):
         pass
 
-
     def ExportListToCSV(self, ipGeoLocObjs, filename):
         return self.__ExportToCSV(ipGeoLocObjs, filename)
 
     def ExportToCSV(self, ipGeoLocObj, filename):
         return self.__ExportToCSV([ipGeoLocObj], filename)
 
-
     def ExportListToXML(self, ipGeoLocObjs, filename):
         return self.__ExportToXML(ipGeoLocObjs, filename)
 
     def ExportToXML(self, ipGeoLocObj, filename):
         return self.__ExportToXML([ipGeoLocObj], filename)
 
-
     def ExportListToTXT(self, ipGeoLocObjs, filename):
         return self.__ExportToTXT(ipGeoLocObjs, filename)
 
     def ExportToTXT(self, ipGeoLocObj, filename):
         return self.__ExportToTXT([ipGeoLocObj], filename)
 
-
     def __ExportToTXT(self, ipGeoLocObjs, filename):
         try:
             with open(filename, 'w') as txtfile:

core/Logger.py

@@ -50,6 +50,7 @@ def __init__(self, nolog=False, verbose=False):
         self.NoLog = nolog
         self.Verbose = verbose
 
+
     def WriteLog(self, messagetype, message):
         filename = '{}.log'.format(datetime.strftime(datetime.now(), "%Y%m%d"))
         path = os.path.join('.', 'logs', filename)

core/Menu.py

@@ -27,7 +27,7 @@
 
 __author__ = 'maldevel'
 __twitter__  = '@maldevel'
-__version__ = '2.0.2'
+__version__ = '2.0.3'
 __year__    = '2015-2016'
 
 

ipgeolocation.py

@@ -28,19 +28,25 @@
 __author__  = 'maldevel'
 
 
-import sys
+import sys, os
 from core.IpGeoLocationLib import IpGeoLocationLib
 from core.Logger import Logger
 from core.Menu import parser,args,banner
 
-
 def main():
 
     # no args provided
     if len(sys.argv) == 1:
         parser.print_help()
         sys.exit(1)
 
+    logsDir = os.path.join(os.getcwd(), 'logs')
+    #resultsDir = os.path.join(os.getcwd(), 'results')
+    if not os.path.exists(logsDir):
+        os.mkdir(logsDir)
+    #if not os.path.exists(resultsDir):
+    #    os.mkdir(resultsDir)
+
     logger = Logger(args.nolog, args.verbose)
 
     #single target or multiple targets