main.py

# main.py

import sys
import os
import pathlib
from src.argumentos import *
from src.ocultar import *
from src.develar import *

def diferente_extension(nombre_original, nombre_resultante):
    arreglo_ruta_1 = os.path.split(nombre_original)
    arreglo_ruta_2 = os.path.split(nombre_resultante)
    if arreglo_ruta_1[1].split(".")[1] == arreglo_ruta_2[1].split(".")[1]:
        return False
    else:
        return True

def uso():
    print("Uso: python3 main.py [-h <txt> <img> <nombre_img_resultante_incluir_misma_extension>] | [-u <img> <nombre_txt_resultante>]")
    sys.exit()

entrada = sys.argv
h = bandera_h(entrada)
u = bandera_u(entrada)

if h == u:
    uso()
elif h:
    nombre_txt = pathlib.Path(__file__).absolute().parent / nombre_archivo_texto(entrada)
    nombre_img_original = pathlib.Path(__file__).absolute().parent / nombre_imagen_para_ocultar(entrada)
    nombre_img_resultante = nombre_imagen_resultante(entrada)
    if nombre_txt == "" or nombre_img_original == "" or nombre_img_resultante == "" or diferente_extension(nombre_img_original, nombre_img_resultante):
        print(diferente_extension(nombre_img_original, nombre_img_resultante))
        uso()
    else:
        ocultar(nombre_img_original, nombre_txt, nombre_img_resultante)
        print("Texto ocultado en la imagen")
elif u:
    nombre_img = pathlib.Path(__file__).absolute().parent / nombre_imagen_para_develar(entrada)
    nombre_txt_resultante = nombre_texto_resultante(entrada)
    if nombre_img == "" or nombre_txt_resultante == "":
        uso()
    else:
        develar(nombre_img, nombre_txt_resultante)
        print("Texto develado de la imagen")