|
|
@ -1,3 +1,4 @@
|
|
|
|
|
|
|
|
from pathlib import Path
|
|
|
|
from corpusparser import Parser
|
|
|
|
from corpusparser import Parser
|
|
|
|
import argparse
|
|
|
|
import argparse
|
|
|
|
import logging
|
|
|
|
import logging
|
|
|
@ -8,37 +9,54 @@ logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
## Main handles command line arguments and writing to files / DB.
|
|
|
|
## Main handles command line arguments and writing to files / DB.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def ssj_to_json_file(sentence_generator, outfile):
|
|
|
|
|
|
|
|
# this funciton is based on the fact that files are parsed sequentially
|
|
|
|
|
|
|
|
data_buffer = []
|
|
|
|
|
|
|
|
for s in sentence_generator:
|
|
|
|
|
|
|
|
sdata = s[1]
|
|
|
|
|
|
|
|
data_buffer += [sdata]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# outfile = Path(outfile)
|
|
|
|
|
|
|
|
with open(outfile, "w") as fp:
|
|
|
|
|
|
|
|
logger.info("Writing to {}".format(outfile))
|
|
|
|
|
|
|
|
json.dump(data_buffer, fp)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def kres_to_json_files()
|
|
|
|
|
|
|
|
return "TODO"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def to_db():
|
|
|
|
|
|
|
|
return "TODO"
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if __name__ == "__main__":
|
|
|
|
parser = argparse.ArgumentParser(description="Parsing corpora kres and ssj500k.")
|
|
|
|
parser = argparse.ArgumentParser(description="Parsing corpora kres and ssj500k.")
|
|
|
|
parser.add_argument('--kres-folder', required=True)
|
|
|
|
parser.add_argument('--kres-folder', required=True)
|
|
|
|
parser.add_argument('--kres-srl-folder', required=True)
|
|
|
|
parser.add_argument('--kres-srl-folder', required=True)
|
|
|
|
parser.add_argument('--ssj-file', required=True)
|
|
|
|
parser.add_argument('--ssj-file', required=True)
|
|
|
|
parser.add_argument('--output', required=False, default=None)
|
|
|
|
parser.add_argument('--output', required=False, default=None)
|
|
|
|
parser.add_argument('--outdir', required=False, default=None)
|
|
|
|
parser.add_argument('--outdir', required=False, default=None)
|
|
|
|
parser.add_argument('--dbaddr', required=False, default=None)
|
|
|
|
parser.add_argument('--dbaddr', required=False, default=None)
|
|
|
|
args = parser.parse_args()
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
|
|
|
|
|
|
# parse ssj
|
|
|
|
# parse ssj
|
|
|
|
logger.info("Parsing ssj500k: {}".format(args.ssj_file))
|
|
|
|
logger.info("Parsing ssj500k: {}".format(args.ssj_file))
|
|
|
|
ssj_parser = Parser(
|
|
|
|
ssj_parser = Parser(
|
|
|
|
corpus="ssj",
|
|
|
|
corpus="ssj",
|
|
|
|
infiles=[args.ssj_file],
|
|
|
|
infiles=[args.ssj_file],
|
|
|
|
output=args.output,
|
|
|
|
)
|
|
|
|
outdir=args.outdir,
|
|
|
|
# res = [x[1]["sid"] for x in ssj_parser.sentence_generator()]
|
|
|
|
)
|
|
|
|
# logger.info("Parsed {} sentences (ssj500k)".format(len(res)))
|
|
|
|
res = [x[1]["sid"] for x in ssj_parser.sentence_generator()]
|
|
|
|
|
|
|
|
logger.info("Parsed {} sentences (ssj500k)".format(len(res)))
|
|
|
|
# ssj to json
|
|
|
|
|
|
|
|
ssj_to_json_file(ssj_parser.sentence_generator(), "/home/voje/workdir/ssj.json")
|
|
|
|
# parse kres
|
|
|
|
|
|
|
|
logger.info("Parsing Kres: {}".format(args.ssj_file))
|
|
|
|
# parse kres
|
|
|
|
kres_parser = Parser(
|
|
|
|
logger.info("Parsing Kres: {}".format(args.ssj_file))
|
|
|
|
corpus="kres",
|
|
|
|
kres_parser = Parser(
|
|
|
|
infiles=[args.kres_folder, args.kres_srl_folder],
|
|
|
|
corpus="kres",
|
|
|
|
output=args.output,
|
|
|
|
infiles=[args.kres_folder, args.kres_srl_folder],
|
|
|
|
outdir=args.outdir,
|
|
|
|
)
|
|
|
|
)
|
|
|
|
res = [x[1]["sid"] for x in kres_parser.sentence_generator()]
|
|
|
|
res = [x[1]["sid"] for x in kres_parser.sentence_generator()]
|
|
|
|
logger.info("Parsed {} sentences (kres)".format(len(res)))
|
|
|
|
logger.info("Parsed {} sentences (kres)".format(len(res)))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Handling output is situational --- implement it outside of Parser.
|
|
|
|
## Handling output is situational --- implement it outside of Parser.
|
|
|
@ -47,12 +65,3 @@ if __name__ == "__main__":
|
|
|
|
# 2. parse and save to DB
|
|
|
|
# 2. parse and save to DB
|
|
|
|
|
|
|
|
|
|
|
|
# TODO
|
|
|
|
# TODO
|
|
|
|
def handle_output(self, sent_ent, xml_file):
|
|
|
|
|
|
|
|
if self.output is None:
|
|
|
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
if self.output == "file":
|
|
|
|
|
|
|
|
outfile = Path(self.outdir) / Path(xml_file.name.split(".")[0]).with_suffix(".json")
|
|
|
|
|
|
|
|
with outfile.open("a") as fp:
|
|
|
|
|
|
|
|
print(sent_ent)
|
|
|
|
|
|
|
|
json.dumps(sent_ent, fp)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|