wechat_assistant.py 文件源码

def user_command(func):
    def weppar(msg):
        db = models.DBSession()
        user = db.query(models.User).filter_by(wechat_id=msg['FromUserName']).first()
        if user:
            if user.name:
                if 'Content' in msg and msg['Content'][0] == '\\':
                    result = command(msg['Content'][1:], user)
                else:
                    result = func(msg, user, db)
            else:
                user.name = msg['Content'].strip()
                result = f'???????: {user.name}, ?????????? \\name_<?????>'
        else:
            user = models.User(wechat_id=msg['FromUserName'])
            db.add(user)
            result = '???????????'
        db.commit()
        db.close()
        return result

    return weppar