Update shroom.py

Yes-but-No · Mar 13, 2023 · 90ce3ee · 90ce3ee
1 parent 3483537
commit 90ce3ee
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/shroom.py b/shroom.py
@@ -46,8 +46,9 @@ def to_dict(self, include_id=True, include_time=True) -> FarmDict:
 
 
 class ShroomFarm:
-  def __init__(self):
-    self._db_client = motor_asyncio.AsyncIOMotorClient()
+  def __init__(self, url: str = "localhost"):
+    self.db_url = url
+    self._db_client = motor_asyncio.AsyncIOMotorClient(url)
     self.shroom_db: motor_asyncio.AsyncIOMotorDatabase = self._db_client["ShroomDB"]
 
     self.farm_db: motor_asyncio.AsyncIOMotorCollection = self.shroom_db["Farm"]