Added peer traffic and session

src/modules/Peer.py

@@ -1,10 +1,12 @@
 """
 Peer
 """
+import datetime
 import os, subprocess, uuid, random, re
+from datetime import timedelta
 
 import jinja2
-
+import sqlalchemy as db
 from .PeerJob import PeerJob
 from .PeerShareLink import PeerShareLink
 from .Utilities import GenerateWireguardPublicKey, ValidateIPAddressesWithRange, ValidateDNSAddress
@@ -232,4 +234,54 @@ class Peer:
         except Exception as e:
             print(e)
             return False
-        return True
+        return True
+    
+    def getSessions(self, startDate: datetime.datetime = None, endDate: datetime.datetime = None):
+        if endDate is None:
+            endDate = datetime.datetime.now()
+        
+        if startDate is None:
+            startDate = (endDate - datetime.timedelta(days=1))
+
+        endDate = endDate.replace(hour=23, minute=59, second=59, microsecond=999999)
+        startDate = startDate.replace(hour=0, minute=0, second=0, microsecond=0)
+            
+        
+        with self.configuration.engine.connect() as conn:
+            result = conn.execute(
+                db.select(
+                    self.configuration.peersTransferTable.c.time
+                ).where(
+                    db.and_(
+                        self.configuration.peersTransferTable.c.id == self.id,
+                        self.configuration.peersTransferTable.c.time <= endDate,
+                        self.configuration.peersTransferTable.c.time >= startDate,
+                    )
+                ).order_by(
+                    self.configuration.peersTransferTable.c.time
+                )
+            ).fetchall()
+        # sessions = []
+        # current_session = [time[0]]
+        # 
+        # for ts in time[1:]:
+        #     if ts - current_session[-1] <= datetime.timedelta(minutes=3):
+        #         current_session.append(ts)
+        #     else:
+        #         sessions.append({
+        #             "duration": self.__duration(current_session[-1], current_session[0]),
+        #             "timestamps": current_session
+        #         })
+        #         current_session = [ts]
+        # sessions.append({
+        #     "duration": self.__duration(current_session[-1], current_session[0]),
+        #     "timestamps": current_session
+        # })
+        return list(map(lambda x : x[0], result))
+    
+    def __duration(self, t1: datetime.datetime, t2: datetime.datetime):
+        delta = t1 - t2
+        
+        hours, remainder = divmod(delta.total_seconds(), 3600)
+        minutes, seconds = divmod(remainder, 60)
+        return f"{int(hours):02}:{int(minutes):02}:{int(seconds):02}"