[DataSource] added the utils.object_collection_name method to utils module. Added also the insert and insert_multi methods to the MongoDb datasource class

lodel/datasource/mongodb/datasource.py

@@ -2,18 +2,10 @@
 import pymongo
 from pymongo import MongoClient
 from pymongo.errors import BulkWriteError
-
 import urllib
 
-# TODO Positionner cette variable dans les settings
-DEFAULT_CONNECTION = {
-    'host': 'localhost',
-    'port': 27017,
-    'login': 'login',  # TODO modifier la valeur
-    'password': 'password',  # TODO modifier la valeur
-    'dbname': 'lodel'
-}
-
+import lodel.datasource.mongodb.utils as utils
+from lodel.settings.settings import Settings
 
 class MongoDbDataSourceError(Exception):
     pass
@@ -21,9 +13,12 @@ class MongoDbDataSourceError(Exception):
 
 class MongoDbDataSource(object):
 
-    MANDATORY_CONNECTION_ARGS = ('host', 'post', 'login', 'password', 'dbname')
+    MANDATORY_CONNECTION_ARGS = ('host', 'port', 'login', 'password', 'dbname')
 
-    def __init__(self, module=pymongo, connection_args=DEFAULT_CONNECTION):
+    ## @brief Instanciates a Database object given a connection name
+    # @param connection_name str
+    def __init__(self, connection_name='default'):
+        connection_args = self._get_connection_args(connection_name)
         login, password, host, port, dbname = MongoDbDataSource._check_connection_args(connection_args)
 
         # Creating of the connection
@@ -32,6 +27,19 @@ class MongoDbDataSource(object):
         # Getting the database
         self.database = self.connection[dbname]
 
+    ## @brief Gets the settings given a connection name
+    # @param connection_name str
+    # @return dict
+    # @TODO Change the return value using the Lodel 2 settings module
+    def _get_connection_args(self, connection_name):
+        return {
+            'host': 'localhost',
+            'port': 27017,
+            'login': 'login',  # TODO modifier la valeur
+            'password': 'password',  # TODO modifier la valeur
+            'dbname': 'lodel'
+        }
+
     ## @brief checks if the connection args are valid and complete
     # @param connection_args dict
     # @return bool
@@ -48,7 +56,8 @@ class MongoDbDataSource(object):
         return (connection_args['login'], urllib.quote_plus(connection_args['password']), connection_args['host'],
                 connection_args['port'], connection_args['dbname'])
 
-    def select(self, target_cls, field_list, filters, rel_filters=None, order=None, group=None, limit=None, offset=0, instanciate=True):
+    def select(self, target_cls, field_list, filters, rel_filters=None, order=None, group=None, limit=None, offset=0,
+               instanciate=True):
         pass
 
     def delete(self, target_cls, uid):
@@ -57,8 +66,29 @@ class MongoDbDataSource(object):
     def update(self, target_cls, uid, **datas):
         pass
 
+    ## @brief Inserts a record in a given collection
+    # @param target_cls Emclass : class of the object to insert
+    # @param datas dict : datas to insert
+    # @return ObjectId : the uid of the inserted record
     def insert(self, target_cls, **datas):
-        pass
-
+        collection_name = utils.object_collection_name(target_cls.__class__)
+        collection = self.database[collection_name]
+        result = collection.insert_one(datas)
+        return result.inserted_id
+
+    ## @brief Inserts a list of records in a given collection
+    # @param target_cls Emclass : class of the objects inserted
+    # @param datas_list
+    # @return int : Number of inserted records in the collection
     def insert_multi(self, target_cls, datas_list):
-        pass
+        collection_name = utils.object_collection_name(target_cls.__class__)
+        collection = self.database[collection_name]
+        bulk = collection.initialize_ordered_bulk_op()
+        for datas_list_item in datas_list:
+            bulk.insert(datas_list_item)
+        try:
+            result = bulk.execute()
+        except BulkWriteError as bwe:
+            pass  # TODO add the behavior in case of an exception => bwe.details['writeErrors'], is a list of error info dicts
+
+        return result['nInserted']

lodel/datasource/mongodb/utils.py

@@ -1 +1,13 @@
-# -*- coding: utf-8 -*-
+# -*- coding: utf-8 -*-
+
+collection_prefix = {
+    'relation': 'rel_',
+    'collection': 'class_'
+}
+
+
+## @brief Returns a collection name given a Emclass name
+# @param class_name str : The class name
+# @return str
+def object_collection_name(class_name):
+    return ("%s%s" % (collection_prefix['object'], class_name)).lower()