update-xrefs: add --work-dir argument

1 files changed, 11 insertions, 4 deletions

diff --git a/bin/update-xrefs b/bin/update-xrefs
index 3170650947..662a7d064c 100755
--- a/bin/update-xrefs
+++ b/bin/update-xrefs
@@ -4,6 +4,7 @@
 #
 # Copyright © 2023 Emilio Pozuelo Monfort <pochu@debian.org>
 
+import argparse
 import os
 
 import setup_paths  # noqa
@@ -73,10 +74,16 @@ def remove_xrefs(cves):
                 cve.annotations.remove(ann)
 
 
-dsa_list = os.path.dirname(__file__) + '/../data/DSA/list'
-dtsa_list = os.path.dirname(__file__) + '/../data/DTSA/list'
-dla_list = os.path.dirname(__file__) + '/../data/DLA/list'
-main_list = os.path.dirname(__file__) + '/../data/CVE/list'
+default_workdir = os.path.join(os.path.dirname(os.path.dirname(__file__)))
+
+parser = argparse.ArgumentParser(description='Update cross-references in CVE list')
+parser.add_argument('--work-dir', help='path to security-tracker repo (default: relative to the script)', default=default_workdir)
+args = parser.parse_args()
+
+dsa_list = args.work_dir + '/data/DSA/list'
+dtsa_list = args.work_dir + '/data/DTSA/list'
+dla_list = args.work_dir + '/data/DLA/list'
+main_list = args.work_dir + '/data/CVE/list'
 
 cves = parsers.cvelist(main_list)
 cve_map = {cve.header.name: cve for cve in cves}