sort-pkgs: for packages dir try evaluting %_topdir

diff --git a/sort-pkgs b/sort-pkgs
index 7b1fa4199e4901c1e459d4f0d0b5e195546ad171..07c1ac340566dce23eb466862a9442f5384c9e82 100755 (executable)
--- a/sort-pkgs
+++ b/sort-pkgs
@@ -16,7 +16,15 @@ BR_PATTERN = re.compile('BuildRequires:\s+(.*?)(\s|$)')
 PACKAGE_PATTERN_WITH_N = re.compile('%package\s+-n\s+(.*)')
 PACKAGE_PATTERN = re.compile('%package\s+(.*)')
 
-DIR = os.getenv("HOME") + '/rpm/packages'
+try:
+    import rpm
+    DIR = rpm.expandMacro('%_topdir')
+except ModuleNotFoundError:
+    try:
+        import subprocess
+        DIR = subprocess.check_output(['rpm', '-E', '%_topdir']).decode('utf-8').strip()
+    except:
+        DIR = os.getenv("HOME") + '/rpm/packages'
 
 BUILD_REQUIRES = {}
 PACKAGES = {}