Merge pull request #130 from lanl/pulido/issue129

Pulido/issue129

Author: jpulidojr

dsi/backends/sqlite.py

@@ -186,15 +186,15 @@ def put_artifacts(self, collection, isVerbose=False):
 
     # Returns text list from query
     def get_artifacts(self, query, isVerbose=False, dict_return = False):
-        if query[:6].lower() == "select":
+        if query[:6].lower() == "select" or query[:6].lower() == "pragma" :
             try:
                 data = self.cur.execute(query).fetchall()
                 if isVerbose:
                     print(data)
             except:
                 raise ValueError("Error in get_artifacts handler: Incorrect SELECT query on the data. Please try again")
         else:
-            raise ValueError("Error in get_artifacts handler: Can only run SELECT queries on the data")
+            raise ValueError("Error in get_artifacts handler: Can only run SELECT or PRAGMA queries on the data")
 
         if dict_return:
             query_cols = [description[0] for description in self.cur.description]

examples/wildfire/wildfire_terminal.py

@@ -0,0 +1,17 @@
+#Loading using plugins and backends
+from dsi.core import Terminal
+
+'''Example access workflow once database has been generated'''
+
+a=Terminal(debug_flag=True)
+a.load_module('backend','Sqlite','back-read', filename='wildfire.db')
+a.transload()
+cnames = a.artifact_handler(interaction_type='get', query = "PRAGMA table_info(wfdata);")
+data = a.artifact_handler(interaction_type='get', query = "SELECT * FROM wfdata;")#, isVerbose = True)
+
+# a.artifact_handler(interaction_type="read")
+clist = [i[1] for i in cnames]
+dlist = list(data)
+print(clist)
+print(dlist)
+# a.unload_module('backend', 'Sqlite', 'back-write')