Require some data from all models in initdb

data/database.py

@@ -4,6 +4,8 @@ import uuid
 import time
 import toposort
 import resumablehashlib
+import sys
+import inspect
 
 from random import SystemRandom
 from datetime import datetime
@@ -861,15 +863,5 @@ class TorrentInfo(BaseModel):
       (('storage', 'piece_length'), True),
     )
 
-
-all_models = [User, Repository, Image, AccessToken, Role, RepositoryPermission, Visibility,
-              RepositoryTag, EmailConfirmation, FederatedLogin, LoginService, QueueItem,
-              RepositoryBuild, Team, TeamMember, TeamRole, LogEntryKind, LogEntry,
-              PermissionPrototype, ImageStorage, BuildTriggerService, RepositoryBuildTrigger,
-              OAuthApplication, OAuthAuthorizationCode, OAuthAccessToken, NotificationKind,
-              Notification, ImageStorageLocation, ImageStoragePlacement,
-              ExternalNotificationEvent, ExternalNotificationMethod, RepositoryNotification,
-              RepositoryAuthorizedEmail, ImageStorageTransformation,
-              TeamMemberInvite, ImageStorageSignature, ImageStorageSignatureKind,
-              AccessTokenKind, Star, RepositoryActionCount, TagManifest, UserRegion,
-              QuayService, QuayRegion, QuayRelease, BlobUpload, DerivedStorageForImage, TorrentInfo]
+is_model = lambda x: inspect.isclass(x) and issubclass(x, BaseModel) and x is not BaseModel
+all_models = [model[1] for model in inspect.getmembers(sys.modules[__name__], is_model)]