projects/Store_emails_in_csv/store_emails.py: PEP8 up

tripleee · tripleee · commit b6b40134fe9a · 2020-10-26T20:57:45.000+02:00
Also, create `def main` so the toplevel can be imported by other
projects if they want to
diff --git a/projects/Store_emails_in_csv/store_emails.py b/projects/Store_emails_in_csv/store_emails.py
@@ -1,13 +1,15 @@
 #!/usr/bin/env python
 
-import imaplib
+import csv
 import email
 from email import policy
-import csv
-import ssl
+import imaplib
 import os
+import ssl
+
 from bs4 import BeautifulSoup
 
+
 credential_path = os.getcwd() + "/credentials.txt"
 csv_path = os.getcwd() + "/mails.csv"
 
@@ -88,8 +90,7 @@ def write_to_csv(mail, writer):
             writer.writerow(row)
 
 
-if __name__ == "__main__":
-
+def main():
     mail, messages = connect_to_mailbox()
 
     total_no_of_mails = int(messages[0])
@@ -104,3 +105,7 @@ def write_to_csv(mail, writer):
             write_to_csv(mail, writer)
         except Exception as e:
             print(e)
+
+
+if __name__ == "__main__":
+    main()
