Merge pull request #217 from zachvalenta/add-dot-command-for-views

add dot command for views

Author: amjith

CHANGELOG.md

@@ -5,6 +5,7 @@
 * Add logs while invoking `\llm`and `\\m+` command. [(#215)](https://github.com/dbcli/litecli/pull/215)
 * Support `--help` in the `\llm`and `\llm+` command. ([#214](https://github.com/dbcli/litecli/pull/214))
 * Make the history file location configurable. ([#206](https://github.com/dbcli/litecli/issues/206))
+* Add dot command to list views.
 
 ### Bug Fixes
 

litecli/packages/special/dbcommands.py

@@ -56,6 +56,40 @@ def list_tables(cur, arg=None, arg_type=PARSED_QUERY, verbose=False):
     return [(None, tables, headers, status)]
 
 
+@special_command(
+    ".views",
+    "\\dv",
+    "List views.",
+    arg_type=PARSED_QUERY,
+    case_sensitive=True,
+    aliases=("\\dv",),
+)
+def list_views(cur, arg=None, arg_type=PARSED_QUERY, verbose=False):
+    if arg:
+        args = ("{0}%".format(arg),)
+        query = """
+            SELECT name FROM sqlite_master
+            WHERE type = 'view' AND name LIKE ? AND name NOT LIKE 'sqlite_%'
+            ORDER BY 1
+        """
+    else:
+        args = tuple()
+        query = """
+            SELECT name FROM sqlite_master
+            WHERE type = 'view' AND name NOT LIKE 'sqlite_%'
+            ORDER BY 1
+        """
+    log.debug(query)
+    cur.execute(query, args)
+    views = cur.fetchall()
+    status = ""
+    if cur.description:
+        headers = [x[0] for x in cur.description]
+    else:
+        return [(None, None, None, "")]
+    return [(None, views, headers, status)]
+
+
 @special_command(
     ".schema",
     ".schema[+] [table]",