feat: add --verbose and --no-mail

Author: xchopin

openlrw/client.py

@@ -19,6 +19,7 @@
 import json
 import sys
 import base64
+import argparse
 
 from openlrw.oneroster import OneRoster
 from openlrw.exceptions import *
@@ -44,6 +45,11 @@ class OpenLRW(object):
 
     URI = ""
 
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--verbose', action='store_true', help='Print all the HTTP calls')
+    parser.add_argument('--no-mail', action='store_true', help='Disable the email server')
+    options = parser.parse_args()
+
     def __init__(self, url, username, password):
         """
         Constructor
@@ -90,7 +96,8 @@ def mail_server(self, subject, message):
         :param message:
         :return:
         """
-        if self._host_mail:
+
+        if self._host_mail and not self.options.nomail:
             self._mail = smtplib.SMTP(str(self._host_mail))
             self._mail.sendmail(self._from_mail, self._to_mail, "Subject: " + subject + " \n\n" + message)
 
@@ -214,10 +221,10 @@ def generate_jwt(self):
         """
         headers = {'X-Requested-With': 'XMLHttpRequest'}
         data = {"username": self._username, "password": self._password}
-        print(OpenLRW.URI)
         try:
             response = requests.post(OpenLRW.URI + Routes.AUTH, headers=headers, json=data)
-            Routes.print_post(self.URI + Routes.AUTH, response)
+            if self.options.verbose:
+                Routes.print_post(self.URI + Routes.AUTH, response)
             res = response.json()
             return res['token']
         except requests.RequestException:

openlrw/oneroster.py

@@ -41,7 +41,8 @@ def http_get(route, jwt):
         """
         from openlrw.client import OpenLRW
         response = requests.get(str(OpenLRW.URI + route), headers={'X-Requested-With': 'XMLHttpRequest', 'Authorization': 'Bearer ' + str(jwt)})
-        Routes.print_get(route, response)
+        if OpenLRW.options.verbose:
+            Routes.print_get(route, response)
 
         if response.status_code == 401:
             raise ExpiredTokenException
@@ -61,7 +62,9 @@ def http_post(route, data, jwt):
         """
         from openlrw.client import OpenLRW
         response = requests.post(str(OpenLRW.URI + route), headers={'X-Requested-With': 'XMLHttpRequest', 'Authorization': 'Bearer ' + str(jwt)}, json=data)
-        Routes.print_post(route, response)
+
+        if OpenLRW.options.verbose:
+            Routes.print_post(route, response)
 
         if response.status_code == 401:
             raise ExpiredTokenException
@@ -82,7 +85,10 @@ def http_delete(route, jwt):
         """
         from openlrw.client import OpenLRW
         response = requests.delete(str(OpenLRW.URI + route), headers={'X-Requested-With': 'XMLHttpRequest', 'Authorization': 'Bearer ' + str(jwt)})
-        Routes.print_delete(route, response)
+
+        if OpenLRW.options.verbose:
+            Routes.print_delete(route, response)
+
         if response.status_code == 401:
             raise ExpiredTokenException(response)
         elif response.status_code == 500:
@@ -101,7 +107,9 @@ def http_patch(route, data, jwt):
         """
         from openlrw.client import OpenLRW
         response = requests.patch(str(OpenLRW.URI + route), headers={'X-Requested-With': 'XMLHttpRequest', 'Authorization': 'Bearer ' + str(jwt)}, json=data)
-        Routes.print_patch(route, response)
+        if OpenLRW.options.verbose:
+            Routes.print_patch(route, response)
+
         if response.status_code == 401:
             raise ExpiredTokenException
         elif response.status_code == 400: