def create_project(self, data):
project_id = self._projects.insert_one(data).inserted_id
# insert log
logs_conn = MongoFactory().get_connection().get_collection("logs")
log = dict(project_id=str(project_id))
log['timelines'] = [dict(created_time=data['created_time'], type=0, creator=data['creator'])]
logs_conn.insert_one(log)
return project_id
class ProjectManager:
def __init__(self):
self._projects = MongoFactory().get_connection().get_collection(collection_name='projects')
def create_project(self, data):
project_id = self._projects.insert_one(data).inserted_id
# insert log
logs_conn = MongoFactory().get_connection().get_collection("logs")
log = dict(project_id=str(project_id))
log['timelines'] = [dict(created_time=data['created_time'], type=0, creator=data['creator'])]
logs_conn.insert_one(log)
return project_id
def is_in_applier(self, username, project_id):
project = self.find_project_by_id(project_id)
appliers = project['applier']
for applier in appliers:
if applier == username:
return 1
return 0
def is_in_team(self, username, project_id):
project = self.find_project_by_id(project_id)
members = project['team']
for member in members:
if member == username:
return 1
return 0
def apply_project(self, applier_name, project_id):
# TODO project_owner
projectapplyed = self.find_project_by_id(project_id)
projectOwner = projectapplyed['creator']
projectName = projectapplyed['name']
if (self.is_in_applier(applier_name, project_id)):
return "已存在"
else:
message.apply(applier_name, project_id, projectName, projectOwner)
return self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$push': {
'applier': applier_name
}
})
def approve_applier(self, applier_name, project_id, message_id):
# TODO project_owner
projectapplyed = self.find_project_by_id(project_id)
projectOwner = projectapplyed['creator']
projectName = projectapplyed['name']
currentPeople = projectapplyed['currentPeople'] + 1
if (self.is_in_team(applier_name, project_id)):
return '已在team中'
else:
message.accept(applier_name, project_id, projectName, projectOwner, message_id)
logs_conn = MongoFactory().get_connection().get_collection("logs")
log = dict(created_time=datetime.now(), type=1, applier=applier_name)
logs_conn.update_one({
'project_id':project_id
},{
'$push': {
'timelines':log
}
})
result = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$pull': {
'applier': applier_name
}
})
result_ = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$push': {
'team': applier_name
}
})
result__ = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$set': {
'currentPeople': currentPeople
}
})
print '同意'
return result and result_ and result__
def quit(self, username, project_id):
# TODO project_owner
if (self.is_in_team(username, project_id)):
project = self.find_project_by_id(project_id)
message.quit(username, project_id,project['name'], project['creator'])
logs_conn = MongoFactory().get_connection().get_collection("logs")
log = dict(created_time=datetime.now(), type=2, applier=username)
logs_conn.update_one({
'project_id':project_id
},{
'$push': {
'timelines':log
}
})
currentPeople = project['currentPeople'] - 1
result = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$pull': {
'team': username
}
})
result__ = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$set': {
'currentPeople': currentPeople
}
})
return result and result__
else:
return '不在项目中'
def kick_out(self, username, project_id):
# TODO project_owner
if (self.is_in_team(username, project_id)):
project = self.find_project_by_id(project_id)
currentPeople = project['currentPeople'] - 1
message.kick_out(username, project_id, "")
result = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$pull': {
'team': username
}
})
result__ = self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$set': {
'currentPeople': currentPeople
}
})
return result and result__
else:
return '不在项目中'
def start_project(self, project_id):
logs_conn = MongoFactory().get_connection().get_collection("logs")
log = dict(created_time=datetime.now(), type=3)
logs_conn.update_one({
'project_id':project_id
},{
'$push': {
'timelines':log
}
})
return self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$set': {
'status': 1
}
})
def finish_project(self, project_id):
logs_conn = MongoFactory().get_connection().get_collection("logs")
log = dict(created_time=datetime.now(), type=4)
logs_conn.update_one({
'project_id':project_id
},{
'$push': {
'timelines':log
}
})
return self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$set': {
'status': 2
}
})
def cancel_project(self, project_id):
return self._projects.update_one({
'_id': ObjectId(project_id)
}, {
'$set': {
'status': statuses.CANCELED
}
})
def project_search(self, input, page=1):
limit = 9
offset = (page - 1) * limit
projects = list(self._projects.find({
'$or':[{'name': {'$regex': input}},{'description':{'$regex':input}}]}
).skip(offset).limit(limit).sort([('created_time', pymongo.DESCENDING)]))
return projects
def find_all_project(self, page=1):
limit = 9
offset = (page - 1) * limit
return self._projects.find().skip(offset).limit(limit).sort([('created_time', pymongo.DESCENDING)])
def project_count(self):
return len(list(self._projects.find())) / 9 + 1
def find_project_by_id(self, project_id):
return self._projects.find_one({
'_id': ObjectId(project_id)
})
def find_project_by_title(self, project_title):
return self._projects.find_one({
'name': project_title
})
def find_all_projects_by_user(self, username, page_no=None, page_size=None):
return self._projects.find({
'creator': username
}).sort([
('created_time', -1)
]).skip((page_no - 1) * page_size).limit(page_size)
def get_projects_count_by_user(self, username):
return self._projects.find({'creator': username}).count()
def find_attend_projects_by_user(self, username, page_no, page_size):
return self._projects.find({
'team': username
}).sort([
('created_time', -1)
]).skip((page_no - 1) * page_size).limit(page_size)
def get_attend_project_count_by_user(self, username):
return self._projects.find({'team': username}).count()