import importlib
from threading import Thread, Lock

import matplotlib.pyplot as plt

from loader import dmt

printnoln = lambda text: print(text, end='', flush=True)
rprint = lambda text: print('\r' + text)

DAYS_NEW_USER = 7
IMAGE_MAGICK = "magick"


def calc_intervals(posts, months=3):
    firstpost = dmt(posts).reduce(lambda acc, e: acc if acc < e['CreationDate'] else e['CreationDate'], lambda acc, e: acc if acc < e else e, lambda: posts[0]['CreationDate'],
                                  "firstpost").getresults()
    lastpost = dmt(posts).reduce(lambda acc, e: acc if acc > e['CreationDate'] else e['CreationDate'], lambda acc, e: acc if acc > e else e, lambda: posts[0]['CreationDate'], "lastpost").getresults()

    # calc quarter beginning
    firstpost = firstpost.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
    if (firstpost.month - 1) % months != 0:
        firstpost = firstpost.replace(month=firstpost.month - ((firstpost.month - 1) % months))
    lastpost = lastpost.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
    if (lastpost.month - 1) % months != 0:
        lastpost = lastpost.replace(month=lastpost.month - ((lastpost.month - 1) % months))
    # add 3 months to last post
    if lastpost.month + months > 12:
        lastpost = lastpost.replace(month=lastpost.month + months - 12, year=lastpost.year + 1)
    else:
        lastpost = lastpost.replace(month=lastpost.month + months)

    cdate = firstpost
    intervals = []
    while cdate < lastpost:
        nextmon = cdate.month + months
        nextquarter = cdate.replace(month=nextmon if nextmon <= 12 else nextmon - 12, year=cdate.year + (0 if nextmon <= 12 else 1))
        print("adding interval: " + cdate.strftime("%d-%m-%Y") + " - " + nextquarter.strftime("%d-%m-%Y"))
        intervals.append((cdate, nextquarter))
        cdate = nextquarter
    return intervals


def imprt(file):
    spec = importlib.util.spec_from_file_location("module.name", file)
    foo = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(foo)
    return foo


class FigSaver():
    def __init__(self):
        self.__lock = Lock()
        self.__threads = []

    def save(self, fig, path, **kwargs):
        thread = Thread(target=self.__dosave, args=(fig, path, kwargs))
        with self.__lock:
            self.__threads.append(thread)
        thread.start()

    def __dosave(self, fig, path, kwargs):
        fig.savefig(path, **kwargs)
        plt.close(fig)

    def join(self):
        with self.__lock:
            for thread in self.__threads:
                thread.join()