added `label` column to `agent_metadata` (#71)
need graham to do migration...not touching any models until that is done
taylor_socfortress committed
Jul 26, 2023 at 10:43 UTC
22cee9bae3b133e9e351b42d469ea7c223ce5a56
1 file changed
+5
backend/app/models/agents.py
+5
@@ -22,6 +22,7 @@ class AgentMetadata(db.Model):
22
ip_address: Column[String] = db.Column(db.String(100))
23
os: Column[String] = db.Column(db.String(100))
24
hostname: Column[String] = db.Column(db.String(100))
25
+ label: Column[String] = db.Column(db.String(100))
26
critical_asset: Column[Boolean] = db.Column(db.Boolean, default=False)
27
last_seen: Column[DateTime] = db.Column(db.DateTime)
28
client_id: Column[String] = db.Column(db.String(100))
@@ -35,6 +36,7 @@ class AgentMetadata(db.Model):
36
ip_address: str,
37
os: str,
38
hostname: str,
39
+ label: str,
40
critical_asset: bool,
41
last_seen: datetime,
42
client_id: str,
@@ -49,6 +51,7 @@ class AgentMetadata(db.Model):
51
:param ip_address: IP address of the agent.
52
:param os: Operating system of the agent.
53
:param hostname: Hostname of the agent.
54
+ :param label: Label of the agent.
55
:param critical_asset: Boolean value indicating if the agent is a critical asset.
56
:param last_seen: Timestamp of when the agent was last seen.
57
:param client_id: Unique ID for the client stored in Velociraptor.
@@ -58,6 +61,7 @@ class AgentMetadata(db.Model):
61
self.ip_address = ip_address
62
self.os = os
63
self.hostname = hostname
64
+ self.label = label
65
self.critical_asset = critical_asset
66
self.last_seen = last_seen
67
self.client_id = client_id
@@ -111,6 +115,7 @@ class AgentMetadataSchema(ma.Schema):
115
"ip_address",
116
"os",
117
"hostname",
118
+ "label",
119
"critical_asset",
120
"last_seen",
121
"client_id",