summaryrefslogtreecommitdiff
path: root/api.py
diff options
context:
space:
mode:
Diffstat (limited to 'api.py')
-rw-r--r--api.py29
1 files changed, 16 insertions, 13 deletions
diff --git a/api.py b/api.py
index 1da9e8d..42adacb 100644
--- a/api.py
+++ b/api.py
@@ -86,21 +86,24 @@ class Worker(object):
logging.info("nothing to do, idling")
await asyncio.sleep(10)
+ async def start(self, number=1):
+ """Start jobs in background."""
+ for _ in range(number):
+ asyncio.ensure_future(self.run())
async def startup():
- for _ in range(1):
- worker = Worker(
- apitoken=os.environ["VAINSOCIAL_APITOKEN"]
- )
- await worker.connect(
- host=os.environ["POSTGRESQL_HOST"],
- port=os.environ["POSTGRESQL_PORT"],
- user=os.environ["POSTGRESQL_USER"],
- password=os.environ["POSTGRESQL_PASSWORD"],
- database=os.environ["POSTGRESQL_DB"]
- )
- await worker.setup()
- await worker.run()
+ worker = Worker(
+ apitoken=os.environ["VAINSOCIAL_APITOKEN"]
+ )
+ await worker.connect(
+ host=os.environ["POSTGRESQL_HOST"],
+ port=os.environ["POSTGRESQL_PORT"],
+ user=os.environ["POSTGRESQL_USER"],
+ password=os.environ["POSTGRESQL_PASSWORD"],
+ database=os.environ["POSTGRESQL_DB"]
+ )
+ await worker.setup()
+ await worker.start()
logging.basicConfig(level=logging.DEBUG)
loop = asyncio.get_event_loop()