def delete(cls,user_id):
handler = DBSession.query(cls).filter_by(user_id=user_id).first()
if handler == None:
return "Username doesn't exist"
if handler != None:
DBSession.delete(handler)
DBSession.flush()
return "ok"
def setUp(self):
"""Setup test fixture for each model test method."""
try:
new_attrs = {}
new_attrs.update(self.attrs)
new_attrs.update(self.do_get_dependencies())
self.obj = self.klass(**new_attrs)
DBSession.add(self.obj)
DBSession.flush()
return self.obj
except:
DBSession.rollback()
raise
def update(cls,user_id,user_name,email_address,display_name,password):
handler = DBSession.query(cls).filter_by(user_id=user_id).first()
if handler == None:
return "Username doesn't exist"
handler.user_name = user_name
handler.email_address = email_address
handler.display_name = display_name
if password != "":
handler.password = password
print("FLUSHABLE")
DBSession.flush()
return "ok"
def add(cls,user_name,email_address,display_name,password):
try:
handler = cls()
handler.user_name = user_name
handler.email_address = email_address
handler.display_name = display_name
handler.password = password
DBSession.add(handler)
DBSession.flush()
return "ok"
except exc.SQLAlchemyError:
DBSession.rollback()
handler = DBSession.query(cls).filter_by(user_name = user_name).all()
if str(handler) != "[]":
return "Username already used"
handler = DBSession.query(cls).filter_by(email_address=email_address).all()
if str(handler) != "[]":
return "Email already used"
return "not ok"
def updateGrid(self):
if self.kw['oper'] == "edit":
my_filters = {self.indexkey: self.kw['id']}
query = DBSession.query(self.model)
for attr, value in my_filters.items():
query = query.filter(getattr(self.model, attr) == value)
item = query.first()
if item is not None:
for column in item.__table__.columns:
if column.name != self.indexkey:
if column.name in self.kw:
if str(column.type) == "BOOLEAN":
newvalue = True if self.kw[
column.name] == "True" else False
else:
newvalue = self.kw[column.name]
setattr(item, column.name, newvalue)
DBSession.flush()
if self.kw['oper'] == "add":
item = self.model()
for column in item.__table__.columns:
if column.name in self.kw:
if (self.indexkey == column.name):
pass
else:
setattr(item, column.name, self.kw[column.name])
DBSession.add(item)
DBSession.flush()
if self.kw['oper'] == "del":
my_filters = {self.indexkey: self.kw['id']}
query = DBSession.query(self.model)
for attr, value in my_filters.items():
query = query.filter(getattr(self.model, attr) == value)
item = query.first()
if item is not None:
DBSession.delete(item)
DBSession.flush()
return dict(error="")