Lubuntu/metrics
8
0
Fork 0
mirror of https://github.com/lubuntu-team/metrics.git synced 2025-10-31 14:04:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a CLI argument for the location of the DB.

Browse Source
This commit is contained in:
Simon Quigley 2020-06-18 20:22:37 -05:00
parent b8fbd5ef6c
commit 9068136a8e
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
metrics
View File

@ -15,13 +15,14 @@
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>. # along with this program. If not, see <https://www.gnu.org/licenses/>.
import argparse
import sqlite3 import sqlite3
from modules.jenkins import JenkinsModule from modules.jenkins import JenkinsModule
ENABLED_MODULES = [JenkinsModule] ENABLED_MODULES = [JenkinsModule]
def sqlite_run(command, db=":memory:", return_output=False): def sqlite_run(command, db, return_output=False):
"""Run the given SQLite command on our db """Run the given SQLite command on our db
command must be a command that SQLite can run command must be a command that SQLite can run
@ -51,9 +52,17 @@ def main(module):
run.append(module.sqlite_setup()) run.append(module.sqlite_setup())
run.append(module.sqlite_add()) run.append(module.sqlite_add())
run.append(module.sqlite_time_range(days=10)) run.append(module.sqlite_time_range(days=10))
print(sqlite_run(run, return_output=True))
# Use --db-location to pass to sqlite
print(sqlite_run(run, db=args.db_location, return_output=True))
if __name__ == "__main__": if __name__ == "__main__":
# Parse CLI arguments
parser = argparse.ArgumentParser()
parser.add_argument("--db-location", type=str, default=":memory:",
help="Specify the location for the SQLite database")
args = parser.parse_args()
for module in ENABLED_MODULES: for module in ENABLED_MODULES:
main(module) main(module)