ignore text files

.gitignore

@@ -1,3 +1,4 @@
 *.exe
-
-*.o
+*.o
+*.txt
+*.log

scripts/import-from-xon.py

@@ -3,10 +3,11 @@ import sqlite3 as sql
 import logging
 import logging.handlers
 
-from os           import listdir, mkdir
 from os.path      import isfile, exists
 from urllib.parse import unquote
 
+import sys, traceback
+
 #------------------------------------------------+
 #  get_list_from_server_txt
 #------------------------------------------------+
@@ -46,8 +47,10 @@ def get_list_from_server_txt(filename):
             output.append(row)
     return output
 
-def init_logging():
-    filename = "_logs/dbimport-%s.log"
+def init_logging(folder, base_file_name="dbimport-%s.log"):
+    if not exists(folder):
+        return False
+    filename = folder + "/" + base_file_name
     i = 0
     while exists(filename % i):
         i += 1
@@ -57,7 +60,7 @@ def init_logging():
     return filename
 
 #------------------------------------------------+
-#   Functions: Clean up.
+#   uid2namefix
 #------------------------------------------------+
 #   Unlike other rows,
 #   the separator character, '/' is part of the value of the second column.
@@ -66,6 +69,7 @@ def init_logging():
 #       \/uid2name/Mnumg2Yh/yxNFDTqGI+YyhlM7QDI0fpEmAaBJ8cI5dU=\Tuxxy
 #   it should become:
 #       ["uid2name", "Mnumg2Yh/yxNFDTqGI+YyhlM7QDI0fpEmAaBJ8cI5dU=", "Tuxxy"]
+
 def uid2namefix(row):
     # quick fix
     # replace first and last occurrence of backslash
@@ -118,74 +122,74 @@ def filters(db):
                 sid.append([ e[0], unquote(e[-1]) ])
     return tt, tr, ti, s, sid
 
-#------------------------------------------------+
-#   Functions: Database Creation
-#------------------------------------------------+
+def insert_to_database(d, s):
+    def insert(c, q, d):
+        for x in d:
+        # possible to do executemany
+        # but want to be able to catch the problematic rows
+        # as it is iterated through.
+        # and proceed with adding OK rows.
+            try:  
+                c.execute(q, x)
+            except sql.ProgrammingError as e:
+                print(e)
+                print(x)
+        return
 
-def inserttodb(c, q, d):
-    for x in d:
-    # possible to do executemany
-    # but want to be able to catch the problematic rows
-    # as it is iterated through.
-    # and proceed with adding OK rows.
-        try:  
-            c.execute(q, x)
-        except sql.ProgrammingError as e:
-            print(e)
-            print(x)
-
-#------------------------------------------------+
-
-# insert new data directly into new database file
-def i(d, s):
     con = sql.connect(d)
     with con:
         csr = con.cursor()
         try:
-            times, ranks, ids, speed, speed_ids = filters(get_list_from_server_txt(s))
+            times, \
+            ranks, \
+            ids,   \
+            speed, \
+            speed_ids = filters(get_list_from_server_txt(s))
             if times:
-                inserttodb(csr, "INSERT OR REPLACE INTO Cts_times VALUES(?, ?, ?, ?)", times)
+                insert(csr, "INSERT OR REPLACE INTO Cts_times VALUES(?, ?, ?, ?)", times)
                 logging.info('\n'.join(y for y in [str(x) for x in times]))
             if ranks:
-                inserttodb(csr, "INSERT OR REPLACE INTO Cts_ranks VALUES(?, ?, ?, ?)", ranks)
+                insert(csr, "INSERT OR REPLACE INTO Cts_ranks VALUES(?, ?, ?, ?)", ranks)
                 logging.info('\n'.join(y for y in [str(x) for x in ranks]))
             if ids:
-                inserttodb(csr, "INSERT OR REPLACE INTO Id2alias VALUES(?, ?, ?)", ids)
+                insert(csr, "INSERT OR REPLACE INTO Id2alias VALUES(?, ?, ?)", ids)
                 logging.info('\n'.join(y for y in [str(x) for x in ids]))
             if speed:
-                inserttodb(csr, "INSERT OR REPLACE INTO Speed VALUES(?, ?)", speed)
+                insert(csr, "INSERT OR REPLACE INTO Speed VALUES(?, ?)", speed)
             if speed_ids:
-                inserttodb(csr, "INSERT OR REPLACE INTO Fastest_players VALUES(?, ?)", speed_ids)
+                insert(csr, "INSERT OR REPLACE INTO Fastest_players VALUES(?, ?)", speed_ids)
         except sql.Error:
             logging.exception("sql error encountered in function 'i'")
             if con:
                 con.rollback()
 
-# 'insert' new data into a file i.e sql query file
-def f(d, s):
-    with open(d, 'w', encoding='utf-8') as h:
-        times, ranks, ids, speed, speed_ids = filters(get_list_from_server_txt(s))
+def write_query(out_file, data):
+    if exists(out_file):
+        print("stopped: output file already exists", file=sys.stderr)
+        return False
+    times, \
+    ranks, \
+    ids,   \
+    speed, \
+    speed_ids = filters(get_list_from_server_txt(data))
+    with open(out_file, 'w', encoding='utf-8') as file_handle:
         for t in times:
-            h.write("INSERT OR REPLACE INTO Cts_times VALUES(%s, %s, %s, %s)\n" % tuple(t))
-            pass
+            file_handle.write("INSERT OR REPLACE INTO Cts_times VALUES(\'%s\', \'%s\', %s, %s);\n" % tuple(t))
         for r in ranks:
-            h.write("INSERT OR REPLACE INTO Cts_ranks VALUES(%s, %s, %s, %s)\n" % tuple(r))
-            pass
+            file_handle.write("INSERT OR REPLACE INTO Cts_ranks VALUES(\'%s\', \'%s\', %s, \'%s\');\n" % tuple(r))
         for i in ids:
-            h.write("INSERT OR REPLACE INTO Id2aslias VALUES(%s, %s, %s)\n" % tuple(i))
-            pass
-        pass
-    pass
+            file_handle.write("INSERT OR REPLACE INTO Id2alias VALUES(\'%s\', \'%s\', \'%s\');\n" % tuple(i))
+    return True
 
 # Test whether repeat rows are added.
-def duplicatestest(d, s):
-    c = sql.connect(d)
+def check_duplicates(database, data):
+    c = sql.connect(database)
     p = True
     with c:
         cs = c.cursor()
         try:
             logging.info("Inserting into database (1/2)")
-            i(d, s)
+            insert_to_database(database, data)
             logging.info("Querying (1/2)")
             cs.execute("SELECT * FROM Cts_times")
             a = cs.fetchall()
@@ -194,7 +198,7 @@ def duplicatestest(d, s):
             cs.execute("SELECT * FROM Id2alias")
             c = cs.fetchall()
             logging.info("Inserting into database (2/2)")
-            i(d, s)
+            insert_to_database(database, data)
             logging.info("Querying (2/2)")
             cs.execute("SELECT * FROM Cts_times")
             x = cs.fetchall()
@@ -218,17 +222,34 @@ def duplicatestest(d, s):
 
 if __name__ == "__main__":
     ap  = argparse.ArgumentParser()
-    ap.add_argument('db')
-    ap.add_argument('src')
-    ap.add_argument('-t', '--test', action='store_true')
-    ap.add_argument('-q', '--sql', action='store_true')
+    ap.add_argument('dest',
+        help="destination, either an sqlite3 database or query file")
+    ap.add_argument('src',
+        help="source, should be data generated by a Xonotic server")
+    ap.add_argument('-t', '--test', 
+        action='store_true',
+        help="test database for duplicates")
+    ap.add_argument('-q', '--export-query', 
+        action='store_true',
+        help="write query file (as opposed to executing / inserting rows into database)")
+    ap.add_argument('-l', '--log', 
+        type=str,
+        help="set folder to store log files")
     args = ap.parse_args()
-    log_file = init_logging()
-    print("Writing log to ", log_file)
-    if args.test:
-        duplicatestest(args.db, args.src)
-    if args.sql:
-        f(args.db, args.src)
+    log_dir  = args.log or "logs"
+    log_file = init_logging(log_dir)
+    if log_file:
+        print("writing log to folder '%s'," % log_dir, log_file, file=sys.stderr)
     else:
-        i(args.db, args.src)
-
+        print("exited: logging not initialized (folder '%s' does not exist)" % log_dir, file=sys.stderr)
+        exit()
+    try:
+        if args.test:
+            check_duplicates(args.dest, args.src)
+        if args.export_query:
+            write_query(args.dest, args.src)
+        else:
+            insert_to_database(args.dest, args.src)
+    except FileNotFoundError:
+        traceback.print_exc()
+        print("\n\t exited: no input file to work with.", file=sys.stderr)