SyDeVoS/iplocate
SyDeVoS/iplocate
4
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

improve readability

Browse Source
This commit is contained in:
devfzn 2024-03-04 15:16:05 -03:00
parent f8133c843b
commit dc83b9082d
1 changed files with 38 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
iplocate.py
View File

@ -127,7 +127,7 @@ def print_ipinfo(ip, tkn=True):
ipr = ip.split('\n')[0] ipr = ip.split('\n')[0]
console.print( console.print(
f'[red][[/red][magenta]{ipr}[/magenta][red]]' f'[red][[/red][magenta]{ipr}[/magenta][red]]'
'no es una IP válida![/red]' ' no es una IP válida![/red]'
) )
@ -205,10 +205,20 @@ def archivo_ips(ips, tkn=True):
def _sync(): def _sync():
console.print( console.print(
'[bold yellow]Sincronizando logs del servidor(bash script)[/bold yellow]') '[bold yellow]Sincronizando logs del servidor'
'(bash script)[/bold yellow]'
)
subprocess.check_call(muevelog+"%s" % "--start", shell=True) 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(): def main():
if len(sys.argv) > 1: if len(sys.argv) > 1:
try: try:
@ -216,10 +226,14 @@ def main():
case '--all': case '--all':
_sync() _sync()
console.print( console.print(
'[bold yellow]Cargando logs en base de datos[/bold yellow]\n') '[bold yellow]Cargando logs en base de datos'
'[/bold yellow]\n'
)
sql_alch.carga_logs() sql_alch.carga_logs()
console.print( console.print(
'[bold yellow]Registro de datos de ipinfo[/bold yellow]') '[bold yellow]Registro de datos de ipinfo'
'[/bold yellow]'
)
sql_alch.registro_ips() sql_alch.registro_ips()
console.print( console.print(
'[bold yellow]Generando mapa de visitas[/bold yellow]') '[bold yellow]Generando mapa de visitas[/bold yellow]')
@ -228,11 +242,15 @@ def main():
_sync() _sync()
case '-c': case '-c':
console.print( console.print(
'[bold yellow]Cargando logs en base de datos[/bold yellow]\n') '[bold yellow]Cargando logs en base de datos'
'[/bold yellow]\n'
)
sql_alch.carga_logs() sql_alch.carga_logs()
case '-g': case '-g':
console.print( console.print(
'[bold yellow]Registro de datos de ipinfo[/bold yellow]') '[bold yellow]Registro de datos de ipinfo'
'[/bold yellow]'
)
sql_alch.registro_ips() sql_alch.registro_ips()
case '-d': case '-d':
console.print( console.print(
@ -241,27 +259,23 @@ def main():
print_ipinfo(ip, None) print_ipinfo(ip, None)
case '-D': case '-D':
console.print( console.print(
'[bold yellow]Consulta por archivo a base de datos:[/bold yellow]') '[bold yellow]Consulta por archivo a base de datos:'
'[/bold yellow]'
)
if isfile(sys.argv[2]): if isfile(sys.argv[2]):
archivo_ips(sys.argv[2], None) archivo_ips(sys.argv[2], None)
else: else:
console.print(f'[red]Archivo [[/red][magenta]{sys.argv[2]}[/magenta]' archivo_invalido(sys.argv[2])
'[red]] no es válido![/red]')
sys.exit(0)
case '-f': case '-f':
if isfile(sys.argv[2]): if isfile(sys.argv[2]):
archivo_ips(sys.argv[2], False) archivo_ips(sys.argv[2], False)
else: else:
console.print(f'[red]Archivo [[/red][magenta]{sys.argv[2]}[/magenta]' archivo_invalido(sys.argv[2])
'[red]] no es válido[/red]')
sys.exit(0)
case '-F': case '-F':
if isfile(sys.argv[2]): if isfile(sys.argv[2]):
archivo_ips(sys.argv[2]) archivo_ips(sys.argv[2])
else: else:
console.print(f'[red]Archivo [[/red][magenta]{sys.argv[2]}[/magenta]' archivo_invalido(sys.argv[2])
'[red]] no es válido[/red]')
sys.exit(0)
case '-h': case '-h':
uso() uso()
exit(0) exit(0)
@ -299,12 +313,18 @@ def main():
pais.upper(), codigo) pais.upper(), codigo)
else: else:
console.print( console.print(
f'[magenta]-q --detalle-pais <pais> [/magenta][red][bold]<CODIGO> inválido ({sys.argv[4]})[/bold][/red]') '[magenta]-q --detalle-pais <pais> '
'[/magenta][red][bold]<CODIGO> '
f'inválido ({sys.argv[4]})'
'[/bold][/red]'
)
else: else:
querys.pt_visita_pais_detalle(pais.upper()) querys.pt_visita_pais_detalle(pais.upper())
case _: case _:
console.print( console.print(
f'[red] query desconocida [bold]{sys.argv[2]}[/bold][/red]') '[red] query desconocida [bold]'
f'{sys.argv[2]}[/bold][/red]'
)
case _: case _:
ip = sys.argv[1] ip = sys.argv[1]
print_ipinfo(ip, False) print_ipinfo(ip, False)