from odoo import models, api, fields
import logging
import requests
import json

_logger = logging.getLogger(__name__)


class IpLookup(models.Model):
    _name = 'ip.lookup'

    date_from = fields.Date(string='Date From', required=True)
    date_to = fields.Date(string='Date To', required=True)
    lookup_line = fields.One2many('ip.lookup.line', 'ip_lookup_id', string='Lookup Lines', auto_join=True)

    def generate_ip_lookup(self):
        query = [
            ('create_date', '>=', self.date_from),
            ('create_date', '<=', self.date_to),
            ('ip_address', '!=', False)
        ]
        logss = self.env['user.activity.log'].search(query)
        ips = []
        for log in logss:
            ips.append(log.ip_address)
        logs = list(dict.fromkeys(ips))

        count = 0
        for log in logs:
            self.env['ip.lookup.line'].create([{
                'ip_lookup_id': self.id,
                'ip_address': log
            }])
            count += 1
        _logger.info('IP Lookup Generated %s' % count)

    def _load_ip_address_lookup(self):
        domain = [
            ('lookup', '=', False),
        ]
        logs = self.env['ip.lookup.line'].search(domain, limit=45, order='create_date asc')
        for log in logs:
            try:
                ipinfo = requests.get('http://ip-api.com/json/%s' % log.ip_address).json()
                del ipinfo['status']
                log.lookup = json.dumps(ipinfo, indent=4, sort_keys=True)
            except:
                log.lookup = ''


class IpLookupLine(models.Model):
    _name = 'ip.lookup.line'

    ip_lookup_id = fields.Many2one('ip.lookup', string='Lookup Ref', required=True, ondelete='cascade', index=True,
                                 copy=False)
    ip_address = fields.Char(string='IP Address')
    lookup = fields.Char(string='Lookup')
    country = fields.Char(string='Country')