From 0b5b5a9a8e131d0af5fff8cf29d32b2e266f6e99 Mon Sep 17 00:00:00 2001 From: Simon Quigley Date: Mon, 8 Jun 2020 17:55:58 -0500 Subject: [PATCH] Actually run the SQLite commands and add a hardcoded list at the top for enabling modules. --- metrics | 22 +++++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) diff --git a/metrics b/metrics index 506fced..0bb6a41 100755 --- a/metrics +++ b/metrics @@ -15,9 +15,25 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . +import sqlite3 from modules.jenkins import JenkinsModule +ENABLED_MODULES = [JenkinsModule] + +def sqlite_run(command, db=":memory:"): + """Run the given SQLite command on our db + + command must be a command that SQLite can run + db must be a valid path to a db, or it's done in memory + """ + conn = sqlite3.connect(db) + c = conn.cursor() + c.execute(command) + conn.commit() + conn.close() + if __name__ == "__main__": - jenkins = JenkinsModule() - print(jenkins.sqlite_setup()) - print(jenkins.sqlite_add()) + for module in ENABLED_MODULES: + module = module() + sqlite_run(module.sqlite_setup()) + sqlite_run(module.sqlite_add())