コード例 #1
0
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')
コード例 #2
0
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)
コード例 #3
0
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, '')
コード例 #4
0
 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)
コード例 #5
0
    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'])
コード例 #6
0
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')
コード例 #7
0
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)
コード例 #8
0
ファイル: mate.py プロジェクト: luckyliaolei/bangni-cloud
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, '')
コード例 #9
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))
コード例 #10
0
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