def related(identifier, direction, color, unicode):
_log.info(f"related to resource id='{identifier}'")
t = _cterm if color else _noterm
dmf = DMF()
try:
resource.identifier_str(identifier, allow_prefix=True)
except ValueError as err:
click.echo(f"{err}")
sys.exit(Code.INPUT_VALUE.value)
_log.debug(f"begin: finding root resource {identifier}")
rsrc_list = list(find_by_id(identifier, dmf=dmf))
n = len(rsrc_list)
if n > 1:
click.echo(f"Too many resources matching `{identifier}`")
sys.exit(Code.INPUT_VALUE)
rsrc = rsrc_list[0]
_log.debug(f"end: finding root resource {identifier}")
# get related resources
_log.debug(f"begin: finding related resources for {identifier}")
outgoing = direction == "out"
rr = list(
dmf.find_related(rsrc, meta=["aliases", "type"], outgoing=outgoing))
_log.debug(f"end: finding related resources for {identifier}")
# stop if no relations
if not rr:
_log.warning(f"no resource related to {identifier}")
click.echo(f"No relations for resource `{identifier}`")
sys.exit(0)
_log.info(f"got {len(rr)} related resources")
# debugging
if _log.isEnabledFor(logging.DEBUG):
dbgtree = '\n'.join([' ' + str(x) for x in rr])
_log.debug(f"related resources:\n{dbgtree}")
# extract uuids & determine common UUID prefix length
uuids = [item[2][resource.Resource.ID_FIELD] for item in rr]
pfx = util.uuid_prefix_len(uuids)
# initialize queue with depth=1 items
q = [item for item in rr if item[0] == 1]
# print root resource
print(_related_item(rsrc.id, rsrc.name, rsrc.type, pfx, t, unicode))
# set up printing style
if unicode:
# connector chars
vrt, vrd, relbow, relbow2, rarr = (
'\u2502',
'\u2506',
'\u2514',
'\u251C',
'\u2500\u2500',
)
# relation prefix and arrow
relpre, relarr = (
['\u2500\u25C0\u2500\u2524', '\u2524'][outgoing],
['\u2502', '\u251C\u2500\u25B6'][outgoing],
)
else:
# connector chars
vrt, vrd, relbow, relbow2, rarr = '|', '.', '+', '+', '--'
# relation prefix and arrow
relpre, relarr = ['<-[', '-['][outgoing], [']-', ']->'][outgoing]
# create map of #items at each level, so we can easily
# know when there are more at a given level, for drawing
n_at_level = {0: 0}
for item in rr:
depth = item[0]
if depth in n_at_level:
n_at_level[depth] += 1
else:
n_at_level[depth] = 1
# print tree
while q:
depth, rel, meta = q.pop()
n_at_level[depth] -= 1
indent = ''.join([
f" {t.blue}{vrd if n_at_level[i - 1] else ' '}{t.normal} "
for i in range(1, depth + 1)
])
print(f"{indent} {t.blue}{vrt}{t.normal}")
rstr = f"{t.blue}{relpre}{t.yellow}{rel.predicate}{t.blue}{relarr}{t.normal}"
if meta["aliases"]:
item_name = meta["aliases"][0]
else:
item_name = meta.get("desc", "-")
istr = _related_item(meta[resource.Resource.ID_FIELD], item_name,
meta["type"], pfx, t, unicode)
# determine correct connector (whether there is another one down the stack)
elbow = relbow if (not q or q[-1][0] != depth) else relbow2
print(f"{indent} {t.blue}{elbow}{rarr}{t.normal}{rstr} {istr}")
new_rr = []
for d2, rel2, _ in rr:
if outgoing:
is_same = rel2.subject == rel.object
else:
is_same = rel2.object == rel.subject
if d2 == depth + 1 and is_same:
q.append((d2, rel2, _))
else:
new_rr.append((d2, rel2, _))
rr = new_rr
def _print_resource_table(resources, show_fields, sort_by, reverse, prefix,
color):
"""Text-mode `ls`.
"""
t = _cterm if color else _noterm
if len(resources) == 0:
print("no resources to display")
return
uuid_len = util.uuid_prefix_len([r.id for r in resources])
full_len = max((len(r.id) for r in resources)) if not prefix else uuid_len
fields = ["id"] + list(show_fields)
nfields = len(fields)
# calculate table body. do this first to get widths.
hdr_fields = [".".join(f) if isinstance(f, tuple) else f for f in fields]
rows, maxwid, widths = [], 60, [len(f) for f in hdr_fields]
for r in resources:
row = []
for i, fld in enumerate(fields):
# transform field for display
try:
transformer = _show_fields[fld]
except KeyError:
transformer = _IdentityField(fld)
transformer.set_value(r)
# if it's a UUID field, add info about unique prefix length
is_id_field = isinstance(transformer, _IdField)
if is_id_field:
transformer.term = t
transformer.pfxlen = uuid_len
transformer.flen = full_len
# extract display string and set field width
s = str(transformer)
slen = full_len if is_id_field else len(s)
if slen > widths[i]:
if slen > maxwid:
s, widths[i] = s[:maxwid], maxwid
else:
widths[i] = slen
row.append(s)
# append sort keys (not displayed)
sort_obj = [_show_fields[fld] for fld in sort_by]
for o in sort_obj:
o.set_value(r)
row.extend([o.value for o in sort_obj])
# add row to table body
rows.append(row)
# sort rows
nsort = len(sort_by)
if nsort > 0:
sort_key_idx = tuple(range(nfields, nfields + nsort))
rows.sort(key=itemgetter(*sort_key_idx), reverse=reverse)
# print table header
hdr_columns = [t.bold + f"{f:{w}}" for f, w in zip(hdr_fields, widths)]
print(" ".join(hdr_columns) + t.normal)
# print table body
for row in rows:
col = []
for i, fld in enumerate(fields):
if i == 0:
col.append(t.red)
elif fld == resource.Resource.TYPE_FIELD:
col.append(t.yellow)
elif fld != "desc":
col.append(t.green)
else:
col.append("")
row_columns = [
f"{col[i]}{f:{w}}{t.normal}"
for i, f, w in zip(range(len(widths)), row[:nfields], widths)
]
print(" ".join(row_columns))
