Skip to content

Commit

Permalink
Minor fixes, 'remove' method for dict.
Browse files Browse the repository at this point in the history
  • Loading branch information
wehriam committed Dec 22, 2010
1 parent b13a1e9 commit fd5bcf0
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 5 deletions.
16 changes: 12 additions & 4 deletions cassorm/dict.py
Expand Up @@ -75,14 +75,20 @@ def __getitem__(self, key):
raise CassandraKeyError(key)

def __setitem__(self, key, value):
return self.cf.insert(self.row_key, {key:value})
self.cf.insert(self.row_key, {key:value})

def __delitem__(self, key):
return self.cf.remove(self.row_key, columns=[key])
self.cf.remove(self.row_key, columns=[key])

def update(self, d):
return self.cf.insert(self.row_key, d)

def remove(self, seq):
if isinstance(seq, list):
self.cf.remove(self.row_key, columns=seq)
elif isinstance(seq, dict):
self.cf.remove(self.row_key, columns=seq.keys())

def keys(self):
try:
return self.cf.get(self.row_key).keys()
Expand Down Expand Up @@ -118,6 +124,8 @@ def __contains__(self, key):

def __len__(self):
return self.cf.get_count(self.row_key)

def __dict__(self):
return dict(self.items())

def __del__(self):
self.cf.remove(self.row_key)

5 changes: 4 additions & 1 deletion cassorm/model.py
Expand Up @@ -104,7 +104,10 @@ def __init__(self, cls, create=False):
self.cls.__name__.lower()))
except:
pass


def all(self):
return self.get_range()

def __getitem__(self, key):
return self.get(key)

Expand Down

0 comments on commit fd5bcf0

Please sign in to comment.