From 3cf974da4d2a5dbe51ed397835ecf5121e632da6 Mon Sep 17 00:00:00 2001 From: Hugo Osvaldo Barrera Date: Tue, 19 Jul 2022 18:51:51 +0200 Subject: [PATCH] Include architectures for dummy data --- schema/gendummydata.py | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/schema/gendummydata.py b/schema/gendummydata.py index aedfda7e..205e26e9 100755 --- a/schema/gendummydata.py +++ b/schema/gendummydata.py @@ -96,6 +96,22 @@ def genFortune(): return fortunes[random.randrange(0, len(fortunes))].replace("'", "") +def genArch() -> str: + arches: list[str] = [] + + if random.randrange(0, 10) < 8: + arches.append("x86_64") + if random.randrange(0, 10) < 6: + arches.append("aarch64") + if random.randrange(0, 10) < 2: + arches.append("riscv") + + if arches: + return ",".join(arches) + else: + return "any" + + # load the words, and make sure there are enough words for users/pkgs # log.debug("Grabbing words from seed file...") @@ -235,9 +251,11 @@ for p in list(seen_pkgs.keys()): s = s % (seen_pkgs[p], p, NOW, NOW, uuid, muid, puid) out.write(s) - s = ("INSERT INTO Packages (ID, PackageBaseID, Name, Version) VALUES " - "(%d, %d, '%s', '%s');\n") - s = s % (seen_pkgs[p], seen_pkgs[p], p, genVersion()) + s = ( + "INSERT INTO Packages (ID, PackageBaseID, Name, Version, Arch) VALUES " + "(%d, %d, '%s', '%s', '%s');\n" + ) + s = s % (seen_pkgs[p], seen_pkgs[p], p, genVersion(), genArch()) out.write(s) count += 1