3131from sentry .backup .helpers import ImportFlags
3232from sentry .backup .sanitize import SanitizableField , Sanitizer
3333from sentry .backup .scopes import ImportScope , RelocationScope
34- from sentry .db .models import (
35- BaseManager ,
36- BaseModel ,
37- BoundedBigAutoField ,
38- control_silo_model ,
39- sane_repr ,
40- )
34+ from sentry .db .models import BaseManager , Model , control_silo_model , sane_repr
4135from sentry .db .models .utils import unique_db_instance
4236from sentry .db .postgres .transactions import enforce_constraints
4337from sentry .integrations .types import EXTERNAL_PROVIDERS , ExternalProviders
@@ -94,13 +88,12 @@ def get_users_with_only_one_integration_for_provider(
9488
9589
9690@control_silo_model
97- class User (BaseModel , AbstractBaseUser ):
91+ class User (Model , AbstractBaseUser ):
9892 __relocation_scope__ = RelocationScope .User
9993 __relocation_custom_ordinal__ = ["username" ]
10094
10195 replication_version : int = 2
10296
103- id = BoundedBigAutoField (primary_key = True )
10497 username = models .CharField (_ ("username" ), max_length = MAX_USERNAME_LENGTH , unique = True )
10598 # this column is called first_name for legacy reasons, but it is the entire
10699 # display name
@@ -368,7 +361,7 @@ def merge_to(from_user: User, to_user: User) -> None:
368361 with enforce_constraints (
369362 transaction .atomic (using = router .db_for_write (OrganizationMemberMapping ))
370363 ):
371- control_side_org_models : tuple [type [BaseModel ], ...] = (
364+ control_side_org_models : tuple [type [Model ], ...] = (
372365 OrgAuthToken ,
373366 OrganizationMemberMapping ,
374367 )
@@ -383,7 +376,7 @@ def merge_to(from_user: User, to_user: User) -> None:
383376 # While it would be nice to make the following changes in a transaction, there are too many
384377 # unique constraints to make this feasible. Instead, we just do it sequentially and ignore
385378 # the `IntegrityError`s.
386- user_related_models : tuple [type [BaseModel ], ...] = (
379+ user_related_models : tuple [type [Model ], ...] = (
387380 Authenticator ,
388381 Identity ,
389382 UserAvatar ,
0 commit comments