2017-10-14 109 views
0

我正在登录另一个网站并获取数据的网站上工作。我正在使用烧瓶并请求库。我的代码从html模板的输入中获取登录名和密码。一切都好,直到我尝试获得另一个页面。然后我失去登录的会话。如何保持会话?

views.py

from LibrusMobile import app 
from flask import render_template, request 
from core.forms import LoginForm 
from core.utils import Site 
import lxml.html 

@app.route('/', methods=['GET', 'POST']) 
def main_page(): 
    form = LoginForm() 
    if request.method == 'POST': 
     username=request.form['username'] 
     password=request.form['password'] 
     path=Site.user_path 
     s=Site.login(username, password) 
     url='https://link.com/index' 
     user_response=Site.get_content(Site, s, url, path) 
     user=', '.join([str(x) for x in user_response]) 

     context={ 
      "user": user.replace("(", "- "),  
     } 
     return render_template("start_page.html", context=context) 
    return render_template("index.html", form=form) 

@app.route('/marks') 
def marks(): 
    url='https://link.com/marks' 
    #Here i have a problem 
    return render_template("marks.html") 

utils.py

import requests 
import lxml.html 


class Site(): 
    user_path  = '//*[@id="user-section"]/b/text()[1]' 

    @staticmethod 
    def login(login, passwd): 
     url = 'https://link.com/login'   
     with requests.session() as s: 
      log_in=s.get(url) 
      log_in_html= lxml.html.fromstring(log_in.text) 
      hidden_inputs = log_in_html.xpath(r'//form//input[@type="hidden"]') 
      form = {x.attrib["name"]: x.attrib["value"] for x in hidden_inputs} 
      form['login']=login 
      form['passwd']=passwd 
      request=s.post(url, data=form, stream=True) 
      return s 



    def get_content(self, session, url, path): 
     response=session.get(url) 
     b=lxml.html.fromstring(response.text) 
     text=b.xpath(path) 
     return text 

回答

0

,可以储存用户名和密码的地方(例如:flask.session),并检查用户是否登录,你有充分的路线上(或也许在app.before_request)。如果没有 - 重定向到登录表单,否则 - 使用存储的登录名/密码请求

+0

我可以将会话保存在Cookie中吗?这是不错的做法吗? –

+0

btw我不想保留cookies中的用户名和密码 –

+0

@AdrianOleszczak查看[Flask-Session](https://pythonhosted.org/Flask-Session/),它是Flask的扩展,它增加了对Server-侧会话到您的应用程序。 –