Use a single parameter for DB connection

Could not get pymongo to work otherwise.

1 files changed, 3 insertions, 5 deletions

diff --git a/kpov_judge/create_disk_images.py b/kpov_judge/create_disk_images.py
index 16bbcab..9073538 100755
--- a/kpov_judge/create_disk_images.py
+++ b/kpov_judge/create_disk_images.py
@@ -61,11 +61,9 @@ if __name__ == '__main__':
     if len(sys.argv) != 1:
         print("Usage: {0}")
         print("Create the pending disk images")
-    db = pymongo.MongoClient(settings.DB_HOST).kpov
-    try:
-        db.authenticate(settings.USERNAME, settings.PASSWORD)
-    except Exception as e:
-        print(str(e))
+
+    db = pymongo.MongoClient(settings.DB_URI).get_default_database()
+
     dev_prefix = settings.GUESTFS_DEV_PREFIX
     l = db.student_computers.find({"disk_urls": {"$exists": False}})
     computers_by_class_task_student = dict()