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)