py-hutun-robot/api/login.py

"""
登录接口
"""

from flask import Blueprint
from flask import request

import api
import huitun
import xhs

login_opt = Blueprint('login', __name__)


@login_opt.route('/huitun', methods=["POST"])
def huitun_login():
    """
    登录接口
    """
    request_body = request.json
    phone = request_body.get('phone')
    browser = huitun.HuiTunBrowser(phone)
    login_result = browser.login(request_body.get('password'))
    return api.success(login_result)


@login_opt.route('/xhs', methods=["POST"])
def xhs_loigin():
    """
    登录接口
    """
    request_body = request.json
    phone = request_body.get('phone')
    browser = xhs.XhsBrowser(phone)
    browser.login()
    return api.success()