def convertToDict(self, type=None, **args):
chat_id = ModelUtils.objectToId(self.chat)
send_time = ModelUtils.timeToStr(self.send_time)
# received: Object用户查询好友请求时返回的数据（accepted 为 False）
if type == "received":
return {
'fid': self.id,
'friend': self.subject.convertToDict('stranger'),
'send_time': self.send_time
}
# send: Subject用户查询好友请求时返回的数据（accepted 为 False）
if type == "send":
return {
'fid': self.id,
'friend': self.object.convertToDict('stranger'),
'send_time': self.send_time
}
# 已成为好友，获取好友的信息（accepted 为 True）
if self.accepted:
uid = args['uid']
# 如果传入参数（查询玩家的uid）为发起方ID
if uid == self.subject_id:
return {
'fid': self.id,
# 则返回的 friend 字段为目标用户的数据
'friend': self.object.convertToDict('friend'),
'send_time': send_time,
'chat_id': chat_id
}
# 如果传入参数（查询玩家的uid）为目标方ID
elif uid == self.object_id:
return {
'fid': self.id,
# 则返回的 friend 字段为发起用户的数据
'friend': self.subject.convertToDict('friend'),
'send_time': send_time,
'chat_id': chat_id
}
# 如果都不是
else:
return {
'fid': self.id,
'subject': self.subject.convertToDict('friend'),
'object': self.object.convertToDict('friend'),
'send_time': send_time,
'chat_id': chat_id
}
# 未成为好友，获取好友的信息（accepted 为 False）
else:
return {
'fid': self.id,
'sid': self.subject_id,
'oid': self.object_id,
'send_time': send_time
}
def getMemberTasks(self, user: User, status=None):
if status is None:
return ModelUtils.getObjectRelatedForFilter(user.tasktake_set,
'task',
task_project=self)
else:
return ModelUtils.getObjectRelatedForFilter(user.tasktake_set,
'task',
task__project=self,
task__status=status)
def convertToDict(self):
read_time = ModelUtils.timeToStr(self.read_time, None)
if read_time is not None:
return {'id': self.id, 'read_time': read_time}
return {'id': self.id}
def convertToDict(self, type=None):
start_date = ModelUtils.timeToStr(self.start_date)
end_date = ModelUtils.timeToStr(self.end_date)
if type == "pr":
prs = self.progressReqs()
return {
'task_name': self.name,
'prs': ModelUtils.objectsToDict(prs)
}
if type == "project":
return {
'id': self.id,
'name': self.name,
'progress': self.progress,
'level': self.level,
'status_id': self.status,
'start_date': start_date,
'end_date': end_date,
}
if type == "detail":
return self.convertToDict()
create_time = ModelUtils.timeToStr(self.create_time)
return {
'id': self.id,
'name': self.name,
'description': self.description,
'project_id': self.project.id,
'project_name': self.project.name,
'parent_id': self.parent.id,
'parent_name': self.parent.name,
'status_id': self.status,
'level': self.level,
'progress': self.progress,
'auto_adjust': self.auto_adjust,
'create_time': create_time,
'start_date': start_date,
'end_date': end_date,
}
def getRoles(self, user: User = None, uid=None):
from .views import ProjectView
if user is not None: uid = user.id
part = ProjectView.Common.getParticipation(uid, self.id)
return ModelUtils.getObjectRelatedForAll(part.participationrole_set,
'role', True)
def converToDict(self, type=None):
create_time = ModelUtils.timeToStr(self.create_time)
result_time = ModelUtils.timeToStr(self.result_time, None)
result = {
'id': self.id,
'creator_id': self.user_id,
'creator_name': self.user.name,
'task_id': self.task_id,
'progress': self.progress,
'desc': self.description,
'create_time': create_time,
}
if result_time:
result['result'] = self.result
result['result_time'] = result_time
return result
def convertToDict(self, type=None):
# 获取索引信息
if type == "index":
return {
'id': self.id,
'name': self.name,
}
if type == "pr":
tasks = self.tasks()
return {'tasks': ModelUtils.objectsToDict(tasks, type='pr')}
# 汇总信息
if type == "total":
sum_tasks, unstart_tasks, started_tasks, \
finished_tasks, progress = self.analayTasks()
return {
'name': "汇总",
'sum_tasks': sum_tasks,
'unstart_tasks': unstart_tasks,
'started_tasks': started_tasks,
'finished_tasks': finished_tasks,
'progress': progress / sum_tasks,
'total': True
}
if type == "notices":
notices = self.notice_set.all()
creator = self.creator()
return {
'id': self.id,
'name': self.name,
'creator_id': creator.id,
'creator_name': creator.name,
'notices': ModelUtils.objectsToDict(notices)
}
# 获取成员视图数据
if type == "members":
ann = self.tasks().annotate(
req_cnt=Count('taskprogress__id', distinct=True))
agg = ann.aggregate(req_cnt=Sum('req_cnt'))
members = self.members()
member_info = ModelUtils.objectsToDict(members,
type="member_info",
proj=self)
member_task = ModelUtils.objectsToDict(members,
type="member_task",
proj=self)
for i in range(members.count()):
# 玩家是否处于空闲状态 = 进行中的任务数是否为 0
member_info[i]['is_free'] = \
member_task[i]['started_tasks'] == 0
member_task[i]['rids'] = member_info[i]['rids']
member_task.append(self.convertToDict('total'))
return {
'req_cnt': agg['req_cnt'],
'member_info': member_info,
'member_task': member_task,
}
chat_id = ModelUtils.objectToId(self.chat)
create_time = ModelUtils.timeToStr(self.create_time)
start_date = ModelUtils.timeToStr(self.start_date)
end_date = ModelUtils.timeToStr(self.end_date)
res = {
'id': self.id,
'name': self.name,
'type_id': self.type,
'description': self.description,
'create_time': create_time,
'start_date': start_date,
'end_date': end_date,
'chat_id': chat_id
}
# 获取详情信息（用于详情页展示）
if type == "detail":
members = ModelUtils.objectsToDict(self.members(),
type="member",
proj=self)
tasks = ModelUtils.objectsToDict(self.tasks(True), type="project")
res['members'] = members
res['tasks'] = tasks
return res
def members(self):
return ModelUtils.getObjectRelatedForAll(self.participation_set,
'user')
class User(models.Model):
class Meta:
verbose_name = verbose_name_plural = "用户"
PASSWORD_SALT = 'd28cb767c4272d8ab91000283c67747cb2ef7cd1'
UN_LEN = 16
PWD_LEN = [8, 32]
EMAIL_REG = r'^[A-Za-z0-9一-龥]+@[a-zA-Z0-9_-]+(\.[a-zA-Z0-9_-]+)+$'
USER_EDUCATIONS = [
(UserEduaction.PrimarySchool.value, '小学'),
(UserEduaction.MiddleSchool.value, '初中'),
(UserEduaction.HighSchool.value, '高中'),
(UserEduaction.Specialty.value, '专科'),
(UserEduaction.Undergraduate.value, '本科'),
(UserEduaction.Master.value, '硕士'),
(UserEduaction.Doctor.value, '博士'),
(UserEduaction.Other.value, '其他'),
]
USER_STATUSES = [
(UserStatus.Offline.value, '离线'),
(UserStatus.Online.value, '在线'),
(UserStatus.Other.value, '其他'),
]
# 用户名
username = models.CharField(max_length=64, verbose_name="用户名")
# 密码
password = models.CharField(max_length=64, verbose_name="密码")
# 邮箱
email = models.CharField(max_length=128, verbose_name="邮箱")
# 姓名
name = models.CharField(blank=True, max_length=12, verbose_name="姓名")
# 性别（False 为男，True 为女）
gender = models.BooleanField(default=False, verbose_name="性别")
# 头像
avatar = models.ImageField(null=True,
blank=True,
verbose_name="头像",
upload_to=ModelUtils.PictureUpload('avatars'))
# 生日
birth = models.DateField(null=True, blank=True, verbose_name="生日")
# 居住地
city = models.CharField(blank=True, max_length=16, verbose_name="居住地")
# 学历
education = models.PositiveSmallIntegerField(default=0,
choices=USER_EDUCATIONS,
verbose_name="学历")
# 职位
duty = models.CharField(blank=True, max_length=16, verbose_name="职位")
# 联系方式
contact = models.CharField(blank=True, max_length=16, verbose_name="联系方式")
# 个人介绍
description = models.CharField(blank=True,
max_length=100,
verbose_name="个人介绍")
# 注册时间
create_time = models.DateTimeField(auto_now_add=True, verbose_name="注册时间")
# 登录状态
status = models.PositiveSmallIntegerField(default=0,
choices=USER_STATUSES,
verbose_name="登录状态")
# 删除标志
is_deleted = models.BooleanField(default=False, verbose_name="删除标志")
def __str__(self):
return "%d.%s(%s)" % (self.id, self.name, self.username)
def convertToDict(self, type=None, **args):
create_time = ModelUtils.timeToStr(self.create_time)
from project_module.models import Project
# 获取陌生人的资料
if type == 'stranger':
return {
'id': self.id,
'username': self.username,
'gender': int(self.gender),
'avatar': self.avatar.name,
'description': self.description,
'status_id': self.status,
}
# 获取队伍成员资料
if type == 'member' and 'proj' in args:
proj: Project = args['proj']
return {
'id': self.id,
'name': self.name,
'contact': self.contact,
'rids': proj.getRoleIds(self),
}
if type == 'member_info' and 'proj' in args:
proj: Project = args['proj']
return {
'id': self.id,
'name': self.name,
'gender': int(self.gender),
'rids': proj.getRoleIds(self),
'email': self.email,
'contact': self.contact,
'status_id': self.status
}
if type == 'member_task' and 'proj' in args:
proj: Project = args['proj']
member_tasks = proj.getMemberTasks(self)
sum_tasks, unstart_tasks, started_tasks, finished_tasks, \
progress = proj.analayTasks(member_tasks)
return {
'id': self.id,
'name': self.name,
'sum_tasks': sum_tasks,
'unstart_tasks': unstart_tasks,
'started_tasks': started_tasks,
'finished_tasks': finished_tasks,
'progress': progress / sum_tasks
}
# 获取好友的资料
if type == 'friend':
return self.convertToDict()
# 获取好友列表（已接受的）
if type == 'friends':
return {
'friends': ModelUtils.objectsToDict(self.getFriends(),
uid=self.id)
}
# 获取发起好友请求列表
if type == 'send_reqs':
return {
'reqs':
ModelUtils.objectsToDict(self.getSendFriendReqs(), type='send')
}
# 获取发起好友请求列表
if type == 'received_reqs':
return {
'reqs':
ModelUtils.objectsToDict(self.getReceivedFriendReqs(),
type='received')
}
# 个人详细资料
return {
'id': self.id,
'username': self.username,
'email': self.email,
'name': self.name,
'gender': int(self.gender),
'avatar': self.avatar.name,
'birth': self.birth,
'city': self.city,
'education_id': self.education,
'duty': self.duty,
'contact': self.contact,
'description': self.description,
'create_time': create_time,
'status_id': self.status,
}
def getFriends(self):
# 找到 subject_id 或 object_id 为自己 id（accepted 为 True）的所有好友关系
return Friend.objects.filter(Q(subject_id=self.id)
| Q(object_id=self.id),
accepted=True)
def getSendFriendReqs(self):
# 找到 subject_id 为自己 id（accepted 为 False）的所有好友关系
# 即返回用户发起的好友请求
return Friend.objects.filter(subject_id=self.id, accepted=False)
def getReceivedFriendReqs(self):
# 找到 object_id 为自己 id（accepted 为 False）的所有好友关系
# 即返回用户接收的好友请求
return Friend.objects.filter(object_id=self.id, accepted=False)
# 登陆
def login(self):
self.status = UserStatus.Online.value
self.save()
# 登出
def logout(self):
self.status = UserStatus.Offline.value
self.save()
# 发送好友请求
# TODO(吴宁): 该函数为实际操作的函数，不需检查重复添加
def addFriend(self, fuid):
Friend.objects.create(subject=self.id, object=fuid)
def convertToDict(self, type=None, **args):
create_time = ModelUtils.timeToStr(self.create_time)
from project_module.models import Project
# 获取陌生人的资料
if type == 'stranger':
return {
'id': self.id,
'username': self.username,
'gender': int(self.gender),
'avatar': self.avatar.name,
'description': self.description,
'status_id': self.status,
}
# 获取队伍成员资料
if type == 'member' and 'proj' in args:
proj: Project = args['proj']
return {
'id': self.id,
'name': self.name,
'contact': self.contact,
'rids': proj.getRoleIds(self),
}
if type == 'member_info' and 'proj' in args:
proj: Project = args['proj']
return {
'id': self.id,
'name': self.name,
'gender': int(self.gender),
'rids': proj.getRoleIds(self),
'email': self.email,
'contact': self.contact,
'status_id': self.status
}
if type == 'member_task' and 'proj' in args:
proj: Project = args['proj']
member_tasks = proj.getMemberTasks(self)
sum_tasks, unstart_tasks, started_tasks, finished_tasks, \
progress = proj.analayTasks(member_tasks)
return {
'id': self.id,
'name': self.name,
'sum_tasks': sum_tasks,
'unstart_tasks': unstart_tasks,
'started_tasks': started_tasks,
'finished_tasks': finished_tasks,
'progress': progress / sum_tasks
}
# 获取好友的资料
if type == 'friend':
return self.convertToDict()
# 获取好友列表（已接受的）
if type == 'friends':
return {
'friends': ModelUtils.objectsToDict(self.getFriends(),
uid=self.id)
}
# 获取发起好友请求列表
if type == 'send_reqs':
return {
'reqs':
ModelUtils.objectsToDict(self.getSendFriendReqs(), type='send')
}
# 获取发起好友请求列表
if type == 'received_reqs':
return {
'reqs':
ModelUtils.objectsToDict(self.getReceivedFriendReqs(),
type='received')
}
# 个人详细资料
return {
'id': self.id,
'username': self.username,
'email': self.email,
'name': self.name,
'gender': int(self.gender),
'avatar': self.avatar.name,
'birth': self.birth,
'city': self.city,
'education_id': self.education,
'duty': self.duty,
'contact': self.contact,
'description': self.description,
'create_time': create_time,
'status_id': self.status,
}