1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # mojo, a Python library for implementing document based databases
5 # Copyright (C) 2013-2014 by Javier Sancho Fernandez <jsf at jsancho dot org>
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 ##############################################################################
23 from cursor import Cursor
26 class Collection(object):
27 def __init__(self, database, table_name):
28 self.database = database
29 self.table_name = str(table_name)
32 return "Collection(%r, %r)" % (self.database, self.table_name)
35 return (self.database.exists() and self.table_name in self.database.collection_names())
37 def _create_field(self, field_name):
39 {'name': 'id', 'type': 'char', 'size': 512, 'primary': True},
40 {'name': 'value', 'type': 'text', 'null': False},
41 {'name': 'number', 'type': 'float'},
43 return self.database.connection._create_table(self.database.db_name, '%s$%s' % (self.table_name, field_name), fields)
45 def _get_fields(self):
46 tables = self.database.connection._get_tables(self.database.db_name)
47 return [str(x[x.find('$')+1:]) for x in filter(lambda x: x.startswith('%s$' % self.table_name), tables)]
50 return self.database.connection._count(self.database.db_name, self.table_name)
52 def find(self, *args, **kwargs):
53 return Cursor(self, *args, **kwargs)
55 def insert(self, doc_or_docs):
56 if not self.database.db_name in self.database.connection.database_names():
57 self.database._create_database()
58 if not self.table_name in self.database.collection_names():
59 self._create_field('_id')
61 if not type(doc_or_docs) in (list, tuple):
66 doc_id = uuid.uuid4().hex
69 self._insert_document(doc_id, doc)
71 if type(doc_or_docs) in (list, tuple):
72 return [d['_id'] for d in docs]
76 def _insert_document(self, doc_id, doc):
77 fields = self._get_fields()
83 'value': msgpack.dumps(doc[f]),
85 if type(doc[f]) in (int, float):
86 values['number'] = doc[f]
87 table_f = '%s$%s' % (self.table_name, f)
88 self.database.connection._insert(self.database.db_name, table_f, values)