1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#!/usr/bin/python
import asyncio
import os
import logging
import json
import asyncpg
import crawler
import joblib.worker
class Apigrabber(joblib.worker.Worker):
def __init__(self, apitoken):
self._apitoken = apitoken
self._pool = None
self._insertquery = ""
super().__init__(jobtype="grab")
async def connect(self, **args):
"""Connect to database."""
logging.warning("connecting to database")
await super().connect(**args)
self._pool = await asyncpg.create_pool(min_size=1,
**args)
async def setup(self):
"""Initialize the database."""
logging.info("initializing database")
async with self._pool.acquire() as con:
await con.execute("""
CREATE TABLE IF NOT EXISTS
match (id TEXT PRIMARY KEY, data JSONB)
""")
root = os.path.realpath(
os.path.join(os.getcwd(), os.path.dirname(__file__)))
with open(root + "/insert.sql", "r", encoding="utf-8-sig") as file:
self._insertquery = file.read()
async def _execute_job(self, jobid, payload, priority):
"""Finish a job."""
api = crawler.Crawler(self._apitoken)
# if a player is queried, pass that information to processor
if "filter[playerNames]" in payload["params"]:
playername = payload["params"]["filter[playerNames]"]
else:
playername = ""
async with self._pool.acquire() as con:
try:
async for data in api.matches(region=payload["region"],
params=payload["params"]):
async with con.transaction():
matchids = await con.fetch(
self._insertquery, json.dumps(data))
logging.debug("%s: inserted %s matches from API into database",
jobid, len(matchids))
payloads = [{
"id": mat["id"],
"playername": playername
} for mat in matchids]
await self._queue.request(jobtype="process",
payload=payloads,
priority=priority)
except crawler.ApiError as error:
raise joblib.worker.JobFailed(error.args[0])
async def startup():
for _ in range(1):
worker = Apigrabber(
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()
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()
|
