safer safe keyword

pombredanne · May 3, 2012 · 674d474 · 674d474
1 parent 9152d83
commit 674d474
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mongoalchemy/session.py b/mongoalchemy/session.py
@@ -69,7 +69,7 @@ def __init__(self, database, safe=False):
         self.safe = safe
 
     @classmethod
-    def connect(self, database, safe=False, *args, **kwds):
+    def connect(self, database, *args, **kwds):
         ''' `connect` is a thin wrapper around __init__ which creates the 
             database connection that the session will use.
             
@@ -80,6 +80,9 @@ def connect(self, database, safe=False, *args, **kwds):
             :param args: arguments for :class:`pymongo.connection.Connection`
             :param kwds: keyword arguments for :class:`pymongo.connection.Connection`
         '''
+        safe = kwds.get('safe', False)
+        if 'safe' in kwds:
+            del kwds['safe']
         conn = Connection(*args, **kwds)
         db = conn[database]
         return Session(db, safe=safe)