Add .import command

Author: zzl0

.gitignore

@@ -6,6 +6,7 @@
 /src
 /test/behave.ini
 /litecli_env
+/.venv
 /.eggs
 
 .vagrant

changelog.md

@@ -1,3 +1,10 @@
+Upcomming:
+==========
+
+Features:
+---------
+* Added `.import` command for importing data from file into table. (Thanks: [Zhaolong Zhu])
+
 1.2.0
 =====
 
@@ -34,4 +41,5 @@ Features:
 [Amjith]: https://blog.amjith.com
 [Zhiming Wang]: https://github.com/zmwangx
 [Irina Truong]: https://github.com/j-bennet
-[Shawn Chapla]: https://github.com/shwnchpl 
+[Shawn Chapla]: https://github.com/shwnchpl
+[Zhaolong Zhu]: https://github.com/zzl0

litecli/packages/completion_engine.py

@@ -110,7 +110,7 @@ def suggest_special(text):
     if cmd in ["\\f", "\\fs", "\\fd"]:
         return [{"type": "favoritequery"}]
 
-    if cmd in ["\\d", "\\dt", "\\dt+", ".schema"]:
+    if cmd in ["\\d", "\\dt", "\\dt+", ".schema", ".import"]:
         return [
             {"type": "table", "schema": []},
             {"type": "view", "schema": []},

litecli/packages/special/dbcommands.py

@@ -1,6 +1,8 @@
 from __future__ import unicode_literals
+import csv
 import logging
 import os
+import sys
 import platform
 import shlex
 from sqlite3 import ProgrammingError
@@ -216,3 +218,47 @@ def read_script(cur, arg, **_):
         script = f.read()
         cur.executescript(script)
     return [(None, None, None, "")]
+
+
+@special_command(
+    ".import",
+    ".import filename table",
+    "Import data from filename into table",
+    arg_type=PARSED_QUERY,
+    case_sensitive=True,
+)
+def import_file(cur, arg=None, **_):
+    args = shlex.split(arg)
+    if len(args) != 2:
+        raise TypeError("Usage: .import filename table")
+
+    filename, table = args
+    cur.execute("PRAGMA table_info(%s)" % table)
+    ncols = len(cur.fetchall())
+    insert_tmpl = 'INSERT INTO "%s" VALUES (?%s)' % (table, ',?' * (ncols - 1))
+
+    with open(filename, "r") as csvfile:
+        dialect = csv.Sniffer().sniff(csvfile.read(1024))
+        csvfile.seek(0)
+        reader = csv.reader(csvfile, dialect)
+
+        cur.execute("BEGIN")
+        ninserted, nignored = 0, 0
+        for i, row in enumerate(reader):
+            if len(row) != ncols:
+                print(
+                    "%s:%d expected %d columns but found %d - ignored" % (
+                        filename, i, ncols, len(row)
+                    ),
+                    file=sys.stderr
+                )
+                nignored += 1
+                continue
+            cur.execute(insert_tmpl, row)
+            ninserted += 1
+        cur.execute("COMMIT")
+
+    status = "Inserted %d rows into %s" % (ninserted, table)
+    if nignored > 0:
+        status += " (%d rows are ignored)" % nignored
+    return [(None, None, None, status)]