rewrite get_suite in an excessively fancy way

Author: mwhudson

livefs_edit/actions.py

@@ -334,6 +334,9 @@ def add_autoinstall_config(ctxt, autoinstall_config):
 
 @register_action()
 def resign_pool(ctxt, dist=None):
+    if dist is None:
+        dist = ctxt.get_suite()
+
     gpgconf = ctxt.tmpfile()
     gpghome = ctxt.tmpdir()
     with open(gpgconf, 'x') as c:
@@ -361,6 +364,7 @@ def resign_pool(ctxt, dist=None):
     with open(key_path, 'w') as new_key:
         run(['gpg', '--home', gpghome, '--export'], stdout=new_key)
 
+
 @register_action()
 def add_debs_to_pool(ctxt, debs: List[str] = ()):
     from debian import deb822

livefs_edit/context.py

@@ -184,9 +184,11 @@ def get_arch(self):
             return fp.read().strip().split()[-2]
 
     def get_suite(self):
-        path=glob.glob(self.p('old/iso/dists/*/Release'))
-        suite=path[0].split(os.sep)[6]
-        return suite
+        from deb822 import Deb822
+        paths = glob.glob(self.p('old/iso/dists/*/Release'))
+        with open(paths[0]) as fp:
+            release = Deb822(fp)
+        return release['Suite']
 
     def edit_squashfs(self, name, *, add_sys_mounts=True):
         lower = self.mount_squash(name)