mongodb.py
62 lines
| 2.1 KiB
| text/x-python
|
PythonLexer
MinRK
|
r3579 | """A TaskRecord backend using mongodb""" | ||
#----------------------------------------------------------------------------- | ||||
# Copyright (C) 2010 The IPython Development Team | ||||
# | ||||
# Distributed under the terms of the BSD License. The full license is in | ||||
# the file COPYING, distributed as part of this software. | ||||
#----------------------------------------------------------------------------- | ||||
from datetime import datetime | ||||
from pymongo import Connection | ||||
MinRK
|
r3604 | from dictdb import BaseDB | ||
MinRK
|
r3631 | |||
#----------------------------------------------------------------------------- | ||||
MinRK
|
r3579 | # MongoDB class | ||
MinRK
|
r3631 | #----------------------------------------------------------------------------- | ||
MinRK
|
r3604 | class MongoDB(BaseDB): | ||
MinRK
|
r3579 | """MongoDB TaskRecord backend.""" | ||
def __init__(self, session_uuid, *args, **kwargs): | ||||
self._connection = Connection(*args, **kwargs) | ||||
self._db = self._connection[session_uuid] | ||||
self._records = self._db['task_records'] | ||||
self._table = {} | ||||
def add_record(self, msg_id, rec): | ||||
"""Add a new Task Record, by msg_id.""" | ||||
# print rec | ||||
obj_id = self._records.insert(rec) | ||||
self._table[msg_id] = obj_id | ||||
def get_record(self, msg_id): | ||||
"""Get a specific Task Record, by msg_id.""" | ||||
return self._records.find_one(self._table[msg_id]) | ||||
def update_record(self, msg_id, rec): | ||||
"""Update the data in an existing record.""" | ||||
obj_id = self._table[msg_id] | ||||
MinRK
|
r3598 | self._records.update({'_id':obj_id}, {'$set': rec}) | ||
MinRK
|
r3579 | |||
def drop_matching_records(self, check): | ||||
"""Remove a record from the DB.""" | ||||
self._records.remove(check) | ||||
def drop_record(self, msg_id): | ||||
"""Remove a record from the DB.""" | ||||
obj_id = self._table.pop(msg_id) | ||||
self._records.remove(obj_id) | ||||
def find_records(self, check, id_only=False): | ||||
"""Find records matching a query dict.""" | ||||
matches = list(self._records.find(check)) | ||||
if id_only: | ||||
MinRK
|
r3598 | return [ rec['msg_id'] for rec in matches ] | ||
else: | ||||
data = {} | ||||
for rec in matches: | ||||
data[rec['msg_id']] = rec | ||||
return data | ||||
MinRK
|
r3579 | |||