From 28e83ac8a32386da494f0630fafd85640a2a76b0 Mon Sep 17 00:00:00 2001 From: schneefux Date: Wed, 8 Mar 2017 17:47:49 +0100 Subject: allow to import apigrabber as a module --- api.py | 31 ++++++++++++++++--------------- 1 file changed, 16 insertions(+), 15 deletions(-) diff --git a/api.py b/api.py index 2e1c3de..c7b8091 100644 --- a/api.py +++ b/api.py @@ -6,7 +6,7 @@ import logging import json import asyncpg -import crawler +from . import crawler import joblib.worker @@ -82,18 +82,19 @@ async def startup(): await worker.start() -logging.basicConfig( - filename=os.path.realpath( - os.path.join(os.getcwd(), - os.path.dirname(__file__))) + - "/logs/apigrabber.log", - filemode="a", - level=logging.DEBUG -) -console = logging.StreamHandler() -console.setLevel(logging.WARNING) -logging.getLogger("").addHandler(console) +if __name__ == "__main__": + logging.basicConfig( + filename=os.path.realpath( + os.path.join(os.getcwd(), + os.path.dirname(__file__))) + + "/logs/apigrabber.log", + filemode="a", + level=logging.DEBUG + ) + console = logging.StreamHandler() + console.setLevel(logging.WARNING) + logging.getLogger("").addHandler(console) -loop = asyncio.get_event_loop() -loop.run_until_complete(startup()) -loop.run_forever() + loop = asyncio.get_event_loop() + loop.run_until_complete(startup()) + loop.run_forever() -- cgit v1.3.1