xevion/the-office
1
0
Fork 0
mirror of https://github.com/Xevion/the-office.git synced 2025-12-08 22:08:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add normalization.meta-update command

Browse Source
This commit is contained in:
Xevion
2022-05-09 11:20:48 -05:00
parent d9f0481cdd
commit 1b3c7fc4e6
1 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
server/normalization/main.py
View File

@@ -472,5 +472,33 @@ def compile() -> None:
logger.info('Completed episode data compiling.')
@cli.command('meta-update')
def meta_update() -> None:
"""Update identifiers.xml with type meta data from meta.json"""
with open(ConstantPaths.META, 'r') as meta_file:
meta_data = json.load(meta_file)
null_count: int = len(list(filter(lambda v: v is None, meta_data.values())))
if null_count > 0:
logger.warning(f"{null_count} characters still have null values.")
with open(ConstantPaths.IDENTIFIERS, 'r') as identifier_file:
identifiers = etree.parse(identifier_file)
for character in identifiers.xpath('//SpeakerList/Speaker/Character'):
character.attrib["type"] = meta_data.get(character.text) or "null"
for character in identifiers.xpath('//SpeakerList/Speaker/Characters/Character'):
character.attrib["type"] = meta_data.get(character.text) or "null"
with open(ConstantPaths.IDENTIFIERS, 'w') as identifier_file:
etree.indent(identifiers, space=" " * 4)
identifier_file.write(etree.tostring(identifiers, encoding=str, pretty_print=True))
if __name__ == '__main__':
cli()