iplocate/iplocate.py

273 lines
12 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 re
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
2022-05-14 03:18:52 -04:00
selfpath = os.path.abspath(os.path.dirname(__file__))
ownip = requests.get('https://ifconfig.me/').text
parser = cfg.ConfigParser()
parser.read(f'{selfpath}/config.cfg')
token = parser.get('iplocate','token')
token = token.strip("'")
muevelog = f'{selfpath}/muevelog.sh '
console = Console()
#tkn=True
2022-05-14 03:18:52 -04:00
# IP validate https://stackoverflow.com/questions/319279/how-to-validate-ip-address-in-python
ip_regx = "^((25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])\.){3}(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])$"
def filtro_ip_propia(ip):
return True if ip != ownip else False
def print_ipinfo(ip, tkn=True):
2022-05-14 03:18:52 -04:00
if (re.search(ip_regx, 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('es dict')
print_tabla_ip(ip_info)
elif isinstance(ip_info, list):
#print('es lista')
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]) #.ljust(24))
visita.append(str(val).split(',')[2].split('=')[1]) #.center(11))
metodo = (str(val).split(',')[4].split('=')[1])
metodo = '---' if metodo == 'None' else metodo #.center(10)
visita.append(metodo)
request = ''.join(str(val).split(',')[5].split('=')[1:])
# configurar wrap en tabla
#request = request[:86]+'...' if len(request) > 90 else request
request = '---' if request == 'None' else request
visita.append(request)
else:
for ind, val in enumerate(tupla):
if ind > 0:
# aqui modificar para representar las nuevas columnas de tabla visita
visita.append(str(val).split(',')[3].split('=')[1]) #.ljust(24)
visita.append(str(val).split(',')[2].split('=')[1]) #.center(11)
metodo = (str(val).split(',')[4].split('=')[1])
metodo = '---' if metodo == 'None' else metodo #.center(10)
visita.append(metodo)
request = ''.join(str(val).split(',')[5].split('=')[1:])
# configurar wrap en tabla
#request = request[:86]+'...' if len(request) > 90 else request
request = '---' if request == 'None' else request
visita.append(request)
lista_visitas.append(visita)
contad+=1
print_tabla_visita(lista_visitas)
else:
console.print(f'[red]Error type(ip_info) = [/red][magenta]{type(ip_info)}[/magenta][red]][/red]')
2022-05-14 03:18:52 -04:00
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):
# color dodger_blue2
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):
# color dodger_blue2
#tbl_v = Table(title=f"[bold][yellow]Visitas registradas:[/yellow] [green]{lista_visitas[0]}[/bold][/green]",
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)
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:
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)
2022-06-01 01:24:55 -04:00
def _sync():
console.print('[bold yellow]Sincronizando logs del servidor(bash script)[/bold yellow]')
subprocess.check_call(muevelog+"%s" % "--start", shell=True)
2022-05-14 03:18:52 -04:00
def main():
if len(sys.argv) > 1:
try:
match sys.argv[1]:
2022-06-01 01:24:55 -04:00
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':
2022-06-01 01:24:55 -04:00
_sync()
case '-c':
2022-05-20 20:34:11 -04:00
console.print('[bold yellow]Cargando logs en base de datos[/bold yellow]\n')
sql_alch.carga_logs()
case '-g':
2022-06-01 01:24:55 -04:00
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:
console.print(f'[red]Archivo [[/red][magenta]{sys.argv[2]}[/magenta]'
'[red]] no es válido![/red]')
sys.exit(0)
case '-f':
if isfile(sys.argv[2]):
archivo_ips(sys.argv[2], False)
else:
console.print(f'[red]Archivo [[/red][magenta]{sys.argv[2]}[/magenta]'
'[red]] no es válido[/red]')
sys.exit(0)
case '-F':
if isfile(sys.argv[2]):
archivo_ips(sys.argv[2])
else:
console.print(f'[red]Archivo [[/red][magenta]{sys.argv[2]}[/magenta]'
'[red]] no es válido[/red]')
sys.exit(0)
case '-h':
uso()
exit(0)
case '-t':
ip = sys.argv[2]
print_ipinfo(ip)
case '-M':
2022-06-01 01:24:55 -04:00
console.print('[bold yellow]Generando mapa de visitas[/bold yellow]')
sql_alch.mapsgen()
case _:
ip = sys.argv[1]
print_ipinfo(ip, False)
except IndexError:
console.print('[red] error sys.args! [/red]')
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():
ayuda = f"""
[bold blue]ipLocate[/bold blue]
[deep_sky_blue1]Consulta información sobre IP(s) disponibles en ipinfo.io con o sin token.
Carga logs de nginx en base de datos. Consulta con 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 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
if __name__ == "__main__":
main()