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