From ad2b97eb837baa2d1ed43d2825c9f337488ba809 Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@tutanota.com>
Date: Fri, 14 Jan 2022 15:16:57 +0100
Subject: [PATCH] Add .env support

---
 README.md |  3 +++
 main.py   | 17 +++++++++++++++--
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index e79366e..998b250 100644
--- a/README.md
+++ b/README.md
@@ -3,6 +3,7 @@
 ## Pré-requis
 ```
 requests-html==0.10.0
+python-dotenv==0.19.2
 ```
 
 ## Utilisation
@@ -12,5 +13,7 @@ python3 main.py "<lien-vers-l'instance-CAS-pour-la-connexion-à-uPortal>" "<pseu
 # Exemple URL : https://cas.XXX.xxx/cas/login?service=https://e-p8.XXX.xxx/uPortal/Login
 ```
 
+Vous pouvez aussi utilisez un fichier `.env` (recommandé) avec : `URL`, `LOGIN` et `PASSWORD` comme nom de variables.
+
 ---
 Testé avec l'[instance de Paris 8](https://e-p8.univ-paris8.fr).
diff --git a/main.py b/main.py
index 481e211..4caa3e4 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,7 @@
+from dotenv import load_dotenv
+from os import environ
 from sys import argv
+
 from requests_html import BaseSession
 
 class Universite:
@@ -56,8 +59,18 @@ class Universite:
             return "WIP"
 
 if __name__ == "__main__":
-    argv = argv[1:]
+    nom = argv.pop(0)
     if len(argv) == 3:
         print(Universite(*argv).recuperationNotes())
     else:
-        print("Merci de renseigner l'URL, le pseudo et le mot de passe (avec des \").")
+        load_dotenv()
+        try:
+            print(Universite(environ["URL"], environ["LOGIN"], environ["PASSWORD"]).recuperationNotes())
+        except:
+            print(f"""Merci de renseigner l'URL, le pseudo et le mot de passe (avec des \"). \
+            \n-> python3 {nom} "URL" "pseudo" "mot-de-passe" \
+            \n--- \
+            \nOu fichier .env contenant ses informations avec les noms (conseillé) : \
+            \n-> URL \
+            \n-> LOGIN \
+            \n-> PASSWORD""")