import os, random
from logging import getLogger, StreamHandler, DEBUG
logger = getLogger(__name__)
handler = StreamHandler(); handler.setLevel(DEBUG)
logger.setLevel(DEBUG)
logger.addHandler(handler)
logger.propagate = False


def get_random_name():
    famous_painters = [
    "Leonardo", "DaVinci",
    "Michelangelo",
    "Pablo", "Picasso",
    "Vincent", "VanGogh",
    "Rembrandt", "VanRijn",
    "Claude", "Monet",
    "Salvador", "Dali",
    "Jackson", "Pollock",
    "Andy", "Warhol",
    "Henri", "Matisse",
    "Georgia", "Keeffe",
    "Edvard", "Munch",
    "Wassily", "Kandinsky",
    "Gustav", "Klimt",
    "Rene", "Magritte",
    "Frida", "Kahlo",
    "Edgar", "Degas",
    "Johannes", "Vermeer",
    "Paul", "Cezanne",
    "Marc", "Chagall",
    ]

    random_painter = random.choice(famous_painters)

    # 4桁の乱数を生成
    rand_num = random.randint(1000, 9999)

    return random_painter+str(rand_num)


def complete(work_dir):
    work_dir = work_dir
    # logger.debug(f"complete :", work_dir)
    return work_dir


def clean(text_output):
    # logger.debug(f"text_output : {text_output}")

    if text_output!="idle_state":
        logger.info(f"clean up : {text_output}.zip")
        os.remove(f"{text_output}.zip")
        return "idle_state"
    else:
        logger.info(f"reset")
        return "idle_state"



def clean_by_name(text_output):
    # logger.debug(f"text_output : {text_output}")
    if text_output!="idle_state":
        text_output, dir_name = text_output.split("+")
        logger.info(f"clean up : {dir_name}.zip")
        os.remove(f"{dir_name}.zip")
        return "idle_state"
    else:
        logger.info(f"reset")
        return "idle_state"