diff --git a/database/Member.ts b/database/Member.ts index 0b2361e..777310b 100644 --- a/database/Member.ts +++ b/database/Member.ts @@ -52,3 +52,5 @@ export default class Member { @prop() public additionalAcknowledgement: MemberAdditionalAcknowledgement[] | undefined; } + +export const MemberModel = getModelForClass(Member); diff --git a/database/Partner.ts b/database/Partner.ts index eaf1079..2c9446f 100644 --- a/database/Partner.ts +++ b/database/Partner.ts @@ -40,7 +40,10 @@ export enum PartnerCommissionType { TRIAL, } -export default class Partner extends Member { +export default class Partner { + @prop({ required: true, unique: true }) + public discordID: string | undefined; + @prop({ required: true }) public roleType: PartnerRoleType | undefined; @@ -56,3 +59,5 @@ export default class Partner extends Member { @prop() public directReport: Partner | string | undefined; } + +export const PartnerModel = getModelForClass(Partner);