Commit b8c43c01 authored by Sarah Abrishami's avatar Sarah Abrishami

added new apis

parent 4f6f5272
...@@ -20,6 +20,15 @@ def return_obj(oid, what, to_dict=True): ...@@ -20,6 +20,15 @@ def return_obj(oid, what, to_dict=True):
return obj return obj
def return_all_objs(what, to_dict=True):
dct = {'ds': Dataset, 'rg': RuleGroup, 'rule': Rule}
objs = db.session.query(dct[what]).all()
if to_dict:
return [obj.to_dict() for obj in objs]
else:
return objs
def rm_obj(oid, what): def rm_obj(oid, what):
dct = {'ds': Dataset, 'rg': RuleGroup, 'rule': Rule} dct = {'ds': Dataset, 'rg': RuleGroup, 'rule': Rule}
obj = db.session.query(dct[what]).filter(dct[what].uid == oid).one() obj = db.session.query(dct[what]).filter(dct[what].uid == oid).one()
...@@ -28,6 +37,6 @@ def rm_obj(oid, what): ...@@ -28,6 +37,6 @@ def rm_obj(oid, what):
return True return True
def return_project_rules(project_id): def return_project_rules(project_id, flat=True):
rules = db.session.query(Rule).filter(Rule.project_id == project_id).all() rules = db.session.query(Rule).filter(Rule.project_id == project_id).all()
return [r.to_dict() for r in rules] return [r.to_dict(flat=flat) for r in rules]
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment