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