From 6e23d752f0e3aa2b82c25e572085ad5bf1baffaa Mon Sep 17 00:00:00 2001
From: Timotej Lazar <timotej.lazar@fri.uni-lj.si>
Date: Wed, 10 Oct 2018 21:52:19 +0200
Subject: Use a single parameter for DB connection

Could not get pymongo to work otherwise.
---
 kpov_judge/create_disk_images.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'kpov_judge/create_disk_images.py')

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()
-- 
cgit v1.2.1