def search_code(code):
# TODO : rewrite this code
dataset = flaten_Dataset('flaten')
results = []
code_rex = re.compile("^%s*" %(code,))
for item in dataset.find({'code': code_rex}):
results.append({key: value
for key, value in item.items() if (key != "_id")
and (key != 'children')})
for result in results:
result['amount'] = int(result['amount'])
return results
def handle(self, *args, **options):
muni = args[0]
year = args[1]
print "bla for the win"
schema_datasset = tree_Dataset("scheme", 0)
tree = Tree.from_dict(schema_datasset.find_one())
tree.update_field("muni", muni)
tree.update_field("year", year)
raw_dataset = raw_Dataset(muni, year)
for line in raw_dataset.find({}):
node = Tree(muni=muni, year=year, **line)
# import pdb; pdb.set_trace()
tree.insert_node(node)
tree.update_amount()
dataset = tree_Dataset(muni, year, clean=True)
dataset = flaten_Dataset("flaten")
# dataset.insert(tree.to_dict())
tree.to_db(dataset)
