add user whitleist

redditrepostsleuth/core/db/databasemodels.py

@@ -350,6 +350,7 @@ class MonitoredSub(Base):
     post_checks = relationship("MonitoredSubChecks", back_populates='monitored_sub', cascade='all, delete', )
     config_revisions = relationship("MonitoredSubConfigRevision", back_populates='monitored_sub', cascade='all, delete')
     config_changes = relationship('MonitoredSubConfigChange', back_populates='monitored_sub', cascade='all, delete')
+    user_whitelist = relationship('UserWhitelist', back_populates='monitored_sub', cascade='all, delete')
 
     def __repr__(self):
         return f'{self.name} | Active: {self.active}'
@@ -416,6 +417,17 @@ def to_dict(self):
         }
 
 
+class UserWhitelist(Base):
+    __tablename__ = 'user_whitelist'
+
+    id = Column(Integer, primary_key=True)
+    username = Column(String(25), nullable=False)
+    monitored_sub_id = Column(Integer, ForeignKey('monitored_sub.id'))
+    ignore_of_detection = Column(Boolean, default=False)
+    ignore_high_volume_detection = Column(Boolean, default=False)
+    ignore_repost_detection = Column(Boolean, default=False)
+
+    monitored_sub = relationship("MonitoredSub", back_populates='user_whitelist', cascade='all, delete')
 
 class MonitoredSubChecks(Base):
     __tablename__ = 'monitored_sub_checked'

redditrepostsleuth/core/db/repository/user_whitelist_repo.py

@@ -0,0 +1,19 @@
+from typing import Optional
+
+from redditrepostsleuth.core.db.databasemodels import UserWhitelist
+
+
+class UserWhitelistRepo:
+
+    def __init__(self, db_session):
+        self.db_session = db_session
+
+    def add(self, item):
+        self.db_session.add(item)
+
+    def get_by_username_and_subreddit(self, username: str, monitored_sub_id: int) -> Optional[UserWhitelist]:
+        return self.db_session.query(UserWhitelist).filter(UserWhitelist.username == username, UserWhitelist.monitored_sub_id == monitored_sub_id).first()
+
+
+    def get_by_username(self, username: str) -> Optional[UserWhitelist]:
+        return self.db_session.query(UserWhitelist).filter(UserWhitelist.username == username).first()

redditrepostsleuth/core/db/uow/unitofwork.py

@@ -30,6 +30,7 @@
 from redditrepostsleuth.core.db.repository.summonsrepository import SummonsRepository
 from redditrepostsleuth.core.db.repository.user_report_repo import UserReportRepo
 from redditrepostsleuth.core.db.repository.user_review_repo import UserReviewRepo
+from redditrepostsleuth.core.db.repository.user_whitelist_repo import UserWhitelistRepo
 
 
 class UnitOfWork:
@@ -170,4 +171,8 @@ def user_review(self) -> UserReviewRepo:
 
     @property
     def post_type(self) -> PostTypeRepo:
-        return PostTypeRepo(self.session)
+        return PostTypeRepo(self.session)
+
+    @property
+    def user_whitelist(self) -> UserWhitelistRepo:
+        return UserWhitelistRepo(self.session)