From: Ross Burton <ross.bur...@arm.com>
Add an argument parser so that the use of the cache or verbose logging
can be enabled/disabled without having to edit the source code.
Signed-off-by: Ross Burton <ross.bur...@arm.com>
---
scripts/abint/abint.py | 45 ++++++++++++++++++++++--------------------
1 file changed, 24 insertions(+), 21 deletions(-)
diff --git a/scripts/abint/abint.py b/scripts/abint/abint.py
index d4fbed7e..1d358836 100755
--- a/scripts/abint/abint.py
+++ b/scripts/abint/abint.py
@@ -1,5 +1,6 @@
#! /usr/bin/env python3
+import argparse
import collections
import dataclasses
import logging
@@ -11,19 +12,6 @@ import arrow
import bugzilla
import requests
-MOCK = False
-# logging.basicConfig(level=logging.DEBUG)
-
-
-def save_pickle(name, data):
- with open(name, "wb") as f:
- pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
-
-
-def load_pickle(name):
- with open(name, "rb") as f:
- return pickle.load(f)
-
http = requests.Session()
@@ -104,15 +92,9 @@ def get_data():
except requests.exceptions.HTTPError as e:
logging.debug(f"Couldn't find build for {builder=}
{build=}")
- save_pickle("bugs.data", bugs)
return bugs
-if MOCK:
-
- def get_data():
- return load_pickle("bugs.data")
-
def last_seen_report(data):
import jinja2
@@ -129,5 +111,26 @@ def last_seen_report(data):
f.write(template.render(bugs=data, start=start, now=arrow.now()))
-data = get_data()
-last_seen_report(data)
+CACHE_NAME = "bugs.data"
+
+if __name__ == "__main__":
+ parser = argparse.ArgumentParser()
+ parser.add_argument("--verbose", action="store_true", help="Enable verbose
logging")
+ parser.add_argument("--cached", action="store_true", help="Use the local
cached data instead of fetching")
+ args = parser.parse_args()
+
+ if args.verbose:
+ logging.basicConfig(level=logging.DEBUG)
+
+ if args.cached:
+ logging.debug(f"Loading cached data from {CACHE_NAME}")
+ with open(CACHE_NAME, "rb") as f:
+ data = pickle.load(f)
+ else:
+ data = get_data()
+ # Always save a cache because it's quick
+ logging.debug(f"Saving data to cache {CACHE_NAME}")
+ with open(CACHE_NAME, "wb") as f:
+ pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
+
+ last_seen_report(data)
--
2.34.1
-=-=-=-=-=-=-=-=-=-=-=-
Links: You receive all messages sent to this group.
View/Reply Online (#62696): https://lists.yoctoproject.org/g/yocto/message/62696
Mute This Topic: https://lists.yoctoproject.org/mt/104791156/21656
Group Owner: yocto+ow...@lists.yoctoproject.org
Unsubscribe: https://lists.yoctoproject.org/g/yocto/unsub
[arch...@mail-archive.com]
-=-=-=-=-=-=-=-=-=-=-=-
