import sys import os import datetime from peewee import * sys.path.append('../') sys.path.append('../../../') from utils.DBModels.ProjectBaseModel import BaseModel class User(BaseModel): userName = CharField() userPwd = IntegerField() createTime = CharField() description = CharField() projectAuthority = CharField() protocolSetting = CharField() userAuthority = CharField() trendAuthority = CharField() @classmethod def getByName(cls, userName): try: return cls.get(cls.userName == userName) except: return False @classmethod def deleteUser(cls, userName): query = cls.delete().where(cls.userName == userName) query.execute() # 创建用户 def createUser(self, userName, userPwd, description, projectAuthority, protocolSetting, userAuthority, trendAuthority): self.userName = userName self.userPwd = userPwd self.description = description self.projectAuthority = projectAuthority self.protocolSetting = protocolSetting self.userAuthority = userAuthority self.trendAuthority = trendAuthority self.createTime = datetime.datetime.strftime(datetime.datetime.now(), '%Y-%m-%d %H:%M:%S') print(self.createTime) self.save()