record multi-arch header in package table

1 files changed, 5 insertions, 4 deletions

diff --git a/readyaml.py b/readyaml.py
index cd9e5db..6940f94 100755
--- a/readyaml.py
+++ b/readyaml.py
@@ -31,12 +31,13 @@ def readyaml(db, stream):
     if pid is not None:
         cur.execute("DELETE FROM content WHERE pid = ?;", (pid,))
         cur.execute("DELETE FROM dependency WHERE pid = ?;", (pid,))
-        cur.execute("UPDATE package SET version = ?, source = ? WHERE id = ?;",
-                    (metadata["version"], metadata["source"], pid))
+        cur.execute("UPDATE package SET version = ?, source = ?, multiarch = ? WHERE id = ?;",
+                    (metadata["version"], metadata["source"],
+                     metadata.get("multiarch"), pid))
     else:
-        cur.execute("INSERT INTO package (name, version, architecture, source) VALUES (?, ?, ?, ?);",
+        cur.execute("INSERT INTO package (name, version, architecture, source, multiarch) VALUES (?, ?, ?, ?, ?);",
                     (package, metadata["version"], metadata["architecture"],
-                     metadata["source"]))
+                     metadata["source"], metadata.get("multiarch")))
         pid = cur.lastrowid
     cur.executemany("INSERT INTO dependency (pid, required) VALUES (?, ?);",
                     ((pid, dep) for dep in metadata["depends"]))