From 16b2f1a3c2a0ec28dfb22b28a9f20e99a35f022e Mon Sep 17 00:00:00 2001
From: InsanePrawn <insane.prawny@gmail.com>
Date: Sun, 11 Sep 2022 06:37:40 +0200
Subject: [PATCH] packages: Pkgbase.update(): fixups

---
 packages/pkgbuild.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/packages/pkgbuild.py b/packages/pkgbuild.py
index 02cf890..3dc53a8 100644
--- a/packages/pkgbuild.py
+++ b/packages/pkgbuild.py
@@ -180,11 +180,12 @@ class Pkgbase(Pkgbuild):
         if lazy and self.sources_refreshed:
             return
         parsed = parse_pkgbuild(self.path, sources_refreshed=True)
-        basepkgs = [p for p in parsed if isinstance(p, Pkgbase)]
-        if not len(basepkgs) == 1:
-            raise Exception(f"error refreshing {self.name}: wrong number of base packages found: {basepkgs}")
+        basepkg = parsed[0]
+        assert isinstance(basepkg, (Pkgbase, SubPkgbuild))
+        if isinstance(basepkg, SubPkgbuild):
+            basepkg = basepkg.pkgbase
         self.sources_refreshed = True
-        self.update(basepkgs[0])
+        self.update(basepkg)
 
 
 class SubPkgbuild(Pkgbuild):