timechecker.py

#!/usr/bin/env python3
from sys import argv
from os import path

if __name__ != '__main__':
	print('Cannot use timechecker as a module')
	exit(1)

if len(argv) < 2:
	print('!> input file path must be given')
	exit(1)

if len(argv) > 3:
	print('!> too many arguments')
	exit(1)

INPUT_FILE = argv[1]
OUTPUT_FILE = f'{path.splitext(INPUT_FILE)[0]}.xml'

print('Reading input data...')
input_text = ''
with open(INPUT_FILE, 'r') as f:
	input_text = f.read()

from regexs.redefs import timex
import xml.etree.ElementTree as et

def indent(el: et.Element, lvl = 0):
	TAB = '\t'
	INDENT = '\n' + lvl * TAB
	if len(el):
		if not el.text or not el.text.strip():
			el.text = INDENT + TAB
		if not el.tail or not el.tail.strip():
			el.tail = INDENT
		for el in el:
			indent(el, lvl + 1)
		if not el.tail or not el.tail.strip():
			el.tail = INDENT
	else:
		if lvl and (not el.tail or not el.tail.strip()):
			el.tail = INDENT

print('Generating output...')
TimeML = et.Element('TimeML')
TEXT = et.Element('TEXT')
TimeML.append(TEXT)
TEXT.text = timex(input_text)[0]

f = open(OUTPUT_FILE, 'w+')
f.write('<?xml version="1.0" encoding="UTF-8"?>\n')
f.write('<!DOCTYPE TimeML>\n')
indent(TimeML)
content = et.tostring(TimeML, encoding = 'UTF-8').decode('utf-8').replace(" /", "/").strip()
content = content.replace('&lt;', '<').replace('&gt;', '>')
f.write(content)
f.close()

print('Done.')
exit(0)