def disable_user(user_id):
session = database.get_session()
user = session.query(UserProfile).filter(UserProfile.UserId == user_id).update({'Status':UserStatus.Disabled})
session.commit()
session.close()
def isexist(classname):
session = database.get_session()
existcount = session.query(Classes).filter(Classes.ClassName == classname).count()
session.close()
if existcount > 0:
return True
else:
return False
def getAll(projectId):
session = database.get_session()
memberInProject = session.query(
Member.UserId).filter(Member.ProjectId == projectId)
members = session.query(UserProfile).filter(
not_(UserProfile.UserId.in_(memberInProject)))
session.close()
return members
def query_case(page_no,page_size,order_by,current_user,current_project):
session = database.get_session()
caseid_list = session.query(CaseId).filter(and_(Creator== current_user, ProjectId==current_project))
case_list = session.query(CaseId,CaseName,Description,CreateDate,LastUpdateDate,Versions,CaseURL).filter(case.CaseId.in_(caseid_list))
(data,row_count,page_count,page_no) = database.query_more(case_list,order_by,page_no,page_size)
session.close()
return (data,row_count,page_count,page_no)
def isexist(modelname, projectid):
session = database.get_session()
existcount = session.query(Model).filter(
Model.ProjectId == projectid).filter(
Model.ModelName == modelname).count()
session.close()
isExist = True if existcount > 0 else False
return isExist
def query(projectid):
session = database.get_session()
if projectid != 'all':
models = session.query(Model).filter(Model.ProjectId == projectid).all()
else:
models = session.query(Model).all()
session.close()
return models
def assign_admin(user_id):
session = database.get_session()
user = session.query(UserProfile).filter(UserProfile.UserId == user_id).one()
user.IsAdmin = not user.IsAdmin
session.commit()
session.close()
def query(projectid):
session = database.get_session()
if projectid != 'all':
models = session.query(Model).filter(
Model.ProjectId == projectid).all()
else:
models = session.query(Model).all()
session.close()
return models
def isexist(classname):
session = database.get_session()
existcount = session.query(Classes).filter(
Classes.ClassName == classname).count()
session.close()
if existcount > 0:
return True
else:
return False
def query():
session = database.get_session()
classlist = session.query(Classes).all()
class_list = []
name_list = []
for i in classlist:
if not i.ClassName in name_list:
class_list.append(i)
name_list.append(i.ClassName)
return class_list
def udpate_case(case_name,description,versions,Case_Id):
session = database.get_session()
case = session.query(case).filter(case.CaseId == Case_Id).first()
case.CaseName = case_name
case.Description = description
case.Versions = versions
case.LastUpdateDate = datetime.now()
session.commit()
session.close()
return True
def query():
session = database.get_session()
classlist = session.query(Classes).all()
class_list = []
name_list = []
for i in classlist:
if not i.ClassName in name_list:
class_list.append(i)
name_list.append(i.ClassName)
return class_list
def addUser(projectId,email):
session = database.get_session()
user = userservice.get(email)
member = Member()
member.ProjectId = projectId
member.UserId = user.UserId
session.add(member)
session.commit()
session.close()
def addUser(projectId, email):
session = database.get_session()
user = userservice.get(email)
member = Member()
member.ProjectId = projectId
member.UserId = user.UserId
session.add(member)
session.commit()
session.close()
def create_one(classname,project,creator):
session = database.get_session()
c = Classes()
c.ClassName = classname.strip()
c.Creator = creator
c.ProjectId = project
c.CreateDate = datetime.now()
c.LastUpdateDate = datetime.now()
session.add(c)
session.commit()
session.close()
def create_one(classname, project, creator):
session = database.get_session()
c = Classes()
c.ClassName = classname.strip()
c.Creator = creator
c.ProjectId = project
c.CreateDate = datetime.now()
c.LastUpdateDate = datetime.now()
session.add(c)
session.commit()
session.close()
def change_password(raw_password,new_password,user_id):
session = database.get_session()
user = session.query(UserProfile).filter(UserProfile.UserId == user_id).first()
if not user.Password == raw_password:
session.close()
return False
user.Password = new_password
session.commit()
session.close()
return True
def update(modelId, modelName, description):
session = database.get_session()
session.query(Model).filter(Model.ModelId == modelId).update({
'ModelName':
modelName.strip(),
'Description':
description.strip(),
'LastUpdateDate':
datetime.now()
})
session.commit()
session.close()
def update(projectId, projectName, introduction):
session = database.get_session()
session.query(Project).filter(Project.ProjectId == projectId).update({
'ProjectName':
projectName.strip(),
'Introduction':
introduction.strip(),
'LastUpdateDate':
datetime.now()
})
session.commit()
session.close()
def create_case(case_name,description,versions,caseurl,creator):
session = database.get_session()
p = case()
p.CaseName = case_name.stSrip()
p.Creator = creator
p.Description = description
p.Versions = versions
p.CaseURL = caseurl
p.CreateDate = datetime.now()
p.LastUpdateDate = datetime.now()
p.DownCount = 0
session.add(p)
session.commit()
session.close()
def delete(projectid):
session = database.get_session()
#先删除项目成员信息
members = session.query(Member).filter(Member.ProjectId == projectid).all()
for m in members:
teamservice.delete(projectid)
#再删除该项目下的模块
models = session.query(Model).filter(Model.ProjectId == projectid).all()
for m in models:
modelservice.delete(m.ModelId)
session.query(Project).filter(Project.ProjectId == projectid).delete()
session.commit()
session.close()
def delete(projectid):
session = database.get_session()
#先删除项目成员信息
members = session.query(Member).filter(Member.ProjectId == projectid).all()
for m in members:
teamservice.delete(projectid)
#再删除该项目下的模块
models = session.query(Model).filter(Model.ProjectId == projectid).all()
for m in models:
modelservice.delete(m.ModelId)
session.query(Project).filter(Project.ProjectId == projectid).delete()
session.commit()
session.close()
def create(project_name, project_introduction, creator):
session = database.get_session()
p = Project()
p.ProjectName = project_name.strip()
p.Creator = creator
p.CreateDate = datetime.now()
p.LastUpdateDate = datetime.now()
p.Introduction = project_introduction
m = Member()
m.UserId = creator
p.Members.append(m)
session.add(p)
session.commit()
session.close()
def create(project_name,project_introduction,creator):
session = database.get_session()
p = Project()
p.ProjectName = project_name.strip()
p.Creator = creator
p.CreateDate = datetime.now()
p.LastUpdateDate = datetime.now()
p.Introduction = project_introduction
m = Member()
m.UserId = creator
p.Members.append(m)
session.add(p)
session.commit()
session.close()
def query(page_no,page_size,order_by,current_user,class_name):
session = database.get_session()
projectid_list = session.query(Member.ProjectId).filter(Member.UserId == current_user)
subdata = ""
subdata_list = []
if class_name != "all":
subdata = session.query(Classes.ProjectId).filter(Classes.ClassName == class_name).all()
for i in subdata:
subdata_list.append(i[0])
project_list = session.query(Project).filter(Project.ProjectId.in_(projectid_list))
(data,row_count,page_count,page_no) = database.query_more(project_list,order_by,page_no,page_size)
session.close()
return (data,subdata_list,row_count,page_count,page_no)
def create(modelname, description, project, creator):
isExist = isexist(modelname, project)
if not isExist:
session = database.get_session()
m = Model()
m.ModelName = modelname.strip()
m.Description = description
m.ProjectId = project
m.Creator = creator
m.CreateDate = datetime.now()
m.LastUpdateDate = datetime.now()
session.add(m)
session.commit()
session.close()
return isExist
def create(modelname,description,project,creator):
isExist = isexist(modelname,project)
if not isExist:
session = database.get_session()
m = Model()
m.ModelName = modelname.strip()
m.Description = description
m.ProjectId = project
m.Creator = creator
m.CreateDate = datetime.now()
m.LastUpdateDate = datetime.now()
session.add(m)
session.commit()
session.close()
return isExist
def udpate_profile(email,nick,user_id):
email = email.strip()
nick = nick.strip()
session = database.get_session()
user = session.query(UserProfile).filter(UserProfile.UserId == user_id).first()
if not user.Email == email:
exist = session.query(UserProfile).filter(UserProfile.Email == email).count() > 0
if exist:
session.close()
return False
else :
user.Email = email
user.Nick = nick
session.commit()
session.close()
return True
def query(page_no, page_size, order_by, current_user, class_name):
session = database.get_session()
projectid_list = session.query(
Member.ProjectId).filter(Member.UserId == current_user)
subdata = ""
subdata_list = []
if class_name != "all":
subdata = session.query(
Classes.ProjectId).filter(Classes.ClassName == class_name).all()
for i in subdata:
subdata_list.append(i[0])
project_list = session.query(Project).filter(
Project.ProjectId.in_(projectid_list))
(data, row_count, page_count,
page_no) = database.query_more(project_list, order_by, page_no, page_size)
session.close()
return (data, subdata_list, row_count, page_count, page_no)
def register(email,nick,password):
email = email.strip()
nick = nick.strip()
session = database.get_session()
exist = session.query(UserProfile).filter(UserProfile.Email == email).count() > 0
userid = -1
if not exist :
user = UserProfile()
user.Email = email
user.Nick = nick
user.Password = password
user.EmailVerify = False
user.Status = UserStatus.Enabled
user.IsAdmin = False
user.RegDate = datetime.now()
session.add(user)
session.commit()
userid = user.UserId
session.close()
return (exist,userid)
def delete(projectId):
session = database.get_session()
session.query(Member).filter(Member.ProjectId == projectId).delete()
session.commit()
session.close()
def delete(class_name):
session = database.get_session()
session.query(Classes).filter(Classes.ClassName == class_name).delete()
session.commit()
session.close()
def isexist(modelname,projectid):
session = database.get_session()
existcount = session.query(Model).filter(Model.ProjectId == projectid).filter(Model.ModelName == modelname).count()
session.close()
isExist = True if existcount > 0 else False
return isExist
def delete(modelId):
session = database.get_session()
session.query(Model).filter(Model.ModelId == modelId).delete()
session.commit()
session.close()
def update(modelId, modelName, description):
session = database.get_session()
session.query(Model).filter(Model.ModelId == modelId).update({'ModelName':modelName.strip(),'Description':description.strip(),'LastUpdateDate':datetime.now()})
session.commit()
session.close()
def delete(class_name):
session = database.get_session()
session.query(Classes).filter(Classes.ClassName == class_name).delete()
session.commit()
session.close()
def delete(modelId):
session = database.get_session()
session.query(Model).filter(Model.ModelId == modelId).delete()
session.commit()
session.close()
# -*- coding: UTF-8 -*-
import sys
from testTeam.models import database, UserProfile, UserStatus
from datetime import datetime
if '-dropcreate' in sys.argv:
database.drop_database()
print(u'删除数据库完成')
database.create_database()
print(u'创建数据库完成')
session = database.get_session()
admin = UserProfile()
admin.Email = '*****@*****.**'
admin.Nick = u'admin'
admin.Password = '******'
admin.Status = UserStatus.Enabled
admin.IsAdmin = True
admin.RegDate = datetime.now()
session.add(admin)
'''
bug = IssueCategory()
bug.CategoryName = u'Bug'
bug.Status = IssueCategoryStatus.Enabled
issue = IssueCategory()
issue.CategoryName = u'Issue'
issue.Status = IssueCategoryStatus.Enabled
# -*- coding: UTF-8 -*-
import sys
from testTeam.models import database,UserProfile,UserStatus
from datetime import datetime
if '-dropcreate' in sys.argv:
database.drop_database()
print(u'删除数据库完成')
database.create_database()
print(u'创建数据库完成')
session = database.get_session()
admin = UserProfile()
admin.Email = '*****@*****.**'
admin.Nick = u'admin'
admin.Password = '******'
admin.Status = UserStatus.Enabled
admin.IsAdmin = True
admin.RegDate = datetime.now()
session.add(admin)
'''
bug = IssueCategory()
bug.CategoryName = u'Bug'
bug.Status = IssueCategoryStatus.Enabled
issue = IssueCategory()
issue.CategoryName = u'Issue'
def get(email):
session = database.get_session()
user = session.query(UserProfile).filter(UserProfile.Email == email).first()
session.close()
return user
def removeUser(projectId, userId):
session = database.get_session()
session.query(Member).filter(Member.ProjectId == projectId).filter(
Member.UserId == userId).delete()
session.commit()
session.close()
def delete(projectId):
session = database.get_session()
session.query(Member).filter(Member.ProjectId==projectId).delete()
session.commit()
session.close()
def removeUser(projectId,userId):
session = database.get_session()
session.query(Member).filter(Member.ProjectId==projectId).filter(Member.UserId==userId).delete()
session.commit()
session.close()
def update(projectId,projectName,introduction):
session = database.get_session()
session.query(Project).filter(Project.ProjectId == projectId).update({'ProjectName':projectName.strip(),'Introduction':introduction.strip(),'LastUpdateDate':datetime.now()})
session.commit()
session.close()
def del_case(Case_Id):
session = database.get_session()
model = session.query(case).filter(case.CaseId == Case_Id).first()
session.delete(model)
session.commit()
return True
def get(projectid):
session = database.get_session()
project = session.query(Project).filter(
Project.ProjectId == projectid).one()
session.close()
return project
def get(projectid):
session = database.get_session()
project = session.query(Project).filter(Project.ProjectId == projectid).one()
session.close()
return project
def getAll(projectId):
session = database.get_session()
memberInProject = session.query(Member.UserId).filter(Member.ProjectId==projectId)
members = session.query(UserProfile).filter(not_(UserProfile.UserId.in_(memberInProject)))
session.close()
return members
def get_user_by_id(user_id):
session = database.get_session()
user = session.query(UserProfile).filter(UserProfile.UserId == user_id).one()
session.close()
return user
