Merge pull request #1 from kumar1202/add-config-arg

Add wal-g config file path as an optional argument

README.md

@@ -15,12 +15,14 @@ If there is no output from the query, then this exporter won't run and give an e
 ## Usage
 
 ```
-usage: wal-g-exporter [-h] --archive_dir ARCHIVE_DIR [--debug] [--version]
+usage: wal-g-exporter [-h] --archive_dir ARCHIVE_DIR [--debug] [--config] CONFIG_FILE_PATH [--version]
 
 optional arguments:
   -h, --help            show this help message and exit
   --archive_dir ARCHIVE_DIR
                         pg_wal/archive_status/ Directory location
+  --config CONFIG_FILE_PATH
+                        file path for wal-g config
   --debug               enable debug log
   --version             show binary version
 ```

exporter.py

@@ -21,8 +21,10 @@
 parser.version = "0.1.0"
 parser.add_argument("--archive_dir",
                     help="pg_wal/archive_status/ Directory location", action="store", required=True)
+parser.add_argument("--config", help="walg config file path", action="store")
 parser.add_argument("--debug", help="enable debug log", action="store_true")
 parser.add_argument("--version", help="show binary version", action="version")
+
 args = parser.parse_args()
 if args.debug:
     logging.basicConfig(level=logging.DEBUG)
@@ -167,8 +169,12 @@ def update_basebackup(self, *unused):
         info('Updating basebackups metrics...')
         try:
             # Fetch remote backup list
-            res = subprocess.run(["wal-g", "backup-list",
-                                  "--detail", "--json"],
+            command = ["wal-g", "backup-list",
+                                  "--detail", "--json"]
+            if args.config:
+                command.extend(["--config", args.config])
+
+            res = subprocess.run(command,
                                  capture_output=True, check=True)
 
             # Check if backup-list return an empty result