Epic 2 APIs

backend/config.py

@@ -1,7 +1,6 @@
 import os
 from dotenv import load_dotenv
 from datetime import timedelta
-
 if os.environ.get("FLASK_ENV") != "production":
     load_dotenv()
 
@@ -23,6 +22,11 @@ class Config(object):
     POSTGRES_DB = os.environ.get("POSTGRES_DB", "police_data")
 
     # Flask-Mail SMTP server settings
+    """
+    Config settings for email sending
+    Put all of the information in your dotenv
+    config file
+    """
     MAIL_SERVER = os.environ.get("MAIL_SERVER")
     MAIL_PORT = os.environ.get("MAIL_PORT")
     MAIL_USE_SSL = bool(os.environ.get("MAIL_USE_SSL"))
@@ -31,9 +35,21 @@ class Config(object):
     MAIL_PASSWORD = os.environ.get("MAIL_PASSWORD")
     MAIL_DEFAULT_SENDER = os.environ.get(
         "MAIL_DEFAULT_SENDER",
-        "National Police Data Coalition <{email}>".format(email=MAIL_USERNAME),
+        "National Police Data Coalition <{email}>".format(
+              email=MAIL_USERNAME),
     )
 
+    """
+    Testing configurations with Mailtrap Email testing, all the configurations
+    will be different--go to mailtrap for more information
+    """
+    # MAIL_SERVER = 'sandbox.smtp.mailtrap.io'
+    # MAIL_PORT = 2525
+    # MAIL_USERNAME = '30a682ceaa0416'
+    # MAIL_PASSWORD = 'dbf502527604b1'
+    # MAIL_USE_TLS = True
+    # MAIL_USE_SSL = False
+
     # Flask-User settings
     USER_APP_NAME = (
         "Police Data Trust"  # Shown in and email templates and page footers

backend/database/models/partner.py

@@ -25,6 +25,41 @@ def get_value(self):
             return 5
 
 
+class Invitation(db.Model):
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    partner_id = db.Column(
+        db.Integer, db.ForeignKey('partner.id'), primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), primary_key=True)
+    role = db.Column(db.Enum(MemberRole), nullable=False)
+    is_accepted = db.Column(db.Boolean, default=False)
+    # default to not accepted invite
+
+    def serialize(self):
+        return {
+            'id': self.id,
+            'partner_id': self.partner_id,
+            'user_id': self.user_id,
+            'role': self.role,
+            'is_accepted': self.is_accepted,
+        }
+
+
+class StagedInvitation(db.Model):
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    partner_id = db.Column(
+        db.Integer, db.ForeignKey('partner.id'), primary_key=True)
+    email = db.Column(db.String, unique=True, primary_key=True)
+    role = db.Column(db.Enum(MemberRole), nullable=False)
+
+    def serialize(self):
+        return {
+            'id': self.id,
+            'partner_id': self.partner_id,
+            'email': self.email,
+            'role': self.role
+        }
+
+
 class PartnerMember(db.Model, CrudMixin):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
@@ -51,6 +86,13 @@ def create(self, refresh: bool = True):
         self.date_joined = datetime.now()
         return super().create(refresh)
 
+    def __repr__(self):
+        """Represent instance as a unique string."""
+        return f"<PartnerMember( \
+        id={self.id}, \
+        partner_id={self.partner_id}, \
+        user_id={self.user_id})>"
+
 
 class Partner(db.Model, CrudMixin):
     def __init__(self, **kwargs):

backend/dto/__init__.py

@@ -1,3 +1,4 @@
 # flake8: noqa: F401
 from .user.register_user import RegisterUserDTO
 from .user.login_user import LoginUserDTO
+from .user.invite_user import InviteUserDTO

backend/dto/user/invite_user.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel, EmailStr
+from enum import Enum
+
+
+class MemberRole(str, Enum):
+    ADMIN = "Administrator"
+    PUBLISHER = "Publisher"
+    MEMBER = "Member"
+    SUBSCRIBER = "Subscriber"
+
+
+class InviteUserDTO(BaseModel):
+    partner_id: int
+    email: EmailStr
+    role:  MemberRole

backend/routes/auth.py

@@ -10,7 +10,7 @@
 from pydantic.main import BaseModel
 from ..auth import min_role_required, user_manager
 from ..mixpanel.mix import track_to_mp
-from ..database import User, UserRole, db
+from ..database import User, UserRole, db, Invitation, StagedInvitation
 from ..dto import LoginUserDTO, RegisterUserDTO
 from ..schemas import UserSchema, validate
 
@@ -87,6 +87,23 @@ def register():
         db.session.add(user)
         db.session.commit()
         token = create_access_token(identity=user.id)
+
+        """
+        code to handle adding staged_invitations-->invitations for users
+        who have just signed up for NPDC
+        """
+        staged_invite = StagedInvitation.query.filter_by(email=user.email).all()
+        if staged_invite is not None and len(staged_invite) > 0:
+            for instance in staged_invite:
+                new_invitation = Invitation(
+                    user_id=user.id,
+                    role=instance.role,
+                    partner_id=instance.partner_id)
+                db.session.add(new_invitation)
+            db.session.commit()
+            StagedInvitation.query.filter_by(email=user.email).delete()
+            db.session.commit()
+
         resp = jsonify(
             {
                 "status": "ok",