iplocate/iplocate.py

406 lines
16 KiB
Python
Raw Normal View History

2022-05-14 03:18:52 -04:00
#!/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
2022-05-14 03:18:52 -04:00
selfpath = os.path.abspath(os.path.dirname(__file__))
parser = cfg.ConfigParser()
parser.read(f'{selfpath}/config.cfg')
2023-12-26 02:11:15 -03:00
token = parser.get('iplocate', 'token')
2022-05-14 03:18:52 -04:00
token = token.strip("'")
muevelog = f'{selfpath}/muevelog.sh '
log_file = f'{selfpath}/log/iplocate.log'
console = Console()
2023-12-26 02:11:15 -03:00
# tkn=True
2022-05-14 03:18:52 -04:00
2023-02-15 03:09:08 -03:00
logging.basicConfig(
2024-03-04 13:45:05 -03:00
filename=log_file,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO
2023-12-26 02:11:15 -03:00
)
2023-02-15 03:09:08 -03:00
logging.info("iplocate inicio")
2023-12-26 02:11:15 -03:00
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)
2023-12-26 02:11:15 -03:00
2023-02-15 03:09:08 -03:00
def log_error(proceso, info):
error_log = f"Proceso: [{proceso}], Info:[{info}]"
logging.error(error_log)
2023-12-26 02:11:15 -03:00
2023-02-15 03:09:08 -03:00
try:
ownip = requests.get('https://ifconfig.me/').text
except Exception as ex:
log_error("Request ip pública", ex)
2023-12-26 02:11:15 -03:00
2022-05-14 03:18:52 -04:00
def filtro_ip_propia(ip):
return True if ip != ownip and not local_ip(ip) else False
2022-05-14 03:18:52 -04:00
2024-03-04 13:45:05 -03:00
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:
2023-12-26 02:11:15 -03:00
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):
2024-03-04 13:45:05 -03:00
print_tabla_visita(visita_como_lista(ip_info))
else:
2024-03-04 13:45:05 -03:00
console.print(
'[red]Error type(ip_info) = [/red][magenta]'
f'{type(ip_info)}[/magenta][red]][/red]'
)
2022-05-14 03:18:52 -04:00
else:
ipr = ip.split('\n')[0]
2024-03-04 13:45:05 -03:00
console.print(
f'[red][[/red][magenta]{ipr}[/magenta][red]]'
2024-03-04 15:16:05 -03:00
' no es una IP válida![/red]'
2024-03-04 13:45:05 -03:00
)
def print_tabla_ip(ip_info):
2023-12-26 02:11:15 -03:00
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:
2023-12-26 02:11:15 -03:00
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)
2023-12-26 02:11:15 -03:00
def print_tabla_visita(lista_visitas):
2024-03-04 13:45:05 -03:00
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")
2024-03-04 13:45:05 -03:00
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)
2022-05-14 03:18:52 -04:00
def archivo_ips(ips, tkn=True):
2022-05-14 03:18:52 -04:00
with open(ips, 'r') as lista:
2023-12-26 02:11:15 -03:00
for linea in lista:
if '\n' in linea:
linea = linea.split('\n')[0]
print_ipinfo(linea, tkn)
2022-05-14 03:18:52 -04:00
sys.exit(0)
2023-12-26 02:11:15 -03:00
2022-06-01 01:24:55 -04:00
def _sync():
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[bold yellow]Sincronizando logs del servidor'
'(bash script)[/bold yellow]'
)
2022-06-01 01:24:55 -04:00
subprocess.check_call(muevelog+"%s" % "--start", shell=True)
2024-03-04 15:16:05 -03:00
def archivo_invalido(archivo):
console.print(
f'[red]Archivo [[/red][magenta]{archivo}'
'[/magenta][red]] no es válido[/red]'
)
sys.exit(0)
2022-05-14 03:18:52 -04:00
def main():
if len(sys.argv) > 1:
2023-12-26 02:11:15 -03:00
try:
match sys.argv[1]:
2022-06-01 01:24:55 -04:00
case '--all':
_sync()
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[bold yellow]Cargando logs en base de datos'
'[/bold yellow]\n'
)
2022-06-01 01:24:55 -04:00
sql_alch.carga_logs()
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[bold yellow]Registro de datos de ipinfo'
'[/bold yellow]'
)
2022-06-01 01:24:55 -04:00
sql_alch.registro_ips()
2024-03-04 13:45:05 -03:00
console.print(
'[bold yellow]Generando mapa de visitas[/bold yellow]')
2022-06-01 01:24:55 -04:00
sql_alch.mapsgen()
case '--sync':
2022-06-01 01:24:55 -04:00
_sync()
case '-c':
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[bold yellow]Cargando logs en base de datos'
'[/bold yellow]\n'
)
sql_alch.carga_logs()
case '-g':
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[bold yellow]Registro de datos de ipinfo'
'[/bold yellow]'
)
sql_alch.registro_ips()
case '-d':
2024-03-04 13:45:05 -03:00
console.print(
'[bold yellow]Consulta a base de datos:[/bold yellow]')
ip = sys.argv[2]
print_ipinfo(ip, None)
case '-D':
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[bold yellow]Consulta por archivo a base de datos:'
'[/bold yellow]'
)
if isfile(sys.argv[2]):
archivo_ips(sys.argv[2], None)
else:
2024-03-04 15:16:05 -03:00
archivo_invalido(sys.argv[2])
case '-f':
if isfile(sys.argv[2]):
archivo_ips(sys.argv[2], False)
else:
2024-03-04 15:16:05 -03:00
archivo_invalido(sys.argv[2])
case '-F':
if isfile(sys.argv[2]):
archivo_ips(sys.argv[2])
else:
2024-03-04 15:16:05 -03:00
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':
2024-03-04 13:45:05 -03:00
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]
2024-03-04 13:45:05 -03:00
fecha_unix = sql_alch.fecha_error_to_epoch(
sys.argv[4]+' 00:00:00'
)
2024-03-04 13:45:05 -03: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]
2024-03-04 13:45:05 -03:00
querys.pt_visita_pais_detalle(
pais.upper(),
codigo
)
else:
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[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 _:
2024-03-04 13:45:05 -03:00
console.print(
2024-03-04 15:16:05 -03:00
'[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)
2022-05-14 03:18:52 -04:00
def uso_consultas():
2024-03-04 13:45:05 -03:00
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)
2023-12-26 02:11:15 -03:00
2022-05-14 03:18:52 -04:00
def uso():
2024-03-04 13:45:05 -03:00
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]
2023-02-15 03:09:08 -03:00
[bold yellow]iploc -hq[/bold yellow] [green]- Ayuda sobre queries.[/green]
2023-12-26 02:11:15 -03:00
[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]
2022-06-01 01:24:55 -04:00
[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]
2022-06-08 02:52:59 -04:00
[bold yellow]iploc --all [/bold yellow][green]- Realiza las 3 operaciones para poblar BD (--sync, -c y -g) y -M.[/green]
2022-05-14 03:18:52 -04:00
"""
console.print(ayuda)
2022-05-14 03:18:52 -04:00
2023-12-26 02:11:15 -03:00
2022-05-14 03:18:52 -04:00
if __name__ == "__main__":
main()