def _delete_null_columns(self): """ executes a delete query to remove columns that have changed to null """ ds = DeleteStatement(self.column_family_name) deleted_fields = False for _, v in self.instance._values.items(): col = v.column if v.deleted: ds.add_field(col.db_field_name) deleted_fields = True elif isinstance(col, Map): uc = MapDeleteClause(col.db_field_name, v.value, v.previous_value) if uc.get_context_size() > 0: ds.add_field(uc) deleted_fields = True if deleted_fields: for name, col in self.model._primary_keys.items(): ds.add_where_clause(WhereClause( col.db_field_name, EqualsOperator(), col.to_database(getattr(self.instance, name)) )) self._execute(ds)
def test_update(self): c = MapDeleteClause('s', {3: 0}, {1: 2, 3: 4, 5: 6}) c._analyze() c.set_context_id(0) self.assertEqual(c._removals, [1, 5]) self.assertEqual(c.get_context_size(), 2) self.assertEqual(str(c), '"s"[%(0)s], "s"[%(1)s]') ctx = {} c.update_context(ctx) self.assertEqual(ctx, {'0': 1, '1': 5})
def test_update(self): c = MapDeleteClause('s', {3: 0}, {1: 2, 3: 4, 5: 6}) c._analyze() c.set_context_id(0) self.assertEqual(c._removals, [1, 5]) self.assertEqual(c.get_context_size(), 2) self.assertEqual(str(c), '"s"[%(0)s], "s"[%(1)s]') ctx = dict() c.update_context(ctx) self.assertEqual(ctx, {'0': 1, '1': 5})