some updates

.gitignore

@@ -1 +1,2 @@
 *.egg-info/
+*/__pycache__/

corpusparser/main.py

@@ -3,6 +3,8 @@ from corpusparser import Parser
 import argparse
 import logging
 import json
+from pymongo import MongoClient
+import sys
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -54,11 +56,23 @@ def kres_to_json_files(sentence_generator, outfolder):
         outfile_buffer += [sdata]
     write_buffer_to_file(current_outfile, outfile_buffer)
 
+def data_to_valdb(sentence_generator, dbaddr, username, password, collection_name):
+    logger.info("Connecting to: {}".format(dbaddr))
+    client = MongoClient(
+        "mongodb://{}".format(dbaddr),
+        username=username,
+        password=password,
+        authSource="valdb",
+        authMechanism='SCRAM-SHA-256'
+    )
+    valdb = client.valdb
+    logger.info("Writing data to {}.".format(collection_name))
+    col = valdb[collection_name]
+    for s in sentence_generator:
+        sdata = s[1]
+        col.insert_one(sdata)
 
 
-def to_db():
-    return "TODO"
-
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Parsing corpora kres and ssj500k.")
     parser.add_argument('--kres-folder', required=True)
@@ -67,36 +81,44 @@ if __name__ == "__main__":
     parser.add_argument('--output', required=False, default=None)
     parser.add_argument('--outdir', required=False, default=None)
     parser.add_argument('--dbaddr', required=False, default=None)
+    parser.add_argument('--dbuser', required=False, default=None)
+    parser.add_argument('--dbpass', required=False, default=None)
     args = parser.parse_args()
 
+
     # parse ssj
     logger.info("Parsing ssj500k: {}".format(args.ssj_file))
     ssj_parser = Parser(
         corpus="ssj",
         infiles=[args.ssj_file],
     )
-    # 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(), args.outdir)
+    if args.output == "file":
+        ssj_to_json_file(ssj_parser.sentence_generator(), args.outdir)
+    elif args.output == "db":
+        data_to_valdb(
+            ssj_parser.sentence_generator(),
+            args.dbaddr,
+            args.dbuser,
+            args.dbpass,
+            collection_name="ssj"
+        )
 
+    sys.exit()
     # parse kres
     logger.info("Parsing Kres: {}".format(args.ssj_file))
     kres_parser = Parser(
         corpus="kres",
         infiles=[args.kres_folder, args.kres_srl_folder],
     )
-    # res = [x[1]["sid"] for x in kres_parser.sentence_generator()]
-    # logger.info("Parsed {} sentences (kres)".format(len(res)))
-
     # kres to json
-    kres_to_json_files(kres_parser.sentence_generator(), args.outdir)
-
-
-## Handling output is situational --- implement it outside of Parser.
-## Parser returns tuples (orig_file, element)
-# 1. parse per-file and output to file (JSON)
-# 2. parse and save to DB
-
-# TODO
+    if args.output == "file":
+        kres_to_json_files(kres_parser.sentence_generator(), args.outdir)
+    elif args.output == "db":
+        data_to_valdb(
+            ssj_parser.sentence_generator(),
+            args.dbaddr,
+            args.dbuser,
+            args.dbpass,
+            collection_name="kres"
+        )

dockerfiles/cjvt-corpusparser-env/Dockerfile

@@ -10,6 +10,7 @@ sshfs
 RUN pip3 install \
 	lxml \
 	argparse \
-	pathlib
+	pathlib \
+	pymongo
 
 ENV PYTHONIOENCODING UTF-8