Как я могу использовать LOCAL VARIABLE в функции, как переменная GLOBAL VARIABLE для другой функции - PullRequest
Новая
       35

Как я могу использовать LOCAL VARIABLE в функции, как переменная GLOBAL VARIABLE для другой функции

0 голосов
/ 09 января 2019

Я написал простой тест на селен, который выполняет аутентификацию на каком-либо сайте, анализирует некоторую часть HTML-кода и отправляет электронную почту в нужный почтовый ящик. Я хочу сделать все красиво, и я решил определить функции для этого кода.

Мой код: 

from selenium.webdriver.support.ui import Select
from selenium.webdriver.chrome.options import Options
from selenium import webdriver
import smtplib

CHROME_PATH = '/usr/bin/google-chrome'
CHROMEDRIVER_PATH = '/usr/local/bin/chromedriver'
WINDOW_SIZE = "1024,768"

def pgi_auth(sitename, region_pgi, login_pgi, password_pgi):
    chrome_options = Options()
    chrome_options.add_argument("--headless")
    chrome_options.add_argument("--window-size=%s" % WINDOW_SIZE)
    chrome_options.binary_location = CHROME_PATH
    driver = webdriver.Chrome(executable_path=CHROMEDRIVER_PATH,chrome_options=chrome_options)
    driver.get(sitename)
    driver.find_element_by_id("LoginId").send_keys(login_pgi)
    driver.find_element_by_id("LoginPassword").send_keys(password_pgi)

    option = Select(
        driver.find_element_by_xpath("//select[@name='Region']")
    ).select_by_value(region_pgi)

    driver.find_element_by_id("submit").click()

    element = driver.find_element_by_id("TotalCurrentCharges-body").text

    current_charges = element.splitlines()[1]
    print(current_charges)
    driver.quit()

def send_email(host, port, from_addr, password, to_addr, subject, current_charges):
    BODY = "\r\n".join((
        "From: %s" % from_addr,
        "To: %s" % to_addr,
        "Subject: %s" % subject ,
        "",
        body_text
    ))
    server = smtplib.SMTP(host, port)
    server.starttls()
    server.login(from_addr, password)
    server.sendmail(from_addr, [to_addr], BODY)
    server.quit()

#for pgi_auth
sitename="https://invoiceaccess.pgiconnect.com"
login_pgi="my_secret_login"
password_pgi="my_secret_password"
region_pgi="us"

#for send_email
host = "mail.example.com"
port = "587"
from_addr = "test.mail@example.com"
password = "my_secret_password"
to_addr = "test.receive@example.com"
subject = "MY BALANCE:"

pgi_auth(sitename, region_pgi, login_pgi, password_pgi)
send_email(host, port, from_addr, password, to_addr, subject, current_charges)

В моей функции pgi_auth я получаю переменную current_charges Я хочу использовать эту переменную вне этой pgi_auth для функции send_email. Конечно, я могу сделать это без функций, но я хочу попытаться сделать все красиво (возможно, я буду использовать config), как я могу получить эту переменную вне функции?

1 Ответ

0 голосов
/ 09 января 2019

вам нужно return ключевое слово в методе 

def pgi_auth(sitename, region_pgi, login_pgi, password_pgi):
    .....
    .....
    return current_charges

....
....
subject = "MY BALANCE:"

current_charges = pgi_auth(sitename, region_pgi, login_pgi, password_pgi)
send_email(host, port, from_addr, password, to_addr, subject, current_charges)
...