427 lines
14 KiB
Python
427 lines
14 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import re
|
|
import math
|
|
import platform
|
|
import xml.etree.ElementTree as ET
|
|
|
|
################################################################################
|
|
# Config #
|
|
################################################################################
|
|
|
|
flags = {
|
|
'c': platform.platform() != 'Windows', # Disable by default on windows, since we use ANSI escape codes
|
|
'b': False,
|
|
'g': False,
|
|
's': False,
|
|
'u': False,
|
|
'h': False,
|
|
'p': False,
|
|
'o': True,
|
|
'i': False,
|
|
}
|
|
flag_descriptions = {
|
|
'c': 'Toggle colors when outputting.',
|
|
'b': 'Toggle showing only not fully described classes.',
|
|
'g': 'Toggle showing only completed classes.',
|
|
's': 'Toggle showing comments about the status.',
|
|
'u': 'Toggle URLs to docs.',
|
|
'h': 'Show help and exit.',
|
|
'p': 'Toggle showing percentage as well as counts.',
|
|
'o': 'Toggle overall column.',
|
|
'i': 'Toggle collapse of class items columns.',
|
|
}
|
|
long_flags = {
|
|
'colors': 'c',
|
|
'use-colors': 'c',
|
|
|
|
'bad': 'b',
|
|
'only-bad': 'b',
|
|
|
|
'good': 'g',
|
|
'only-good': 'g',
|
|
|
|
'comments': 's',
|
|
'status': 's',
|
|
|
|
'urls': 'u',
|
|
'gen-url': 'u',
|
|
|
|
'help': 'h',
|
|
|
|
'percent': 'p',
|
|
'use-percentages': 'p',
|
|
|
|
'overall': 'o',
|
|
'use-overall': 'o',
|
|
|
|
'items': 'i',
|
|
'collapse': 'i',
|
|
}
|
|
table_columns = ['name', 'brief_description', 'description', 'methods', 'constants', 'members', 'signals']
|
|
table_column_names = ['Name', 'Brief Desc.', 'Desc.', 'Methods', 'Constants', 'Members', 'Signals']
|
|
colors = {
|
|
'name': [36], # cyan
|
|
'part_big_problem': [4, 31], # underline, red
|
|
'part_problem': [31], # red
|
|
'part_mostly_good': [33], # yellow
|
|
'part_good': [32], # green
|
|
'url': [4, 34], # underline, blue
|
|
'section': [1, 4], # bold, underline
|
|
'state_off': [36], # cyan
|
|
'state_on': [1, 35], # bold, magenta/plum
|
|
}
|
|
overall_progress_description_weigth = 10
|
|
|
|
|
|
################################################################################
|
|
# Utils #
|
|
################################################################################
|
|
|
|
def validate_tag(elem, tag):
|
|
if elem.tag != tag:
|
|
print('Tag mismatch, expected "' + tag + '", got ' + elem.tag)
|
|
sys.exit(255)
|
|
|
|
|
|
def color(color, string):
|
|
if flags['c']:
|
|
color_format = ''
|
|
for code in colors[color]:
|
|
color_format += '\033[' + str(code) + 'm'
|
|
return color_format + string + '\033[0m'
|
|
else:
|
|
return string
|
|
|
|
ansi_escape = re.compile(r'\x1b[^m]*m')
|
|
|
|
|
|
def nonescape_len(s):
|
|
return len(ansi_escape.sub('', s))
|
|
|
|
|
|
################################################################################
|
|
# Classes #
|
|
################################################################################
|
|
|
|
class ClassStatusProgress:
|
|
|
|
def __init__(self, described=0, total=0):
|
|
self.described = described
|
|
self.total = total
|
|
|
|
def __add__(self, other):
|
|
return ClassStatusProgress(self.described + other.described, self.total + other.total)
|
|
|
|
def increment(self, described):
|
|
if described:
|
|
self.described += 1
|
|
self.total += 1
|
|
|
|
def is_ok(self):
|
|
return self.described >= self.total
|
|
|
|
def to_configured_colored_string(self):
|
|
if flags['p']:
|
|
return self.to_colored_string('{percent}% ({has}/{total})', '{pad_percent}{pad_described}{s}{pad_total}')
|
|
else:
|
|
return self.to_colored_string()
|
|
|
|
def to_colored_string(self, format='{has}/{total}', pad_format='{pad_described}{s}{pad_total}'):
|
|
ratio = self.described / self.total if self.total != 0 else 1
|
|
percent = round(100 * ratio)
|
|
s = format.format(has=str(self.described), total=str(self.total), percent=str(percent))
|
|
if self.described >= self.total:
|
|
s = color('part_good', s)
|
|
elif self.described >= self.total / 4 * 3:
|
|
s = color('part_mostly_good', s)
|
|
elif self.described > 0:
|
|
s = color('part_problem', s)
|
|
else:
|
|
s = color('part_big_problem', s)
|
|
pad_size = max(len(str(self.described)), len(str(self.total)))
|
|
pad_described = ''.ljust(pad_size - len(str(self.described)))
|
|
pad_percent = ''.ljust(3 - len(str(percent)))
|
|
pad_total = ''.ljust(pad_size - len(str(self.total)))
|
|
return pad_format.format(pad_described=pad_described, pad_total=pad_total, pad_percent=pad_percent, s=s)
|
|
|
|
|
|
class ClassStatus:
|
|
|
|
def __init__(self, name=''):
|
|
self.name = name
|
|
self.has_brief_description = True
|
|
self.has_description = True
|
|
self.progresses = {
|
|
'methods': ClassStatusProgress(),
|
|
'constants': ClassStatusProgress(),
|
|
'members': ClassStatusProgress(),
|
|
'signals': ClassStatusProgress()
|
|
}
|
|
|
|
def __add__(self, other):
|
|
new_status = ClassStatus()
|
|
new_status.name = self.name
|
|
new_status.has_brief_description = self.has_brief_description and other.has_brief_description
|
|
new_status.has_description = self.has_description and other.has_description
|
|
for k in self.progresses:
|
|
new_status.progresses[k] = self.progresses[k] + other.progresses[k]
|
|
return new_status
|
|
|
|
def is_ok(self):
|
|
ok = True
|
|
ok = ok and self.has_brief_description
|
|
ok = ok and self.has_description
|
|
for k in self.progresses:
|
|
ok = ok and self.progresses[k].is_ok()
|
|
return ok
|
|
|
|
def make_output(self):
|
|
output = {}
|
|
output['name'] = color('name', self.name)
|
|
|
|
ok_string = color('part_good', 'OK')
|
|
missing_string = color('part_big_problem', 'MISSING')
|
|
|
|
output['brief_description'] = ok_string if self.has_brief_description else missing_string
|
|
output['description'] = ok_string if self.has_description else missing_string
|
|
|
|
description_progress = ClassStatusProgress(
|
|
(self.has_brief_description + self.has_description) * overall_progress_description_weigth,
|
|
2 * overall_progress_description_weigth
|
|
)
|
|
items_progress = ClassStatusProgress()
|
|
|
|
for k in ['methods', 'constants', 'members', 'signals']:
|
|
items_progress += self.progresses[k]
|
|
output[k] = self.progresses[k].to_configured_colored_string()
|
|
|
|
output['items'] = items_progress.to_configured_colored_string()
|
|
|
|
output['overall'] = (description_progress + items_progress).to_colored_string('{percent}%', '{pad_percent}{s}')
|
|
|
|
if self.name.startswith('Total'):
|
|
output['url'] = color('url', 'http://docs.godotengine.org/en/latest/classes/')
|
|
if flags['s']:
|
|
output['comment'] = color('part_good', 'ALL OK')
|
|
else:
|
|
output['url'] = color('url', 'http://docs.godotengine.org/en/latest/classes/class_{name}.html'.format(name=self.name.lower()))
|
|
|
|
if flags['s'] and not flags['g'] and self.is_ok():
|
|
output['comment'] = color('part_good', 'ALL OK')
|
|
|
|
return output
|
|
|
|
def generate_for_class(c):
|
|
status = ClassStatus()
|
|
status.name = c.attrib['name']
|
|
for tag in list(c):
|
|
|
|
if tag.tag == 'brief_description':
|
|
status.has_brief_description = len(tag.text.strip()) > 0
|
|
|
|
elif tag.tag == 'description':
|
|
status.has_description = len(tag.text.strip()) > 0
|
|
|
|
elif tag.tag in ['methods', 'signals']:
|
|
for sub_tag in list(tag):
|
|
descr = sub_tag.find('description')
|
|
status.progresses[tag.tag].increment(len(descr.text.strip()) > 0)
|
|
|
|
elif tag.tag in ['constants', 'members']:
|
|
for sub_tag in list(tag):
|
|
status.progresses[tag.tag].increment(len(sub_tag.text.strip()) > 0)
|
|
|
|
elif tag.tag in ['theme_items']:
|
|
pass # Ignore those tags, since they seem to lack description at all
|
|
|
|
else:
|
|
print(tag.tag, tag.attrib)
|
|
|
|
return status
|
|
|
|
|
|
################################################################################
|
|
# Arguments #
|
|
################################################################################
|
|
|
|
input_file_list = []
|
|
input_class_list = []
|
|
|
|
for arg in sys.argv[1:]:
|
|
if arg.startswith('--'):
|
|
flags[long_flags[arg[2:]]] = not flags[long_flags[arg[2:]]]
|
|
elif arg.startswith('-'):
|
|
for f in arg[1:]:
|
|
flags[f] = not flags[f]
|
|
elif arg.endswith('.xml'):
|
|
input_file_list.append(arg)
|
|
else:
|
|
input_class_list.append(arg)
|
|
|
|
if flags['i']:
|
|
for r in ['methods', 'constants', 'members', 'signals']:
|
|
index = table_columns.index(r)
|
|
del table_column_names[index]
|
|
del table_columns[index]
|
|
table_column_names.append('Items')
|
|
table_columns.append('items')
|
|
|
|
if flags['o'] == (not flags['i']):
|
|
table_column_names.append('Overall')
|
|
table_columns.append('overall')
|
|
|
|
if flags['u']:
|
|
table_column_names.append('Docs URL')
|
|
table_columns.append('url')
|
|
|
|
|
|
################################################################################
|
|
# Help #
|
|
################################################################################
|
|
|
|
if len(input_file_list) < 1 or flags['h']:
|
|
if not flags['h']:
|
|
print(color('section', 'Invalid usage') + ': At least one classes.xml file is required')
|
|
print(color('section', 'Usage') + ': doc_status.py [flags] <classes.xml> [class names]')
|
|
print('\t< and > signify required parameters, while [ and ] signify optional parameters.')
|
|
print('\tNote that you can give more than one classes file, in which case they will be merged on-the-fly.')
|
|
print(color('section', 'Available flags') + ':')
|
|
possible_synonym_list = list(long_flags)
|
|
possible_synonym_list.sort()
|
|
flag_list = list(flags)
|
|
flag_list.sort()
|
|
for flag in flag_list:
|
|
synonyms = [color('name', '-' + flag)]
|
|
for synonym in possible_synonym_list:
|
|
if long_flags[synonym] == flag:
|
|
synonyms.append(color('name', '--' + synonym))
|
|
|
|
print(('{synonyms} (Currently ' + color('state_' + ('on' if flags[flag] else 'off'), '{value}') + ')\n\t{description}').format(
|
|
synonyms=', '.join(synonyms),
|
|
value=('on' if flags[flag] else 'off'),
|
|
description=flag_descriptions[flag]
|
|
))
|
|
sys.exit(0)
|
|
|
|
|
|
################################################################################
|
|
# Parse class list #
|
|
################################################################################
|
|
|
|
class_names = []
|
|
classes = {}
|
|
|
|
for file in input_file_list:
|
|
tree = ET.parse(file)
|
|
doc = tree.getroot()
|
|
|
|
if 'version' not in doc.attrib:
|
|
print('Version missing from "doc"')
|
|
sys.exit(255)
|
|
|
|
version = doc.attrib['version']
|
|
|
|
for c in list(doc):
|
|
if c.attrib['name'] in class_names:
|
|
continue
|
|
class_names.append(c.attrib['name'])
|
|
classes[c.attrib['name']] = c
|
|
|
|
class_names.sort()
|
|
|
|
if len(input_class_list) < 1:
|
|
input_class_list = class_names
|
|
|
|
|
|
################################################################################
|
|
# Make output table #
|
|
################################################################################
|
|
|
|
table = [table_column_names]
|
|
table_row_chars = '+- '
|
|
table_column_chars = '|'
|
|
|
|
total_status = ClassStatus('Total')
|
|
|
|
for cn in input_class_list:
|
|
if not cn in classes:
|
|
print('Cannot find class ' + cn + '!')
|
|
sys.exit(255)
|
|
|
|
c = classes[cn]
|
|
validate_tag(c, 'class')
|
|
status = ClassStatus.generate_for_class(c)
|
|
|
|
if flags['b'] and status.is_ok():
|
|
continue
|
|
if flags['g'] and not status.is_ok():
|
|
continue
|
|
|
|
total_status = total_status + status
|
|
out = status.make_output()
|
|
row = []
|
|
for column in table_columns:
|
|
if column in out:
|
|
row.append(out[column])
|
|
else:
|
|
row.append('')
|
|
|
|
if 'comment' in out and out['comment'] != '':
|
|
row.append(out['comment'])
|
|
|
|
table.append(row)
|
|
|
|
|
|
################################################################################
|
|
# Print output table #
|
|
################################################################################
|
|
|
|
if len(table) == 1:
|
|
print(color('part_big_problem', 'No classes suitable for printing!'))
|
|
sys.exit(0)
|
|
|
|
if len(table) > 2:
|
|
total_status.name = 'Total = {0}'.format(len(table) - 1)
|
|
out = total_status.make_output()
|
|
row = []
|
|
for column in table_columns:
|
|
if column in out:
|
|
row.append(out[column])
|
|
else:
|
|
row.append('')
|
|
table.append(row)
|
|
|
|
table_column_sizes = []
|
|
for row in table:
|
|
for cell_i, cell in enumerate(row):
|
|
if cell_i >= len(table_column_sizes):
|
|
table_column_sizes.append(0)
|
|
|
|
table_column_sizes[cell_i] = max(nonescape_len(cell), table_column_sizes[cell_i])
|
|
|
|
divider_string = table_row_chars[0]
|
|
for cell_i in range(len(table[0])):
|
|
divider_string += table_row_chars[1] * (table_column_sizes[cell_i] + 2) + table_row_chars[0]
|
|
print(divider_string)
|
|
|
|
for row_i, row in enumerate(table):
|
|
row_string = table_column_chars
|
|
for cell_i, cell in enumerate(row):
|
|
padding_needed = table_column_sizes[cell_i] - nonescape_len(cell) + 2
|
|
if cell_i == 0:
|
|
row_string += table_row_chars[2] + cell + table_row_chars[2] * (padding_needed - 1)
|
|
else:
|
|
row_string += table_row_chars[2] * math.floor(padding_needed / 2) + cell + table_row_chars[2] * math.ceil((padding_needed / 2))
|
|
row_string += table_column_chars
|
|
|
|
print(row_string)
|
|
|
|
if row_i == 0 or row_i == len(table) - 2:
|
|
print(divider_string)
|
|
|
|
print(divider_string)
|
|
|
|
if total_status.is_ok() and not flags['g']:
|
|
print('All listed classes are ' + color('part_good', 'OK') + '!')
|