import sys
import os
import datetime
from peewee import *
sys.path.append('../')
sys.path.append('../../../')
from utils.DBModels.BaseModel import *



class Project(BaseModel):
    projectName = CharField()
    createTime = CharField()
    description = CharField()

    # 查询工程是否存在
    @classmethod
    def getByName(cls, projectName):
        try:
            return cls.get(cls.projectName == str(projectName))
        except Exception as e:
            return print(e)


    # 删除工程
    @classmethod
    def deleteProject(cls, name):
        try:
            query = cls.delete().where(cls.projectName == name)
            query.execute()
            return 'ok'
        except Exception as e:
            return e

    # 创建工程
    def createProject(self, name, des):
        self.projectName = name
        self.description = des
        self.createTime = datetime.datetime.strftime(datetime.datetime.now(),'%Y-%m-%d %H:%M:%S')
        # print(self.createTime)
        self.save()

class ClientDB(BaseModel):
    currentProject = CharField()
    value = CharField()

    # 查询工程是否存在
    @classmethod
    def getByName(cls):
        try:
            return cls.get(cls.currentProject == '0')
        except Exception as e:
            print(e)


    def createMes(self, value):
        self.value = value
        self.currentProject = '0'
        self.save()