#!/usr/bin/python import datetime import asyncio import socket import sys import json import aiohttp.web import aiohttp_route_decorator import aiohttp_wsgi import database import crawler import queries route = aiohttp_route_decorator.RouteCollector() db = database.Database() async def recrawl(): """Gets the latest matches and inserts them into the database.""" print("getting recent matches") api = crawler.Crawler() # TODO: insert API version (force update if changed) # TODO: create database indices # get or put when the last crawl was executed try: last_match_update = await db.meta("last_match_update") except KeyError: last_match_update = datetime.datetime(1, 1, 1).isoformat() await db.meta("last_match_update", last_match_update) nowiso = datetime.datetime.now().isoformat() # crawl and upsert matches = await api.matches_since(last_match_update) if len(matches) > 0: print("got a lot new data items: " + str(len(matches))) await db.upsert(matches, True) await db.meta("last_match_update", nowiso) asyncio.ensure_future(recrawl_soon()) async def recrawl_soon(): """Calls `recrawl` after 60 seconds.""" print("crawler sleeping, Zzzzzz…") await asyncio.sleep(60) asyncio.ensure_future(recrawl()) @route("/matches") async def api_matches(request): data = (await db.select(queries.queries["recent-matches"]))[0]["data"] return aiohttp.web.Response(text=str(data)) @route("/status") async def api_status(_): resp = json.dumps({"version": "0.1.0"}) return aiohttp.web.Response(text=resp) loop = asyncio.get_event_loop() loop.run_until_complete(db.connect("postgres://vgstats@localhost/vgstats")) loop.run_until_complete(queries.load_queries("queries/")) loop.create_task(recrawl()) app = aiohttp.web.Application(loop=loop) route.add_to_router(app.router) aiohttp.web.run_app(app)