diff --git a/data/database.py b/data/database.py index 0e724c6b0..9726adf5b 100644 --- a/data/database.py +++ b/data/database.py @@ -1173,5 +1173,12 @@ class BitTorrentPieces(BaseModel): ) -is_model = lambda x: inspect.isclass(x) and issubclass(x, BaseModel) and x is not BaseModel +beta_classes = set([ManifestLayerScan, Tag, BlobPlacementLocation, ManifestLayer, ManifestList, + BitTorrentPieces, MediaType, Label, ManifestBlob, BlobUploading, Blob, + ManifestLayerDockerV1, BlobPlacementLocationPreference, ManifestListManifest, + Manifest, DerivedImage, BlobPlacement]) +is_model = lambda x: (inspect.isclass(x) and + issubclass(x, BaseModel) and + x is not BaseModel and + x not in beta_classes) all_models = [model[1] for model in inspect.getmembers(sys.modules[__name__], is_model)]