web_visits.py

import os
import json

from tornado.web import RequestHandler

from pygeoip import GeoIP
import dateutil.parser

from app.models import Visit
from app import settings


class VisitController(RequestHandler):

    def post(self):
        self.set_status(204)
        data = json.loads(self.request.body)

        ip = self.request.remote_ip
        if ip:
            data['ip'] = ip
        else:
            return self.finish()

        geo_db = GeoIP(os.path.join(settings.DATA_ROOT, 'GeoLiteCity.dat'))
        try:
            city = geo_db.record_by_addr(ip)
        except:
            return self.finish()
        else:
            data.update({'city': city['city'],
                         'country_code': city['country_code'],
                         'country_name': city['country_name'],
                         'continent': city['continent']})

        data['date'] = dateutil.parser.parse(data['date'])

        try:
            Visit.create(**data)
        except:
            pass

        return self.finish()