julyThird/Jt.py

from sys import flags
from color_printing import ERROR, INFO, STATUS, WARNING, cstr, log_print
import datetime
import pickle
import random
import sys
import time
from hashlib import md5
from os import listdir


class Base:
    def __init__(self):
        self.create_time = time.time()
        self.create_date: datetime.datetime = datetime.datetime.utcfromtimestamp(
            self.create_time)
        self.str_time = time.strftime(
            "%Y%m%d%H%M%S", time.gmtime(self.create_time))
        self.__gene_id()

    def __gene_id(self):
        self.__id_seed: str = self.str_time + str(random.randint(10000, 99999))
        self.id: str = md5(self.__id_seed.encode("UTF-8")).hexdigest()


class Entry(Base):
    def __init__(self, title: str = "未命名事项"):
        Base.__init__(self)

        if title == "未命名事项":
            self.title: str = title + self.str_time[4:8]
        else:
            self.title: str = title

        self.__content: str = "事项内容"
        self.set_priority()

        self.set_start_date()
        self.set_deadline(self.create_date + datetime.timedelta(days=7))
        self.time_total : datetime.timedelta = self.deadline-self.start_date
        self.time_left : datetime.timedelta = self.deadline - datetime.datetime.today()
        # self.sub_entries_dict: dict = {}

    def set_priority(self, priority: float = 0):
        if priority < -10:
            print("最低优先级为-10，已自动设置为-10")
            priority = -10.0
        elif priority > 10:
            print("最高优先级为10，已自动设置为10")
            priority = 10.0
        self.priority: float = priority

    def set_deadline(self, deadline_date: datetime.datetime):
        """This function will set the entry's deadline date

        Args:
            deadline_date (float): The deadline date must be datetime structure.

        Returns:
            bool : If successfully set the deadline will return 0, else 1.
        """
        if deadline_date < self.start_date:
            print("\033[1;31mERROR: Let it RIP if it's dead already\033[0m")
            return 1
        else:
            self.deadline = deadline_date
            return 0

    def set_start_date(self, start_date: datetime.datetime = datetime.datetime(1970, 1, 1, 0, 0, 0)):
        if start_date == datetime.datetime(1970, 1, 1, 0, 0, 0):
            self.start_date = self.create_date
        else:
            self.start_date = start_date

    def set_content(self, content: str):
        self.__content = content

    def timestamp_converter(self, time_in):
        """This function can convert timestamp to date string and do reversely.

        Args:
            time_in (float/ str): Timestamp as float or date string like "YYYY/mm/dd HH:MM".

        Returns:
            timestamp or date string.
        """
        try:
            _time_in = float(time_in)
            return datetime.datetime.utcfromtimestamp(_time_in)
        except ValueError:
            return time.mktime(time.strptime(time_in, "%Y/%m/%d %H:%M"))

    def str_to_date(self, date_in: str) -> datetime.datetime:
        """Convert date string like "%Y/%m/%d %H:%M" or "%Y/%m/%d %H:%M" to datetime structure

        Args:
            date_in (str): "%Y/%m/%d %H:%M", 2020/07/03 18:23

        Returns:
            datetime
        """
        try:
            date = datetime.datetime.strptime(date_in, "%Y/%m/%d")
            return date
        except ValueError:
            date = datetime.datetime.strptime(date_in, "%Y/%m/%d %H:%M")
            return date

    def date_input(self) -> datetime.datetime:
        while 1:
            try:
                str_in = input(
                    cstr("Type in the date (%Y/%m/%d or %Y/%m/%d %H:%M): ", "g"))
                date :datetime.datetime = self.str_to_date(str_in)
                return date
            except:
                print("Care the correct date format.")
            

class Coordinate(Base):
    def __init__(self):
        Base.__init__(self)
        self.name = "未命名坐标系" + self.str_time
        # Base.__gene_id(self)
        self.entries_dict = {}
        self.entries_num = 0

    def show_entries(self):
        print(self.entries_dict)

    def add_entry(self, entry_in: Entry):
        self.entries_dict[entry_in.id] = entry_in
        self.entries_num += 1

    def del_entry(self, entry_in: Entry):
        try:
            del self.entries_dict[entry_in.id]
            self.entries_num -= 1
        except KeyError:
            print("\033[1;31mERROR: Entry doesn't exist\033[0m")

    def save_coord(self):
        with open("coord/" + self.name + ".crd", "wb") as f:
            pickle.dump(self, f, pickle.HIGHEST_PROTOCOL)

    def __str__(self) -> str:
        return (
            "Title: "
            + self.name
            + "\nId: "
            + self.id
            + "\nEntries number: "
            + str(self.entries_num)
        )


class Interface:
    def __init__(self):
        self.coordinate_dict: dict = {}
        self.exit: bool = False
        self.func_dict = {
            "exit": self.__exit,
            "open": self.__open_coord,
            "ls": self.__list_coord,
            "show": self.__show_now_coord,
            "entry": self.__new_entry,
            "chcrd": self.__coord_changer,
        }
        self.command_list: list = list(self.func_dict.keys())

    def start(self):
        try:
            print(
                "*******************************\n*Welcome to Jt mission manager*\n*******************************\n"
            )
            try:
                import completer
                completer.start_completer(self.command_list)
                log_print(STATUS, 'Auto complete: ON')
                # print("\033[1;32mAuto complete: ON\033[0m")
            except ModuleNotFoundError:
                log_print(WARNING, 'Auto complete: OFF. Module Not Found')
                # print("\033[1;31mWARNING: Auto complete: OFF. Module Not Found\033[0m")
            self.__coord_changer([])
            while not (self.exit):
                command = input("\033[1;32m> \033[0m").split()
                try:
                    func_name = command[0]
                    try:
                        func = self.func_dict[func_name]
                        func(command[1:])
                    except KeyError:
                        print("\033[1;31mERROR: Command not found\033[0m")
                except IndexError:
                    pass
        except KeyboardInterrupt:
            self.__exit(1)

    def __open_coord(self, coord_file: str):
        with open(coord_file, "rb") as f:
            self.now_coord: Coordinate = pickle.load(f)

    def __list_coord(self, arg):
        """Detect the coord folder if there are any files. 

        Args:
            arg (none): Meaningless, just give 1 or anything

        Returns:
            int: If multiple files exist, return 0, if empty, return -1, if only one, return 1.
        """
        self.files = listdir("coord")
        if len(self.files) == 1:
            print(cstr(f"Only one coordinate was found.\n{self.files}", 'y'))
            return 1
        if self.files:
            i = 0
            for f in self.files:
                print(i, f)
                i += 1
            return 0
        else:
            return -1

    def __coord_changer(self, arg: list):
        try:
            self.now_coord
            self.__save_now_coord()
        except AttributeError:
            pass

        list_result = self.__list_coord(0)
        
        if list_result==0:
            print("Select a coordinate to open:")
            coord_file = self.files[
                int(input("\033[1;32mType in the coordinate number:\033[0m"))
            ]
            self.__open_coord("coord/" + coord_file)
        elif list_result==1:
            print(cstr("Opening default coordinate.",'b'))
            self.__open_coord('coord/'+ self.files[0])
        elif list_result==-1:
            print(cstr('No coordinate exist','r'))

    def __save_now_coord(self, save_withou_asking:bool = False):
        if save_withou_asking:
            save_or_not = 'y'
        else:
            save_or_not: str = input("Save the opened coordinate? Y/n")
        if save_or_not in ["Y", "y", "1"]:
            self.now_coord.save_coord()
            return 0
        elif save_or_not in ["N", "n", "0"]:
            return 0
        else:
            self.__save_now_coord()

    def __new_entry(self, arg: list):
        e = Entry(arg[0])
        e.set_deadline(e.date_input())
        print(cstr("Input the content of this entry, use \"Ctrl+D\" to end.\n", 'g'))
        e.content = "\n".join(sys.stdin.readlines())
        print(e.content)
        print(e.time_left)

    def __new_coord(self, arg: list):
        c = Coordinate()
        c.name = arg[0]
        raise NotImplementedError

    def __show_now_coord(self, arg: list):
        print(self.now_coord)

    def __add_coordinate(self, c: Coordinate):
        if not c.id in self.coordinate_dict:
            self.coordinate_dict[c.id] = c
        else:
            print("\033[1;31mERROR: It's already in there\033[0m")

    def __exit(self, arg):
        log_print(WARNING, "Exiting...")
        self.__save_now_coord(True)
        log_print(INFO, "Coordinate saved.")
        self.exit = True


def main():
    i = Interface()
    i.start()


if __name__ == "__main__":
    main()