[ADD] [SQUASH] Initial login.

Author: ultimatecoder

yamq/server.py

@@ -1,9 +1,9 @@
 import argparse
 import asyncio
+import logging
 import os
 from functools import partial
 
-
 import subject
 import stomp
 import observer
@@ -127,7 +127,7 @@ def connection_lost(self, exc):
         type=str,
         nargs="?",
         default="localhost",
-        help=("Host name on which the server will bind to.")
+        help="Host name on which the server will bind to."
     )
     parser.add_argument(
         "-p", "--port",
@@ -138,12 +138,20 @@ def connection_lost(self, exc):
             "Port number on which the server will listen for incoming requests."
         )
     )
+    parser.add_argument(
+        "-l", "--log",
+        type=str,
+        nargs='?',
+        default="DEBUG",
+        choices= ["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG", "NOTSET"],
+        help="Decide the level of log the application should throw."
+    )
     args = parser.parse_args()
     event_loop = asyncio.get_event_loop()
     coro = event_loop.create_server(STOMP_Server, args.host, args.port)
     server = event_loop.run_until_complete(coro)
 
-    print("Server started on {}:{}".format(args.host, args.port))
+    logging.info("Server started on {}:{}".format(args.host, args.port))
 
     try:
         event_loop.run_forever()