diff --git a/modules/database.py b/modules/database.py index ea625de..705d3c1 100644 --- a/modules/database.py +++ b/modules/database.py @@ -24,10 +24,11 @@ db = db_client.get_database(name=db_config["name"]) collections = db.list_collection_names() -for collection in ["albums", "photos", "tokens"]: +for collection in ["users", "albums", "photos", "tokens"]: if not collection in collections: db.create_collection(collection) +col_users = db.get_collection("users") col_albums = db.get_collection("albums") col_photos = db.get_collection("photos") col_tokens = db.get_collection("tokens") \ No newline at end of file