from matplotlib.figure import Figure


class Graph:
    def __init__(
        self, x, y, labels, mult_subplots=0, xl="x", yl="y", linetype="k", lims=False
    ):
        # словарь из настроек?
        self.labels = labels
        self.__x = x
        self.__y = y
        self.__fig = Figure(figsize=(5, 4), dpi=100)
        self.__chb = mult_subplots
        self.__ax = self.__fig.add_subplot()

        self.draw_n_func_plot()
        
        self.draw_subplots()

        self.__ax.set_xlabel(xl)
        self.__ax.set_ylabel(yl)
        self.__ax.legend()
        self.figure.tight_layout()

    def add_vline(ax, x=0, ymin=0, ymax=1):
        ax.axvline(x, ymin, ymax)
        # принимаем плот и на нем же рисуем

    def add_hline(
        ax,
        y=0,
        xmin=0,
        xmax=1,
    ):
        ax.axhline(y, xmin, xmax)

    def draw_n_func_plot(
        self,
    ):  # много графиков на одном холсте
        if self.__chb == 0 and len(self.__x) > 1:
            for i in range(len(self.__x)):
                self.__ax.plot(self.__x[i], self.__y[i], label=self.labels[i])
                #новое(старое) условие против проскока
        if self.__chb == 0 and len(self.__x) == 1:
            self.__ax.plot(self.__x[0], self.__y[0], label=self.labels[0]) 

    

    def draw_subplots(
        self,
    ):
        if self.__chb == 1 and len(self.__x) > 1:
            for i in range(len(self.__x)):
                n = int(f"{len(self.__x)}1{i+1}")

                axes = self.__fig.add_subplot(n)
                axes.set_title(f"График №{i+1}")
                axes.plot(self.__x[i], self.__y[i], label=self.labels[i])
                        #новое(старое) условие против проскока
        if self.__chb ==1  and len(self.__x) == 1:
            self.__ax.plot(self.__x[0], self.__y[0], label=self.labels[0])

    @property
    def figure(
        self,
    ):
        return self.__fig