def delfile():
file_hash = request.args.get('file_hash')
users.update_one({'_id': ObjectId(session['user_id'])},
{'$pullAll': {
'files': [file_hash]
}})
return tojson(1, 'delete success')
def hello_world():
user_id = request.args.get('user_id')
if request.method == 'POST':
profile = dict(parse.parse_qsl(request.form.get('profile')))
condition = dict(parse.parse_qsl(request.form.get('condition')))
profile['images'] = request.form.getlist('images')
fs = request.files.getlist('files')
for f in fs:
u = uuid.uuid1().hex
f.save('static/pictures/' + u + '.jpg')
profile['images'].append(u)
users.update_one({
'_id': user_id
}, {
'$set': {
'profile': profile,
'condition': condition
}
}, upsert=True)
return json.dumps({
'res': 0
})
else:
user = users.find_one({
'_id': user_id
}) or {}
profile = user.get('profile', {})
condition = user.get('condition', {})
return render_template('edit.html', profile=profile, condition=condition)
def success():
file_hash = request.args.get('file_hash')
files.update_one({'_id': file_hash},
{'$currentDate': {
'success_date': True
}})
users.update_one({'_id': ObjectId(session['user_id'])},
{'$push': {
'files': file_hash
}})
return tojson(True, '')
def changepassword():
form = ChangePassword()
user = users.find_one({'username': current_user.get_id()})
if form.validate_on_submit():
users.update_one(user, {
'$set': {
'password': generate_password_hash(form.password.data)
}
})
flash('Password Changed', 'info')
return redirect(url_for('index'))
return render_template('changepassword.html',
title="Change Password",
form=form)
def answer(self, question_id, question_index):
last_qu, next_qu = questions.find_one({
'_id': question_id
}, {
'content': {
'$slice': [question_index, 2]
}
})['content']
if self.msg.type == 'text' and self.msg.content == '过':
users.update_one({
'_id': self.msg.source
}, {
'$set': {
'answering.question_index': question_index + 1
}
})
else:
# 下面是验证答案的有效性。
# 类型验证
if last_qu.get('type') != self.msg.type:
return self.text_reply('你的回答类型有误, 回答要求是%s类型。' % last_qu.get('type'))
# 正则验证
regex = last_qu.get('regex')
if regex and not re.match(regex, self.msg.content):
return self.text_reply(last_qu['error'])
if last_qu.get('type') == 'image':
self.download_img()
users.update_one({
'_id': self.msg.source
}, {
'$push': {
'profile.' + last_qu['filed']: self.msg.media_id
}
})
return self.text_reply('嗯嗯，再来一张？回复“过”跳过该问题')
users.update_one({
'_id': self.msg.source
}, {
'$set': {
question_id + '.' + last_qu['filed']: self.msg.content,
'answering.question_index': question_index + 1
}
})
if not next_qu.get('filed'): # 间接判断是否为最后一问，但最好还是直接判断(idx==len)以增强可读性和代码健壮性。
users.update_one({
'_id': self.msg.source
}, {
'$unset': {
'answering': ''
} # 最后一个问题没有答案，所以在回复之前即可删掉answering状态字段。
})
return self.text_reply(next_qu['question'])
def mate():
filename = request.args.get('filename')
file_hash = request.args.get('file_hash')
if not files.find_one({'_id': file_hash}):
files.insert_one({
'_id': file_hash,
'filename': filename,
'chunks': []
}) # {chunk_hash: '', nodes: []}
users.update_one({'_id': ObjectId(session['user_id'])},
{'$push': {
'files': file_hash
}})
return tojson(0, '')
else:
return tojson(1, 'Already have this file')
def recommend_page():
user_id = request.args.get('user_id')
if request.method == 'POST':
action = request.form.get('action')
value = request.form.get('value') == 'true'
uu = request.form.get('uuid')
users.update_one({
'_id': user_id,
'likes.uuid': uu}, {
'$set': {"likes.$." + action: value}
})
return json.dumps({'res': 0})
else:
idx = request.args.get('idx', 0)
recommend_user, like = recommend(user_id, int(idx))
profile = recommend_user.get('profile', {})
condition = recommend_user.get('condition', {})
return render_template('recommend.html', profile=profile, condition=condition, like=like)
def mate():
filename = request.args.get('filename')
file_hash = request.args.get('file_hash')
if files.find_one({
'_id': file_hash}):
return tojson(1, 'success')
elif users.find_one({
'files': {
'$all': [file_hash]}}):
return tojson(2, 'file exist')
else:
files.insert_one({
'_id': file_hash,
'filename': filename,
'chunks': []}) # {chunk_hash: '', nodes: []}
users.update_one({
'_id': ObjectId(session['user_id'])}, {
'$push': {
'files': file_hash}})
return tojson(0, '')
def msg_hendle(self):
user = users.find_one({
'_id': self.msg.source
})
if not user:
user = {
"_id": self.msg.source,
"profile": {},
"condition": {}
}
users.insert_one(user)
if self.msg.type == 'event':
if self.msg.event == 'subscribe':
return self.text_reply(
'感谢你关注本订阅号，我们承诺不泄露隐私信息，你的信息将会在取消关注本订阅号后自动删除，回复“资料”启动智能小客服采集资料，也可以点击\n%s\n通过网页进行填写。由于公众号的限制我们不能主动给你发消息，只能被动恢复消息，所以并不是我们不理你。有事儿没事儿多回复“在吗”，将会有不一样的惊喜等着你哦。' % self.edit_url)
elif self.msg.event == "unsubscribe":
pass
else:
pass
if user.get('answering'): # 用户正在回答问题, 调用anwser函数处理答案。
return self.answer(user['answering']['question_id'], user['answering']['question_index'])
else: # 用户发了段新信息，判断是否为key
if self.msg.type != 'text':
return self.text_reply('额，看不是很懂耶。')
question = questions.find_one({
'key': self.msg.content
}, {
'_id': 1
})
if question:
if user.get(question['_id']):
return self.text_reply('已经回答过该问题, 查看资料请访问')
users.update_one({
'_id': self.msg.source
}, {
'$set': {
'answering': {
"question_time": self.msg.create_time,
"question_id": question['_id'],
"question_index": 0
}
}
})
first_qu = questions.find_one({
'_id': question['_id']
}, {
'content': {
'$slice': [0, 1]
}
})['content'][0]
return self.text_reply(first_qu['question'])
elif self.msg.content == 'test':
return self.text_reply(self.get_url('test', '测试'))
if not user.get('profile'):
return self.text_reply('你好，是不是有点无聊了，你好像还没上传基本资料，你可以点击\n%s\n或回复“资料”启动智能小客服采集资料？' % self.edit_url)
if not user.get('condition'):
return self.text_reply('你好，是不是有点无聊了，你好像还没上传心动条件，你可以点击\n%s\n或回复“条件”启动智能小客服采集资料？' % self.edit_url)
return self.text_reply(
'你好，欢迎来到小红娘公众号，快来看看小红娘都有哪些功能吧:\n\n%s\n\n%s\n\n%s\n\n%s' % (self.recommend_url, self.info_url, self.edit_url, self.achievement_url))
def recommend(user_id, idx):
today = datetime.date.today()
recommended = list(users.aggregate(
[{
"$match": {
"_id": user_id
}
}, {
"$unwind": "$likes"
}, {
"$match": {
'likes.date': {
'$gte': datetime.datetime(today.year, today.month, today.day)
}
}
}, {
"$lookup": {
"from": "users",
"localField": "likes.user_id",
"foreignField": "_id",
"as": "nodes"
}
}, {
"$project": {
"_id": 0,
"gender": '$profile.gender',
"likes": 1,
"nodes": { '$arrayElemAt': ["$nodes", 0]}
}
}, {
'$skip': idx
}, {
'$limit': 1
}]))
if not recommended:
user = users.find_one({
'_id': user_id
})
gender = user['profile'].get('gender', '女') == '男' and '女' or '男'
charm = user.get('charm', 0.5)
expect = user.get('expect', 0.5)
recommend_user = users.find_one({
'profile.gender': gender
}) # recommend arithmetic.
uu = base64.b64encode(uuid.uuid1().bytes).decode()
like = {
'user_id': recommend_user['_id'],
'uuid': uu,
'date': datetime.datetime.now(),
'like': False,
'store': False
}
users.update_one({
'_id': user_id
}, {
'$push': {
'likes': like
}
}, upsert=True)
else:
recommend_user = recommended[0]['nodes']
like = recommended[0]['likes']
return recommend_user, like