Просмотр исходного кода

Merge pull request #59 from martino87r/master

FIx import in docs and allow multiple calls for sender.close()
Mikael Koli 3 лет назад
Родитель
Сommit
c5c57b43bf
2 измененных файлов с 6 добавлено и 5 удалено
  1. 2 2
      docs/tutorials/sending.rst
  2. 4 3
      redmail/email/sender.py

+ 2 - 2
docs/tutorials/sending.rst

@@ -13,7 +13,7 @@ is configured. At minimum, sending an email requires:
 
 .. code-block:: python
 
-    from email import EmailSender
+    from redmail import EmailSender
     email = EmailSender(host='localhost', port=0)
 
     email.send(
@@ -166,4 +166,4 @@ methods:
         )
         ...
     finally:
-        email.close()
+        email.close()

+ 4 - 3
redmail/email/sender.py

@@ -440,8 +440,9 @@ class EmailSender:
 
     def close(self):
         "Close (quit) the connection"
-        self.connection.quit()
-        self.connection = None
+        if self.connection:
+            self.connection.quit()
+            self.connection = None
 
     def get_server(self) -> smtplib.SMTP:
         "Connect and get the SMTP Server"
@@ -555,4 +556,4 @@ class EmailSender:
     @user_name.setter
     def user_name(self, user):
         warnings.warn("Attribute user_name was renamed as username. Please use username instead.", FutureWarning)
-        self.username = user
+        self.username = user