From 777c9a0c70f3554b9b489d9f5b1d1eb34eae9244 Mon Sep 17 00:00:00 2001
From: Scoder12 <34356756+Scoder12@users.noreply.github.com>
Date: Mon, 27 Jul 2020 14:46:04 -0700
Subject: [PATCH] Don't crash when imported but DB url is not present

---
 src/replit/database/__init__.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/replit/database/__init__.py b/src/replit/database/__init__.py
index cabc78d..c16fac3 100644
--- a/src/replit/database/__init__.py
+++ b/src/replit/database/__init__.py
@@ -276,4 +276,12 @@ class ReplitDb(dict):
         return f"<ReplitDb(db_url={self.db_url!r})>"
 
 
-db = ReplitDb(os.environ["REPLIT_DB_URL"])
+db_url = os.environ.get("REPLIT_DB_URL")
+if db_url:
+    db = ReplitDb(db_url)
+else:
+    print(
+        "Warning: REPLIT_DB_URL does not exist, are we running on repl.it? Database will not function.",
+        file=stderr,
+    )
+    db = None