PKDevTools 0.13.20250208.218__tar.gz → 0.13.20250208.220__tar.gz

{pkdevtools-0.13.20250208.218 → pkdevtools-0.13.20250208.220}/PKDevTools/classes/DBManager.py

@@ -410,3 +410,120 @@ class DBManager:
                 self.conn.close()
                 self.conn = None
         return scanners[0].userIds if len(scanners) > 0 else []
+
+    def updateAlertSubscriptionModel(self,userID,charge:float,scanId:str):
+        try:
+            success = False
+            result = self.connection().execute("""
+                                            UPDATE alertsubscriptions
+                                            SET 
+                                                balance = balance - ?,
+                                                scannerJobs = scannerJobs || ';' || ?
+                                            WHERE userId = ?;
+                                        """, (charge, scanId, userID))
+            if result.rows_affected > 0:
+                default_logger().debug(f"User: {userID} updated with balance and scannerJobs!")
+                success = self.topUpScannerJobs(scanId,userID)
+        except Exception as e: # pragma: no cover
+            print(f"Could not updateAlertSubscriptionModel UserID: {userID}\n{e}")
+            default_logger().debug(e, exc_info=True)
+            pass
+        finally:
+            if self.conn is not None:
+                self.conn.close()
+                self.conn = None
+        return success
+
+    def topUpAlertSubscriptionBalance(self,userID,topup:float):
+        """
+        Handles a new user insertion as well as update.
+
+        If UserID does not exist, a new row is inserted with the given Balance.
+        If UserID already exists, only the Balance is updated by adding the new amount 
+        (Balance + excluded.Balance).
+        """
+        try:
+            success = False
+            result = self.connection().execute("""
+                                            INSERT INTO alertsubscriptions (userId, balance) 
+                                            VALUES (?, ?) 
+                                            ON CONFLICT(userId) DO UPDATE 
+                                            SET balance = balance + excluded.balance;
+                                        """, (userID,topup))
+            if result.rows_affected > 0:
+                default_logger().debug(f"User: {userID} topped up with balance !")
+                success = True
+        except Exception as e: # pragma: no cover
+            print(f"Could not topUpAlertSubscriptionBalance UserID: {userID}\n{e}")
+            default_logger().debug(e, exc_info=True)
+            pass
+        finally:
+            if self.conn is not None:
+                self.conn.close()
+                self.conn = None
+        return success
+
+    def topUpScannerJobs(self,scanId,userID):
+        """
+        Handles a new user insertion as well as update for a given scanId.
+
+        If scanId does not exist, a new row is inserted with the given userId.
+        If scanId already exists, only the users is updated by adding the new userId 
+        """
+        try:
+            success = False
+            result = self.connection().execute("""
+                                            INSERT INTO scannerjobs (scannerId, users) 
+                                            VALUES (?, ?) 
+                                            ON CONFLICT(scannerId) DO UPDATE 
+                                            SET users = users || ';' || excluded.users;
+                                        """, (scanId,userID))
+            if result.rows_affected > 0:
+                default_logger().debug(f"User: {userID} added to scanId !")
+                success = True
+        except Exception as e: # pragma: no cover
+            print(f"Could not topUpScannerJobs UserID: {userID}\n{e}")
+            default_logger().debug(e, exc_info=True)
+            pass
+        finally:
+            if self.conn is not None:
+                self.conn.close()
+                self.conn = None
+        return success
+
+    def resetScannerJobs(self):
+        """
+        Truncates scannerJobs and clears all jobs from users' alert subscriptions
+        """
+        try:
+            success1 = False
+            result = self.connection().execute("DELETE from scannerjobs")
+            if result.rows_affected > 0:
+                default_logger().debug(f"scannerJobs truncated !")
+                success1 = True
+        except Exception as e: # pragma: no cover
+            print(f"Could not deleteScannerJobs \n{e}")
+            default_logger().debug(e, exc_info=True)
+            pass
+        finally:
+            if self.conn is not None:
+                self.conn.close()
+                self.conn = None
+        try:
+            success2 = False
+            result = self.connection().execute("""
+                                            UPDATE alertsubscriptions
+                                            SET scannerJobs = ''
+                                        """)
+            if result.rows_affected > 0:
+                default_logger().debug(f"alertsubscriptions updated with cleaned up scannerJobs!")
+                success2 = True
+        except Exception as e: # pragma: no cover
+            print(f"Could not deleteScannerJobs \n{e}")
+            default_logger().debug(e, exc_info=True)
+            pass
+        finally:
+            if self.conn is not None:
+                self.conn.close()
+                self.conn = None
+        return success1 and success2

pkdevtools-0.13.20250208.220/PKDevTools/classes/__init__.py

@@ -0,0 +1 @@
+VERSION='0.13.20250208.220'

{pkdevtools-0.13.20250208.218 → pkdevtools-0.13.20250208.220}/PKDevTools.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.2
 Name: PKDevTools
-Version: 0.13.20250208.218
+Version: 0.13.20250208.220
 Summary: A general day-to-day toolset for PKScreener repos
 Home-page: https://github.com/pkjmesra/PKDevTools
-Download-URL: https://github.com/pkjmesra/PKDevTools/archive/v0.13.20250208.218.zip
+Download-URL: https://github.com/pkjmesra/PKDevTools/archive/v0.13.20250208.220.zip
 Author: pkjmesra
 Author-email: pkjmesra@gmail.com
 License: OSI Approved (MIT)

{pkdevtools-0.13.20250208.218 → pkdevtools-0.13.20250208.220}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.2
 Name: PKDevTools
-Version: 0.13.20250208.218
+Version: 0.13.20250208.220
 Summary: A general day-to-day toolset for PKScreener repos
 Home-page: https://github.com/pkjmesra/PKDevTools
-Download-URL: https://github.com/pkjmesra/PKDevTools/archive/v0.13.20250208.218.zip
+Download-URL: https://github.com/pkjmesra/PKDevTools/archive/v0.13.20250208.220.zip
 Author: pkjmesra
 Author-email: pkjmesra@gmail.com
 License: OSI Approved (MIT)

pkdevtools-0.13.20250208.218/PKDevTools/classes/__init__.py

@@ -1 +0,0 @@
-VERSION='0.13.20250208.218'