Add dict style SystemJS item access PYTHON-190

Author: behackett

pymongo/database.py

@@ -705,13 +705,22 @@ def __init__(self, database):
  def __setattr__(self, name, code):
  self._db.system.js.save({"_id": name, "value": Code(code)}, safe=True)
 
+ def __setitem__(self, name, code):
+ self.__setattr__(name, code)
+
  def __delattr__(self, name):
  self._db.system.js.remove({"_id": name}, safe=True)
 
+ def __delitem__(self, name):
+ self.__delattr__(name)
+
  def __getattr__(self, name):
  return lambda *args: self._db.eval("function() { return %s.apply(this,"
  "arguments); }" % name, *args)
 
+ def __getitem__(self, name):
+ return self.__getattr__(name)
+
  def list(self):
  """Get a list of the names of the functions stored in this database.
 

test/test_database.py

@@ -494,9 +494,16 @@ def test_system_js(self):
  self.assertEqual('add', db.system.js.find_one()['_id'])
  self.assertEqual(1, db.system.js.count())
  self.assertEqual(6, db.system_js.add(1, 5))
-
  del db.system_js.add
  self.assertEqual(0, db.system.js.count())
+ 
+ db.system_js['add'] = "function(a, b) { return a + b; }"
+ self.assertEqual('add', db.system.js.find_one()['_id'])
+ self.assertEqual(1, db.system.js.count())
+ self.assertEqual(6, db.system_js['add'](1, 5))
+ del db.system_js['add']
+ self.assertEqual(0, db.system.js.count())
+ 
  if version.at_least(db.connection, (1, 3, 2, -1)):
  self.assertRaises(OperationFailure, db.system_js.add, 1, 5)