406 lines
16 KiB
Python
Executable File
406 lines
16 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import requests
|
|
import configparser as cfg
|
|
from os.path import isfile
|
|
import sql_alch
|
|
from rich import box
|
|
from rich.console import Console
|
|
from rich.table import Table
|
|
import consultas.querys_sqlite as querys
|
|
import logging
|
|
|
|
selfpath = os.path.abspath(os.path.dirname(__file__))
|
|
parser = cfg.ConfigParser()
|
|
parser.read(f'{selfpath}/config.cfg')
|
|
token = parser.get('iplocate', 'token')
|
|
token = token.strip("'")
|
|
muevelog = f'{selfpath}/muevelog.sh '
|
|
log_file = f'{selfpath}/log/iplocate.log'
|
|
console = Console()
|
|
# tkn=True
|
|
|
|
logging.basicConfig(
|
|
filename=log_file,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO
|
|
)
|
|
logging.info("iplocate inicio")
|
|
|
|
|
|
def valid_ip(ip_str):
|
|
ip = ip_str.split('.')
|
|
if len(ip) != 4:
|
|
return False
|
|
else:
|
|
for segm in ip:
|
|
if not segm.isdigit() or not (255 >= int(segm) >= 0):
|
|
return False
|
|
return True
|
|
|
|
|
|
def local_ip(ip_str):
|
|
ip = ip_str.split('.')
|
|
if ip[0] == '192' and ip[1] == '168':
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def log_usage(proceso, info):
|
|
usage_log = f"Proceso:[{proceso}] - Info: [{info}]"
|
|
logging.info(usage_log)
|
|
|
|
|
|
def log_error(proceso, info):
|
|
error_log = f"Proceso: [{proceso}], Info:[{info}]"
|
|
logging.error(error_log)
|
|
|
|
|
|
try:
|
|
ownip = requests.get('https://ifconfig.me/').text
|
|
except Exception as ex:
|
|
log_error("Request ip pública", ex)
|
|
|
|
|
|
def filtro_ip_propia(ip):
|
|
return True if ip != ownip and not local_ip(ip) else False
|
|
|
|
|
|
def visita_como_lista(ip_info):
|
|
lista_visitas = []
|
|
contad = 0
|
|
for tupla in ip_info:
|
|
visita = []
|
|
if contad < 1:
|
|
for ind, val in enumerate(tupla):
|
|
if ind == 0:
|
|
ip_dict = dict()
|
|
for dato in str(val).split(';'):
|
|
ip_dict[dato.split("=")[0]] = dato.split("=")[1]
|
|
print_tabla_ip(ip_dict)
|
|
else:
|
|
visita.append(str(val).split(',')[3].split('=')[1])
|
|
visita.append(str(val).split(',')[2].split('=')[1])
|
|
metodo = (str(val).split(',')[4].split('=')[1])
|
|
metodo = '---' if metodo == 'None' else metodo
|
|
visita.append(metodo)
|
|
request = ''.join(str(val).split(',')[5].split('=')[1:])
|
|
request = '---' if request == 'None' else request
|
|
visita.append(request)
|
|
else:
|
|
for ind, val in enumerate(tupla):
|
|
if ind > 0:
|
|
visita.append(str(val).split(',')[3].split('=')[1])
|
|
visita.append(str(val).split(',')[2].split('=')[1])
|
|
metodo = (str(val).split(',')[4].split('=')[1])
|
|
metodo = '---' if metodo == 'None' else metodo
|
|
visita.append(metodo)
|
|
request = ''.join(str(val).split(',')[5].split('=')[1:])
|
|
request = '---' if request == 'None' else request
|
|
visita.append(request)
|
|
lista_visitas.append(visita)
|
|
contad += 1
|
|
return lista_visitas
|
|
|
|
|
|
def print_ipinfo(ip, tkn=True):
|
|
if valid_ip(ip):
|
|
try:
|
|
ip_info = sql_alch.consulta_ip(ip, tkn)
|
|
except Exception as ex:
|
|
print(f'Exception sql_alch.consulta_ip({ip})\n', ex)
|
|
ip_info = None
|
|
if isinstance(ip_info, dict):
|
|
print_tabla_ip(ip_info)
|
|
elif isinstance(ip_info, list):
|
|
print_tabla_visita(visita_como_lista(ip_info))
|
|
else:
|
|
console.print(
|
|
'[red]Error type(ip_info) = [/red][magenta]'
|
|
f'{type(ip_info)}[/magenta][red]][/red]'
|
|
)
|
|
else:
|
|
ipr = ip.split('\n')[0]
|
|
console.print(
|
|
f'[red][[/red][magenta]{ipr}[/magenta][red]]'
|
|
' no es una IP válida![/red]'
|
|
)
|
|
|
|
|
|
def print_tabla_ip(ip_info):
|
|
tbl_ip = Table(box=box.ROUNDED, highlight=True, border_style="dim plum1")
|
|
tbl_ip.add_column("IP", justify="left", style="bold #005fff", no_wrap=True)
|
|
tbl_ip.add_column(f"{ip_info['ip']}", justify="left", style="#00ff5f")
|
|
try:
|
|
if 'host' in ip_info:
|
|
tbl_ip.add_row("HOSTNAME", ip_info['host'])
|
|
elif 'hostname' in ip_info:
|
|
tbl_ip.add_row("HOSTNAME", ip_info['hostname'])
|
|
if 'anycast' in ip_info:
|
|
anycast = 'Si' if ip_info['anycast'] else 'No'
|
|
tbl_ip.add_row("ANYCAST", anycast)
|
|
if 'cuidad' in ip_info:
|
|
tbl_ip.add_row("CUIDAD", ip_info['cuidad'])
|
|
elif 'city' in ip_info:
|
|
tbl_ip.add_row("CUIDAD", ip_info['city'])
|
|
if 'region' in ip_info:
|
|
tbl_ip.add_row("REGION", ip_info['region'])
|
|
if 'pais' in ip_info:
|
|
tbl_ip.add_row("PAIS", ip_info['pais'])
|
|
elif 'country' in ip_info:
|
|
tbl_ip.add_row("PAIS", ip_info['country'])
|
|
if 'geoloc' in ip_info:
|
|
tbl_ip.add_row("GEOLOC", ip_info['geoloc'])
|
|
elif 'loc' in ip_info:
|
|
tbl_ip.add_row("GEOLOC", ip_info['loc'])
|
|
if 'organizacion' in ip_info:
|
|
tbl_ip.add_row("ORGANIZ.", ip_info['organizacion'])
|
|
elif 'org' in ip_info:
|
|
tbl_ip.add_row("ORGANIZ.", ip_info['org'])
|
|
if 'fecha_reg' in ip_info:
|
|
tbl_ip.add_row("FECHA REG", ip_info['fecha_reg'])
|
|
if 'tzone' in ip_info:
|
|
tbl_ip.add_row("TimeZone", ip_info['tzone'])
|
|
elif 'timezone' in ip_info:
|
|
tbl_ip.add_row("TimeZone", ip_info['timezone'])
|
|
if 'cod_post' in ip_info:
|
|
tbl_ip.add_row("COD POST", ip_info['cod_post'])
|
|
elif 'postal' in ip_info:
|
|
tbl_ip.add_row("COD POST", ip_info['postal'])
|
|
except Exception as ex:
|
|
print('Exception ipl.print_tabla_ip(): ', ex)
|
|
try:
|
|
console.print(tbl_ip)
|
|
except Exception as ex:
|
|
print('Exception print(tabla_ip): ', ex)
|
|
|
|
|
|
def print_tabla_visita(lista_visitas):
|
|
tbl_v = Table(box=box.ROUNDED, show_lines=False, row_styles=[
|
|
"dim", ""], border_style="dark_orange3")
|
|
tbl_v.add_column("Fecha visita", justify="center",
|
|
style="bright_yellow", no_wrap=True)
|
|
tbl_v.add_column("Codigo", justify="center", style="bold dodger_blue2")
|
|
tbl_v.add_column("Metodo", justify="center", style="bright_magenta")
|
|
tbl_v.add_column("Request", justify="left", style="#00ff5f",
|
|
overflow='fold', no_wrap=False)
|
|
for item in lista_visitas:
|
|
tbl_v.add_row(item[0], item[1], item[2], item[3])
|
|
|
|
console.print(tbl_v)
|
|
|
|
|
|
def archivo_ips(ips, tkn=True):
|
|
with open(ips, 'r') as lista:
|
|
for linea in lista:
|
|
if '\n' in linea:
|
|
linea = linea.split('\n')[0]
|
|
print_ipinfo(linea, tkn)
|
|
sys.exit(0)
|
|
|
|
|
|
def _sync():
|
|
console.print(
|
|
'[bold yellow]Sincronizando logs del servidor'
|
|
'(bash script)[/bold yellow]'
|
|
)
|
|
subprocess.check_call(muevelog+"%s" % "--start", shell=True)
|
|
|
|
|
|
def archivo_invalido(archivo):
|
|
console.print(
|
|
f'[red]Archivo [[/red][magenta]{archivo}'
|
|
'[/magenta][red]] no es válido[/red]'
|
|
)
|
|
sys.exit(0)
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) > 1:
|
|
try:
|
|
match sys.argv[1]:
|
|
case '--all':
|
|
_sync()
|
|
console.print(
|
|
'[bold yellow]Cargando logs en base de datos'
|
|
'[/bold yellow]\n'
|
|
)
|
|
sql_alch.carga_logs()
|
|
console.print(
|
|
'[bold yellow]Registro de datos de ipinfo'
|
|
'[/bold yellow]'
|
|
)
|
|
sql_alch.registro_ips()
|
|
console.print(
|
|
'[bold yellow]Generando mapa de visitas[/bold yellow]')
|
|
sql_alch.mapsgen()
|
|
case '--sync':
|
|
_sync()
|
|
case '-c':
|
|
console.print(
|
|
'[bold yellow]Cargando logs en base de datos'
|
|
'[/bold yellow]\n'
|
|
)
|
|
sql_alch.carga_logs()
|
|
case '-g':
|
|
console.print(
|
|
'[bold yellow]Registro de datos de ipinfo'
|
|
'[/bold yellow]'
|
|
)
|
|
sql_alch.registro_ips()
|
|
case '-d':
|
|
console.print(
|
|
'[bold yellow]Consulta a base de datos:[/bold yellow]')
|
|
ip = sys.argv[2]
|
|
print_ipinfo(ip, None)
|
|
case '-D':
|
|
console.print(
|
|
'[bold yellow]Consulta por archivo a base de datos:'
|
|
'[/bold yellow]'
|
|
)
|
|
if isfile(sys.argv[2]):
|
|
archivo_ips(sys.argv[2], None)
|
|
else:
|
|
archivo_invalido(sys.argv[2])
|
|
case '-f':
|
|
if isfile(sys.argv[2]):
|
|
archivo_ips(sys.argv[2], False)
|
|
else:
|
|
archivo_invalido(sys.argv[2])
|
|
case '-F':
|
|
if isfile(sys.argv[2]):
|
|
archivo_ips(sys.argv[2])
|
|
else:
|
|
archivo_invalido(sys.argv[2])
|
|
case '-h':
|
|
uso()
|
|
exit(0)
|
|
case '-hq':
|
|
uso_consultas()
|
|
exit(0)
|
|
case '-t':
|
|
ip = sys.argv[2]
|
|
print_ipinfo(ip)
|
|
case '-M':
|
|
console.print(
|
|
'[bold yellow]Generando mapa de visitas[/bold yellow]')
|
|
sql_alch.mapsgen()
|
|
case '-q':
|
|
match sys.argv[2]:
|
|
case '-p':
|
|
pais = sys.argv[3]
|
|
querys.pt_html_cod_pais(pais.upper())
|
|
case '--top':
|
|
top = sys.argv[3]
|
|
querys.pt_top_paises(top)
|
|
case '--pais-desde':
|
|
pais = sys.argv[3]
|
|
fecha_local = sys.argv[4]
|
|
fecha_unix = sql_alch.fecha_error_to_epoch(
|
|
sys.argv[4]+' 00:00:00'
|
|
)
|
|
querys.pt_sel_pais_fecha(
|
|
pais.upper(),
|
|
fecha_unix,
|
|
fecha_local
|
|
)
|
|
case '--detalle-pais':
|
|
pais = sys.argv[3]
|
|
if len(sys.argv) > 4:
|
|
if sys.argv[4].isnumeric():
|
|
codigo = sys.argv[4]
|
|
querys.pt_visita_pais_detalle(
|
|
pais.upper(),
|
|
codigo
|
|
)
|
|
else:
|
|
console.print(
|
|
'[magenta]-q --detalle-pais <pais> '
|
|
'[/magenta][red][bold]<CODIGO> '
|
|
f'inválido ({sys.argv[4]})'
|
|
'[/bold][/red]'
|
|
)
|
|
else:
|
|
querys.pt_visita_pais_detalle(pais.upper())
|
|
case _:
|
|
console.print(
|
|
'[red] query desconocida [bold]'
|
|
f'{sys.argv[2]}[/bold][/red]'
|
|
)
|
|
case _:
|
|
ip = sys.argv[1]
|
|
print_ipinfo(ip, False)
|
|
except IndexError as ex:
|
|
console.print('[red] error sys.args! [/red]', ex)
|
|
finally:
|
|
sys.exit(0)
|
|
|
|
console.print("[green]Ingresa una[/green] [bold blue]IP [/bold blue]"
|
|
"[green]o [/green][bold blue]s [/bold blue]"
|
|
"[green]para salir:[/green]")
|
|
while True:
|
|
console.print("[bold blue] -> [/bold blue]", end='')
|
|
ip = input()
|
|
if ip in 'sq0SQnN':
|
|
exit(0)
|
|
print_ipinfo(ip)
|
|
|
|
|
|
def uso_consultas():
|
|
ayuda = """
|
|
[bold blue]ipLocate[/bold blue]
|
|
|
|
[deep_sky_blue1]Reportes según consultas a base de datos.
|
|
|
|
Uso: [deep_sky_blue1][bold][yellow]iploc -q[/yellow][blue] <consulta>[/bold][/blue]
|
|
|
|
[deep_sky_blue1]Consultas a base de datos:[deep_sky_blue1]
|
|
[bold yellow]-p[/bold yellow] [blue]<pais>[/blue] [green]- Conteo de respuestas html para <pais> (ejs. CL AR)[/green]
|
|
[bold yellow]--top [/bold yellow][blue]<n>[/blue] [green]- Visitas top <n> paises[/green]
|
|
[bold yellow]--pais-desde[/bold yellow][blue] <pais> <fecha>[/blue] [green]- Detalle visitas <pais> desde <fecha> (ej. 2022/9/19)[/green]
|
|
[bold yellow]--detalle-pais[/bold yellow][blue] <pais> opc(<cod>)[/blue] [green] - Muestra al detalle las visitas desde <pais>,
|
|
filtro por codigo opcional.[/green]
|
|
"""
|
|
console.print(ayuda)
|
|
|
|
|
|
def uso():
|
|
ayuda = """
|
|
[bold blue]ipLocate[/bold blue]
|
|
|
|
[deep_sky_blue1]Consulta en ipinfo.io (con o sin token) información sobre IP(s).
|
|
Carga logs de nginx en base de datos.
|
|
Consulta en ipinfo.io y registra en base de datos.
|
|
Consultas y reportes según información en la base de datos.[/deep_sky_blue1]
|
|
|
|
[bold yellow]iploc -h[/bold yellow] [green]- Muestra esta ayuda.[/green]
|
|
[bold yellow]iploc -hq[/bold yellow] [green]- Ayuda sobre queries.[/green]
|
|
|
|
[bold blue]Consultas ipinfo.io:[/bold blue]
|
|
[bold yellow]iploc[/bold yellow] [blue]<IP>[/blue] [green]- Consulta la información de <IP> disponible en ipinfo.io.[/green]
|
|
[bold yellow]iploc -f [/bold yellow][blue]<archivo> [/blue] [green]- Consulta info. de las IPs en <archivo> (ipinfo.io).[/green]
|
|
[bold yellow]iploc -t [/bold yellow][blue]<IP>[/blue] [green]- Consulta la info. de <IP> usando 'token' de ipinfo.io,
|
|
especificado en config.cfg.[/green]
|
|
[bold yellow]iploc -F [/bold yellow][blue]<archivo> [/blue] [green]- Consulta info. de las IPs en <archivo> (token ipinfo.io).[/green]
|
|
|
|
[bold blue]Consultas base de datos:[/bold blue]
|
|
[bold yellow]iploc -d [/bold yellow][blue]<IP> [/blue] [green]- Consulta la información de <IP> disponible en base de datos.[/green]
|
|
[bold yellow]iploc -D [/bold yellow][blue]<archivo> [/blue] [green]- Consulta info. de las IPs en <archivo> (base de datos).[/green]
|
|
[bold yellow]iploc -M [/bold yellow][green]- Genera mapa según registro en BD (cod. 200 y otros).[/green]
|
|
|
|
[bold blue]Operaciones base de datos:[/bold blue]
|
|
[bold yellow]iploc --sync [/bold yellow][green]- Sincroniza logs del servidor (bash script).[/green]
|
|
[bold yellow]iploc -c [/bold yellow][green]- Carga logs en base de datos.[/green]
|
|
[bold yellow]iploc -g [/bold yellow][green]- Guarda ipinfo de IPs sin registro en la BD.[/green]
|
|
[bold yellow]iploc --all [/bold yellow][green]- Realiza las 3 operaciones para poblar BD (--sync, -c y -g) y -M.[/green]
|
|
"""
|
|
console.print(ayuda)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|