utils: allow overriding current architecture name ...

... acbs will now read ARCH override value from /etc/autobuild/ab4cfg.sh

Author: liushuyu

acbs/ab4cfg.py

@@ -5,16 +5,25 @@
 from acbs import bashvar
 from acbs.const import AUTOBUILD_CONF_DIR
 
-from typing import Optional
+from typing import Optional, Dict
 
 
-def is_in_stage2_file(abcfg_path: str) -> bool:
+def _parse_abcfg(abcfg_path: str) -> Dict[str, str]:
     with open(abcfg_path) as f:
         vars = bashvar.eval_bashvar(f.read(), filename=abcfg_path)
         assert isinstance(vars, dict)
-        stage2_val: Optional[str] = vars.get('ABSTAGE2')
-        return stage2_val == '1'
-    return False
+        return vars
+
+
+def get_arch_override(abcfg_path: str) -> Optional[str]:
+    vars = _parse_abcfg(abcfg_path)
+    return vars.get('ARCH')
+
+
+def is_in_stage2_file(abcfg_path: str) -> bool:
+    vars = _parse_abcfg(abcfg_path)
+    stage2_val: Optional[str] = vars.get('ABSTAGE2')
+    return stage2_val == '1'
 
 
 def is_in_stage2_env() -> bool:

acbs/utils.py

@@ -10,6 +10,7 @@
 from typing import Dict, List, Optional, Sequence, Tuple, cast
 
 from acbs import __version__
+from acbs.ab4cfg import get_arch_override
 from acbs.base import ACBSPackageInfo, ACBSSourceInfo
 from acbs.const import (
     ANSI_BROWN,
@@ -18,6 +19,7 @@
     ANSI_RED,
     ANSI_RST,
     ANSI_YELLOW,
+    AUTOBUILD_CONF_DIR,
 )
 from acbs.crypto import check_hash_hashlib_inner
 
@@ -104,7 +106,11 @@ def get_arch_name() -> Optional[str]:
 
     :returns: architecture name
     """
+    abcfg_path = os.path.join(AUTOBUILD_CONF_DIR, 'ab4cfg.sh')
     try:
+        arch_override = get_arch_override(abcfg_path)
+        if arch_override:
+            return arch_override
         output = subprocess.check_output(['dpkg', '--print-architecture'])
         return output.decode('utf-8').strip()
     except Exception: