Script to get a code from a letter without authorization

выглядит солидно, оформлено красиво

 

Топ

 

А че так можно было?

 

Пару комментов по коду - надо всё таки разделять код на зоны ответственности, а то у тебя всё в одной куче (хорошо хоть проверку письма вынес в отдельный файл) - я слегка модифицировал код и оставил пару комментов

import email
import imaplib
import re

from PyQt5.QtCore import QThread, pyqtSignal


class GetCode(QThread):
send_code = pyqtSignal(str)

def __init__(self, mail_data, regular_string):
super().__init__()
self.mail_data = mail_data
self.regular_string = regular_string

def get_last_email_text(imap_server, username, password):
# Подключение к почтовому серверу
mail = imaplib.IMAP4_SSL(imap_server)
mail.login(username, password)

# Выбор почтового ящика (в данном случае - входящие)
mail.select('INBOX')

# Поиск последнего письма
result, data = mail.search(None, 'ALL')

# Тут может оказаться что почтовый ящик пустой и в нем нет писем вообще - ты это не проверяешь
last_email_id = data[0].split()[-1]
result, data = mail.fetch(last_email_id, '(RFC822)')

# Извлечение текста письма
raw_email = data[0][1]
email_message = email.message_from_bytes(raw_email)
text = ""

if email_message.is_multipart():
for part in email_message.walk():
if part.get_content_type() == 'text/plain':
text = part.get_payload(decode=True).decode('utf-8')
else:
text = email_message.get_payload(decode=True).decode('utf-8')

# Закрытие соединения с почтовым сервером
mail.close()
mail.logout()

return text


def run(self):
try:
# это лучше делать снаружи класса и в конструктор передавать уже распаршенные данные
# типа GetCode(IMAP_SERVER, username, password), потому что это не обязанность метода GetCode знать в каком формате у тебя данные - это знает UI
split_string = self.mail_data.split(":")
USERNAME = split_string[0]
PASSWORD = split_string[1]
IMAP_SERVER = 'imap.rambler.ru'
except IndexError:
self.send_code.emit("Неверные данные")
return

try:
email_text = get_last_email_text(IMAP_SERVER, USERNAME, PASSWORD)

# Извлечение кода с помощью регулярного выражения
match = re.search(self.regular_string, email_text)
if match:
# а вот тут ты очень оптимистично думаешь что переданная в конструктор регулярка будет иметь хоть одну группу
# (или что код будет именно в первой группе)
code = match.group(1)
self.send_code.emit(code)
else:
self.send_code.emit("Код не найден.")
except imaplib.IMAP4_SSL.error as e:
self.send_code.emit("Ошибка подключения")
except imaplib.IMAP4.error as e:
self.send_code.emit("Неверные данные")
except Exception as e:
print(e)

Python

import email

import imaplib

import re



from PyQt5.QtCore import QThread, pyqtSignal





class GetCode(QThread):

    send_code = pyqtSignal(str)



    def __init__(self, mail_data, regular_string):

        super().__init__()

        self.mail_data = mail_data

        self.regular_string = regular_string



    def get_last_email_text(imap_server, username, password):

        # Подключение к почтовому серверу

        mail = imaplib.IMAP4_SSL(imap_server)

        mail.login(username, password)



        # Выбор почтового ящика (в данном случае - входящие)

        mail.select('INBOX')



        # Поиск последнего письма

        result, data = mail.search(None, 'ALL')



        # Тут может оказаться что почтовый ящик пустой и в нем нет писем вообще - ты это не проверяешь

        last_email_id = data[0].split()[-1]

        result, data = mail.fetch(last_email_id, '(RFC822)')



        # Извлечение текста письма

        raw_email = data[0][1]

        email_message = email.message_from_bytes(raw_email)

        text = ""



        if email_message.is_multipart():

            for part in email_message.walk():

                if part.get_content_type() == 'text/plain':

                    text = part.get_payload(decode=True).decode('utf-8')

        else:

            text = email_message.get_payload(decode=True).decode('utf-8')



        # Закрытие соединения с почтовым сервером

        mail.close()

        mail.logout()



        return text





    def run(self):

        try:

            # это лучше делать снаружи класса и в конструктор передавать уже распаршенные данные

            # типа GetCode(IMAP_SERVER, username, password), потому что это не обязанность метода GetCode знать в каком формате у тебя данные - это знает UI

            split_string = self.mail_data.split(":")

            USERNAME = split_string[0]

            PASSWORD = split_string[1]

            IMAP_SERVER = 'imap.rambler.ru'

        except IndexError:

            self.send_code.emit("Неверные данные")

            return



        try:

            email_text = get_last_email_text(IMAP_SERVER, USERNAME, PASSWORD)



            # Извлечение кода с помощью регулярного выражения

            match = re.search(self.regular_string, email_text)

            if match:

                # а вот тут ты очень оптимистично думаешь что переданная в конструктор регулярка будет иметь хоть одну группу

                # (или что код будет именно в первой группе)

                code = match.group(1)

                self.send_code.emit(code)

            else:

                self.send_code.emit("Код не найден.")

        except imaplib.IMAP4_SSL.error as e:

            self.send_code.emit("Ошибка подключения")

        except imaplib.IMAP4.error as e:

            self.send_code.emit("Неверные данные")

        except Exception as e:

            print(e)

 

Доработал скрипт на обработку ошибки пустой почты

 

переход на pyside6
реворк UI
реворк бека

 

НикотиновыйКодер , должен ещё рабаотать?