You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
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()
|
|
projectType = IntegerField()
|
|
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, proType, des):
|
|
self.projectName = name
|
|
self.projectType = proType
|
|
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()
|
|
|
|
|