diff --git a/main.py b/main.py index f4e91af7..4a77ed23 100644 --- a/main.py +++ b/main.py @@ -389,6 +389,18 @@ def update_data_for_package(package: str) -> None: (package, check_name, check_score), ) + classifiers = resp["info"].get("classifiers") or [] + for classifier in classifiers: + db.execute( + """ + INSERT OR IGNORE INTO classifiers ( + package_name, + name + ) VALUES (?, ?); + """, + (package, classifier), + ) + return package @@ -551,6 +563,16 @@ def get_google_assured_oss_packages(http: urllib3.PoolManager) -> set[str]: ); """ ) + _DB.execute( + """ + CREATE TABLE IF NOT EXISTS classifiers ( + package_name TEXT, + name TEXT, + PRIMARY KEY (package_name, name), + FOREIGN KEY (package_name) REFERENCES packages(name) + ); + """ + ) _DB.execute( """ CREATE INDEX IF NOT EXISTS idx_packages_name ON packages (name);