1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
# Copyright (c) 2015, Menno Smits
# Released subject to the New BSD License
# Please see http://en.wikipedia.org/wiki/BSD_licenses
import imaplib
import socket
from typing import Optional
class IMAP4WithTimeout(imaplib.IMAP4):
def __init__(self, address: str, port: int, timeout: Optional[float]) -> None:
self._timeout = timeout
imaplib.IMAP4.__init__(self, address, port)
def open(
self, host: str = "", port: int = 143, timeout: Optional[float] = None
) -> None:
# This is overridden to make it consistent across Python versions.
self.host = host
self.port = port
self.sock = self._create_socket(timeout)
self.file = self.sock.makefile("rb")
def _create_socket(self, timeout: Optional[float] = None) -> socket.socket:
return socket.create_connection(
(self.host, self.port), timeout if timeout is not None else self._timeout
)
|
