From 85d5092028406ad7df87901d5e93e1006911d954 Mon Sep 17 00:00:00 2001 From: Jesse Hitch Date: Sun, 18 Dec 2022 00:04:40 +0100 Subject: [PATCH] add server variable --- powerline/segments/common/mail.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/powerline/segments/common/mail.py b/powerline/segments/common/mail.py index 852fe2c9..01f029be 100644 --- a/powerline/segments/common/mail.py +++ b/powerline/segments/common/mail.py @@ -19,7 +19,7 @@ class EmailIMAPSegment(KwThreadedSegment): interval = 60 @staticmethod - def key(username, password, username_variable, password_variable, server='imap.gmail.com', port=IMAP4_SSL_PORT, folder='INBOX', use_ssl=None, **kwargs): + def key(username, password, server='imap.gmail.com', port=IMAP4_SSL_PORT, folder='INBOX', use_ssl=None, **kwargs): if use_ssl is None: use_ssl = (port == IMAP4_SSL_PORT) # catch if user set custom mail credential env variables @@ -27,6 +27,8 @@ class EmailIMAPSegment(KwThreadedSegment): username = os.environ.get(username_variable, None) if password_variable: password = os.environ.get(password_variable, None) + if server_variable: + server = os.enviorn.get(server_variable, server) return _IMAPKey(username, password, server, port, folder, use_ssl) @@ -68,12 +70,14 @@ email_imap_alert = with_docstring(EmailIMAPSegment(), login username :param str password: login password +:param str server: + e-mail server :param str username_variable: name of environment variable to check for login username :param str password_variable: name of environment variable to check for login password -:param str server: - e-mail server +:param str server_variable: + name of environment variable to check for email server :param int port: e-mail server port :param str folder: