如何在Django的视图之间传递对象?

时间:2020-01-19 03:07:27

标签: javascript python django ajax selenium-chromedriver

背景故事/问题

我有一个运行各种作业的Web应用程序。这些工作之一包括运行登录到linkedin的python脚本。

当我运行脚本时,将启动一个无头铬实例,并导航至linkedin登录页面并提交适当的凭据。到目前为止,一切正常,但有时linkedin会将我的请求发送到检查点网址,并提示我提交发送到我的电子邮件的特殊密码,然后再继续。

在我看来,我很难理解如何实现这种条件逻辑。 基本上,我希望有一种方法可以在Web应用程序中输入密码。

预期结果

从django视图中运行脚本以登录到linkedin,如果到达检查点,则提示用户输入密码,否则,如果没有到达检查点并且登录成功,则继续输入密码。

我尝试过的

  • Google-找不到与我的问题有关的任何内容。

  • 下面的代码-下面的视图显示了我当前的尝试。我知道这是可以做到的,但是我不确定是否应该尝试在单个视图中构建它,还是应该将我的Chrome实例传递给另一个视图(对象无法序列化,所以我不确定我该怎么做;也许我可以将chrome id传递给我的webdriver的另一个实例,但是我认为那不是最好的方法?)

  • Ajax? -我实际上没有尝试过,但是我知道这是一个选择。

注释**

下面包含的代码Ive就是一切。但是要考虑的唯一重要部分是视图,我认为添加其余部分可能会阐明逻辑并帮助某人帮助我。希望这不是太冗长!

查看

def verify_linkedin(request):
    """Used for verifying login to linkedin with email pincode"""
    ln = LinkedinScraper()
    if ln.login_to_linkedin(username=os.getenv("LINKEDIN_USERNAME"), password=os.getenv("LINKEDIN_PASSWORD")):
        return HttpResponse("<h1>You are verified</h1>")
    else:
        if request.method == "POST":
            form = VerifyLinkedinForm(request.POST)
            if form.is_valid():
                print(form.cleaned_data)
                pin_code = form['verification_key']
                ln.authenticate_linkedin_login(pin_code)
                return redirect("home")
        else:
            form = VerifyLinkedinForm()

    context = {"form": form}
    return render(request, "linkedin_verify.html", context)

表格

class VerifyLinkedinForm(forms.Form):
    verification_key = forms.CharField()
    helper = FormHelper()
    helper.form_method = "post"
    helper.layout = Layout(
        'verification_key',
        FormActions(
            Submit("submit", "Verify Linkedin", style="display: block; margin: auto; margin-bottom:2em;",
                css_class="btn btn-success", ),
        )
    )

模板

{% extends 'base.html' %}
{% load crispy_forms_tags %}

{% block content %}
  {% crispy form %}
{% endblock %}

Webscraper后端

class LinkedinScraper(WebScraper):
    """Class for logging into linkedin with chromedriver and scraping linkedin"""

    def __init__(self, headless=True, **kwargs):
        super(LinkedinScraper, self).__init__()
        self.setup_driver(headless=headless)
        self.data = {}
        self.pause = 10
        self.retries = 0
        self.max_retries = 3

    def login_to_linkedin(self, username: str, password: str) -> bool:
        """ Navigates to Linkedin login page and logs in with credentials"""
        login_url = "https://www.linkedin.com/login"
        self.driver.get(login_url)
        user_element = self.driver.find_element("id", "username")
        pass_element = self.driver.find_element("id", "password")
        login_btn = self.driver.find_element_by_class_name(
            "login__form_action_container"
        )
        self.login_and_authenticate(login_btn, user_element, pass_element, username, password)
        if 'checkpoint' in self.driver.current_url:
            return False
        else:
            return True

    def authenticate_linkedin_login(self, pin_code: str):
        """ Sometimes when logging into linkedin from a new ip address, linkedin will flag account for suscipicous access
        If this happens we need to use a pin code sent to email and input it into the form.

        This should be immedidately called after login_and_authenticate returns False
        """
        if 'checkpoint' in self.driver.current_url:
            verification_element = self.driver.find_element_by_xpath("//input[@id='input__email_verification_pin']")
            print(pin_code)
            self.random_send_keys(element=verification_element, keys=pin_code)
            submit_element = self.driver.find_element("id", "email-pin-submit-button")
            submit_element.click()
            return True
        else:
            raise BaseException(f"Driver not on correct url, current url is {self.driver.current_url} -- Should contain checkpoint")

1 个答案:

答案 0 :(得分:0)

有两种方法可以解决此问题。 一种方法是在视图外部创建对象实例,这有点麻烦,而且完全不可扩展,但是我能够通过Webrowser使用硒验证我的电子邮件帐户。我不必在视图之间传递对象,因为它是全局的。

第二种方法是不使用硒。我最终要做的是。 如果有人对此方法感兴趣,请给我留言或研究pyppeteer。