From 013b116eb5eb6aca5861d6d54ae194b945ccbda2 Mon Sep 17 00:00:00 2001
From: pycook <pycook@126.com>
Date: Wed, 21 Feb 2024 18:10:44 +0800
Subject: [PATCH] feat(acl): login channel add ssh options (#396)

---
 cmdb-api/api/lib/perm/acl/audit.py | 1 +
 cmdb-api/api/models/acl.py         | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/cmdb-api/api/lib/perm/acl/audit.py b/cmdb-api/api/lib/perm/acl/audit.py
index f741345..1682e36 100644
--- a/cmdb-api/api/lib/perm/acl/audit.py
+++ b/cmdb-api/api/lib/perm/acl/audit.py
@@ -389,6 +389,7 @@ class AuditCRUD(object):
                        logout_at=logout_at,
                        ip=request.headers.get('X-Real-IP') or request.remote_addr,
                        browser=request.headers.get('User-Agent'),
+                       channel=request.values.get('channel', 'web'),
                        )
 
         if logout_at is None:
diff --git a/cmdb-api/api/models/acl.py b/cmdb-api/api/models/acl.py
index e3396c3..730a012 100644
--- a/cmdb-api/api/models/acl.py
+++ b/cmdb-api/api/models/acl.py
@@ -356,7 +356,7 @@ class AuditLoginLog(Model2):
     __tablename__ = "acl_audit_login_logs"
 
     username = db.Column(db.String(64), index=True)
-    channel = db.Column(db.Enum('web', 'api'), default="web")
+    channel = db.Column(db.Enum('web', 'api', 'ssh'), default="web")
     ip = db.Column(db.String(15))
     browser = db.Column(db.String(256))
     description = db.Column(db.String(128))