From 67e239ebadf7ee03b2bc582235d8b04300378e32 Mon Sep 17 00:00:00 2001
From: spla <spla@mastodont.cat>
Date: Wed, 3 Aug 2022 09:10:20 +0200
Subject: [PATCH] handle of ModuleNotFoundError

---
 edit_status.py | 5 ++++-
 inactives.py   | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/edit_status.py b/edit_status.py
index ed3398f..5c11dc8 100644
--- a/edit_status.py
+++ b/edit_status.py
@@ -8,7 +8,10 @@ import os
 import sys
 import os.path
 import psycopg2
-from prettytable import PrettyTable
+try:
+    from prettytable import PrettyTable
+except ModuleNotFoundError as mod_not_found:
+    print(f"{mod_not_found}. Run 'pip install -r requirements.txt' and try again")
 
 def get_user(email):
 
diff --git a/inactives.py b/inactives.py
index 3486598..ff6c85c 100644
--- a/inactives.py
+++ b/inactives.py
@@ -2,7 +2,10 @@ import os
 import sys
 from datetime import datetime, timezone, timedelta
 import time
-import psycopg2
+try:
+    import psycopg2
+except ModuleNotFoundError as mod_not_found:
+    print(f"{mod_not_found}'. Run 'pip install -r requirements.txt' and try again")
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 import smtplib