Added feature #525

src/modules/AmneziaWireguardConfiguration.py

@@ -159,6 +159,15 @@ class AmneziaWireguardConfiguration(WireguardConfiguration):
             sqlalchemy.Column('Info', sqlalchemy.Text),
             extend_existing=True
         )
+        
+        self.peersHistoryEndpointTable = sqlalchemy.Table(
+            f'{dbName}_history_endpoint', self.metadata,
+            sqlalchemy.Column('id', sqlalchemy.String(255), nullable=False),
+            sqlalchemy.Column('endpoint', sqlalchemy.String(255), nullable=False),
+            sqlalchemy.Column('time',
+                              (sqlalchemy.DATETIME if self.DashboardConfig.GetConfig("Database", "type")[1] == 'sqlite' else sqlalchemy.TIMESTAMP)),
+            extend_existing=True
+        )
 
         self.metadata.create_all(self.engine)
 

src/modules/Peer.py

@@ -236,6 +236,20 @@ class Peer:
             return False
         return True
     
+    def getEndpoints(self):
+        result = []
+        with self.configuration.engine.connect() as conn:
+            result = conn.execute(
+                db.select(
+                    self.configuration.peersHistoryEndpointTable.c.endpoint
+                ).group_by(
+                    self.configuration.peersHistoryEndpointTable.c.endpoint
+                ).where(
+                    self.configuration.peersHistoryEndpointTable.c.id == self.id
+                )
+            ).mappings().fetchall()
+        return list(result)
+    
     def getTraffics(self, interval: int = 30, startDate: datetime.datetime = None, endDate: datetime.datetime = None):
         if startDate is None and endDate is None:
             endDate = datetime.datetime.now()

src/modules/WireguardConfiguration.py

@@ -452,11 +452,9 @@ class WireguardConfiguration:
                                             self.peersTable.columns.id == i['PublicKey']
                                         )
                                     )
-                                
                             tmpList.append(Peer(tempPeer, self))
                 except Exception as e:
                     print(f"[WGDashboard] {self.Name} getPeers() Error: {str(e)}")
-                        
         else:
             with self.engine.connect() as conn:
                 existingPeers = conn.execute(self.peersTable.select()).mappings().fetchall()