Minor refactoring

stamparm · stamparm · commit c5d3198101db · 2016-01-07T21:46:20.000+01:00
diff --git a/sqlmapapi.py b/sqlmapapi.py
@@ -19,10 +19,11 @@
 from lib.utils.api import client
 from lib.utils.api import server
 
-if __name__ == "__main__":
+def main():
     """
     REST-JSON API main function
     """
+
     # Set default logging level to debug
     logger.setLevel(logging.DEBUG)
 
@@ -45,3 +46,6 @@
         client(args.host, args.port)
     else:
         apiparser.print_help()
+
+if __name__ == "__main__":
+    main()
