def get(self, label_leaf_id):
dbm = access.DBMan(LOST_CONFIG)
identity = get_jwt_identity()
user = dbm.get_user_by_id(identity)
if not user.has_role(roles.DESIGNER):
dbm.close_session()
return "You are not authorized.", 401
else:
label_tree = LabelTree(dbm, root_id=label_leaf_id)
ldf = label_tree.to_df()
dbm.close_session()
f = BytesIO()
ldf.to_csv(f)
f.seek(0)
resp = make_response(f.read())
resp.headers[
"Content-Disposition"] = f"attachment; filename={label_tree.root.name}.csv"
resp.headers["Content-Type"] = "blob"
return resp
import pandas as pd
logging.basicConfig(level=logging.INFO, format='(%(levelname)s): %(message)s')
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description='Export a label tree to a csv file')
parser.add_argument('tree_name',
nargs='?',
action='store',
help='Name of the label tree.')
parser.add_argument('out_path',
nargs='?',
action='store',
help='Path to store the label tree in csv style.')
args = parser.parse_args()
lostconfig = config.LOSTConfig()
dbm = access.DBMan(lostconfig)
root_leaf = next(
filter(lambda x: x.name == args.tree_name, dbm.get_all_label_trees()),
None)
if root_leaf is None:
logging.warning('LabelTree "{}" not present in database!'.format(
args.tree_name))
else:
tree = LabelTree(dbm, root_leaf=root_leaf)
df = tree.to_df()
df.to_csv(args.out_path, index=False)
logging.info('Exported label tree to: {}'.format(args.out_path))
dbm.close_session()