improve database_service

src/database_service.py

@@ -2,10 +2,14 @@
 import logging
 import settings
 from nameko.rpc import rpc
+from nameko.runners import ServiceRunner
 
-logging.basicConfig()
+logging.basicConfig(
+    filename='logs/database_rpc_service.txt',
+    level=logging.INFO
+)
 LOG = logging.getLogger("DatabaseEnQueueService")
-LOG.setLevel(logging.DEBUG)
+
 
 
 class DatabaseService(object):
@@ -16,10 +20,10 @@ def create_database(self, engine, name, cpu, mem):
         call_args = bytes(json.dumps(
             {"engine": engine, "name": name, "cpu": cpu, "mem": mem}
         ), 'utf-8')
-        LOG.debug("Args Converted")
+        LOG.info("Args Converted")
 
         settings.CREATE_DATABASE_QUEUE.put(call_args)
-        LOG.debug("Queue updated")
+        LOG.info("Queue updated")
 
         return "Create request sent to queue"
 
@@ -28,9 +32,27 @@ def destroy_database(self, engine, name, cpu, mem):
         call_args = bytes(json.dumps(
             {"engine": engine, "name": name, "cpu": cpu, "mem": mem}
         ), 'utf-8')
-        LOG.debug("Args Converted")
+        LOG.info("Args Converted")
 
         settings.DESTROY_DATABASE_QUEUE.put(call_args)
-        LOG.debug("Queue updated")
+        LOG.info("Queue updated")
 
         return "Destroy request sent to queue"
+
+def main():
+    config = {
+        '_log': LOG,
+        'AMQP_URI': 'amqp://guest:guest@localhost'
+    }
+    runner = ServiceRunner(config)
+    runner.add_service(DatabaseService)
+
+    runner.start()
+
+    try:
+        runner.wait()
+    except (KeyboardInterrupt, SystemExit):
+        runner.kill()
+
+if __name__ == '__main__':
+    main()