➕ Add basic repo functions for invite codes

backend/README.md

@@ -1,6 +1,6 @@
 # Thunderbird Appointment Backend
 
-This is the backend component of Thunderbird Appointment written in Python using FastAPI, SQLAlchemy, and pytest. 
+This is the backend component of Thunderbird Appointment written in Python using FastAPI, SQLAlchemy, and pytest.
 
 ## Installation / Running
 
@@ -20,7 +20,7 @@ You will want to ensure any variable ending with `_SECRET` has a secret value as
 
 ### Authentication
 
-This project is deployed with Mozilla Accounts (known as fxa in the code.) Since Mozilla Accounts is for internal use you will need to use password authentication. Note: password authentication does not currently have a registration flow. 
+This project is deployed with Mozilla Accounts (known as fxa in the code.) Since Mozilla Accounts is for internal use you will need to use password authentication. Note: password authentication does not currently have a registration flow.
 
 ## Commands
 

backend/src/appointment/database/repo.py

@@ -557,7 +557,33 @@ def schedule_has_slot(db: Session, schedule_id: int, slot_id: int):
     return db_slot and db_slot.schedule_id == schedule_id
 
 
-"""External Connections repository functions"""
+"""INVITES repository functions
+"""
+
+
+def use_invite_code(db: Session, code: str, subscriber_id: int):
+    """set existing invite code status to revoked"""
+    db_invite = db.query(models.Invite).filter(models.Invite.code == code).first()
+    if db_invite.status == models.InviteStatus.active:
+        db_invite.subscriber_id = subscriber_id
+        db.commit()
+        db.refresh(db_invite)
+        return True
+    else:
+        return False
+
+
+def revoke_invite_code(db: Session, code: str):
+    """set existing invite code status to revoked"""
+    db_invite = db.query(models.Invite).filter(models.Invite.code == code).first()
+    db_invite.status = models.InviteStatus.revoked
+    db.commit()
+    db.refresh(db_invite)
+    return True
+
+
+"""External Connections repository functions
+"""
 
 
 def create_subscriber_external_connection(db: Session, external_connection: ExternalConnection):