def handle(self, *args, **kwargs): print(args) database = Database() pkl_file = open(args[0], 'rb') instances = pickle.load(pkl_file) collection = database.get_collection(args[1]) for instance in instances: collection.insert(instance) print("Added", instance['_id']) pkl_file.close()
def handle(self, *args, **kwargs): print(args) database = Database() pkl_file = open(args[0], 'rb') instances = pickle.load(pkl_file) collection = database.get_collection(args[1]) for instance in instances: collection.insert(instance) print("Added", instance['_id']) pkl_file.close()
def setUp(self): # pylint: disable=C0103 """Open a database connection.""" super(TestEmbedModificationValidationB, self).setUp() # Create a new test model instance_loader = InstanceLoader(database='test', validation=False) instance_loader.add_instances(six.itervalues(EMBEDDED_MODELS_B)) # Kill any test existing instances database = Database(database_name='test') self.collection = database.get_collection('article') self.collection.remove() # Add the test article self.collection.insert(TEST_ARTICLE_B)
def setUp(self): # pylint: disable=C0103 """Open a database connection.""" super(TestEmbedModificationValidationB, self).setUp() # Create a new test model instance_loader = InstanceLoader( database='test', validation=False) instance_loader.add_instances(six.itervalues(EMBEDDED_MODELS_B)) # Kill any test existing instances database = Database(database_name='test') self.collection = database.get_collection('article') self.collection.remove() # Add the test article self.collection.insert(TEST_ARTICLE_B)
def setUp(self): # pylint: disable=C0103 """Open a database connection and load the models.""" super(MagEmbedTestCase, self).setUp() # Create test models EMBEDDED_MODELS['article']['_permissions'] = { 'create': True, 'read': True, 'update': True, 'delete': True, } instance_loader = InstanceLoader(database='test', validation=False) instance_loader.add_instances(tuple(six.itervalues(EMBEDDED_MODELS))) # Kill any test existing instances database = Database(database_name='test') collection = database.get_collection('article') collection.remove()
def setUp(self): # pylint: disable=C0103 """Open a database connection and load the models.""" super(MagEmbedTestCase, self).setUp() # Create test models EMBEDDED_MODELS['article']['_permissions'] = { 'create': True, 'read': True, 'update': True, 'delete': True, } instance_loader = InstanceLoader( database='test', validation=False) instance_loader.add_instances(tuple(six.itervalues(EMBEDDED_MODELS))) # Kill any test existing instances database = Database(database_name='test') collection = database.get_collection('article') collection.remove()