From 61dd64c0e68b8fd4e43f8ceb03bfeffce89de1ce Mon Sep 17 00:00:00 2001
From: Joachim Desroches <joachim.desroches@epfl.ch>
Date: Thu, 16 Sep 2021 14:51:49 +0200
Subject: [PATCH] Make DB name an argument.

---
 prometheus-odoo-license-exporter | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/prometheus-odoo-license-exporter b/prometheus-odoo-license-exporter
index a6b2341..ffe7c7b 100755
--- a/prometheus-odoo-license-exporter
+++ b/prometheus-odoo-license-exporter
@@ -4,17 +4,22 @@
 from prometheus_client import start_http_server, Gauge
 from datetime import datetime
 from time import sleep
+
 import psycopg2
+import sys
 
 
 USER = "odoo"
 HTTP_PORT = 9402
 INTERVAL = 10
 
-# FIXME: Make a parameter
-DBNAME = "edurable_staging"
-
 if __name__ == '__main__':
+    if len(sys.argv) == 2:
+        DBNAME = sys.argv[1]
+    else:
+        sys.exit("Invalid number of arguments. " +
+                 "Usage: {} [dbname]".format(sys.argv[0]))
+
     # Setup prometheus Gauge
     prom_days = Gauge(
             'odoo_license_remaining_days',