Fixing syntax errors in db/models.py

This commit is contained in:
Salvatore Orlando 2011-07-05 01:08:18 +01:00
parent c8be977cdc
commit 704ed9a4e1
1 changed files with 2 additions and 21 deletions

View File

@ -19,10 +19,9 @@
import uuid
from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.exc import IntegrityError
from sqlalchemy import Column, String, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relation
from sqlalchemy.orm import relation, object_mapper
BASE = declarative_base()
@ -30,24 +29,6 @@ BASE = declarative_base()
class QuantumBase(object):
"""Base class for Quantum Models."""
def save(self, session=None):
"""Save this object."""
if not session:
session = get_session()
session.add(self)
try:
session.flush()
except IntegrityError, e:
if str(e).endswith('is not unique'):
raise exception.Duplicate(str(e))
else:
raise
def delete(self, session=None):
"""Delete this object."""
# TODO: this method does not do anything at the moment
self.save(session=session)
def __setitem__(self, key, value):
setattr(self, key, value)