From fc80647a41a895867b48500f7f6a0e2feb2a9b01 Mon Sep 17 00:00:00 2001
From: Matthew <matthew@staff.libraryofcode.org>
Date: Mon, 1 Apr 2024 20:04:34 -0400
Subject: [PATCH] addtl changes for Member.ts and Partner.ts

---
 database/Member.ts  | 2 ++
 database/Partner.ts | 7 ++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

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);