style: format all Python files with Black
Apply Black formatter to the entire codebase for consistent code style. Co-Authored-By: Claude (glm-5) <noreply@anthropic.com>
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
"""
|
||||
BIPUsers DAO - Data access object for user authentication and management
|
||||
"""
|
||||
|
||||
from typing import Optional, Dict, Any, List
|
||||
from db.base_dao import BaseDAO
|
||||
from db.connection import get_connection
|
||||
@@ -22,7 +23,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Dict with user info if authentication successful, None otherwise
|
||||
Returns: {id, username, user_type}
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
@@ -43,13 +44,15 @@ class BIPUsersDAO(BaseDAO):
|
||||
results = db.execute_query(sql, (username, password))
|
||||
if results:
|
||||
return {
|
||||
'id': results[0]['ID'],
|
||||
'username': results[0]['UserName'],
|
||||
'user_type': results[0]['UserType']
|
||||
"id": results[0]["ID"],
|
||||
"username": results[0]["UserName"],
|
||||
"user_type": results[0]["UserType"],
|
||||
}
|
||||
return None
|
||||
|
||||
def authenticate_by_computer_name(self, computer_name: str) -> Optional[Dict[str, Any]]:
|
||||
def authenticate_by_computer_name(
|
||||
self, computer_name: str
|
||||
) -> Optional[Dict[str, Any]]:
|
||||
"""
|
||||
Authenticate a user using computer name (silent login)
|
||||
|
||||
@@ -60,7 +63,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Dict with user info if authentication successful, None otherwise
|
||||
Returns: {id, username, user_type}
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# Note: Column name is 'ComputerNmae' (typo in database schema)
|
||||
@@ -81,9 +84,9 @@ class BIPUsersDAO(BaseDAO):
|
||||
results = db.execute_query(sql, (computer_name,))
|
||||
if results:
|
||||
return {
|
||||
'id': results[0]['ID'],
|
||||
'username': results[0]['UserName'],
|
||||
'user_type': results[0]['UserType']
|
||||
"id": results[0]["ID"],
|
||||
"username": results[0]["UserName"],
|
||||
"user_type": results[0]["UserType"],
|
||||
}
|
||||
return None
|
||||
|
||||
@@ -94,7 +97,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Returns:
|
||||
List of user dictionaries: [{id, username, user_type, create_time}]
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
if self._db_type == DatabaseType.MYSQL:
|
||||
@@ -114,15 +117,17 @@ class BIPUsersDAO(BaseDAO):
|
||||
results = db.execute_query(sql)
|
||||
return [
|
||||
{
|
||||
'id': row['ID'],
|
||||
'username': row['UserName'],
|
||||
'user_type': row['UserType'],
|
||||
'create_time': row['CreateTime']
|
||||
"id": row["ID"],
|
||||
"username": row["UserName"],
|
||||
"user_type": row["UserType"],
|
||||
"create_time": row["CreateTime"],
|
||||
}
|
||||
for row in results
|
||||
]
|
||||
|
||||
def create_user(self, username: str, password: str, user_type: str, computer_name: str = '') -> bool:
|
||||
def create_user(
|
||||
self, username: str, password: str, user_type: str, computer_name: str = ""
|
||||
) -> bool:
|
||||
"""
|
||||
Create a new user with optional computer name for silent login
|
||||
|
||||
@@ -135,7 +140,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Returns:
|
||||
True if successful, False otherwise
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
@@ -185,7 +190,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Returns:
|
||||
True if successful, False otherwise
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
@@ -221,7 +226,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Returns:
|
||||
True if successful, False otherwise
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
@@ -256,7 +261,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Returns:
|
||||
True if successful, False otherwise
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
@@ -289,7 +294,7 @@ class BIPUsersDAO(BaseDAO):
|
||||
Returns:
|
||||
True if username exists, False otherwise
|
||||
"""
|
||||
table_name = self._convert_sql('[dbo].[BIPUsers]')
|
||||
table_name = self._convert_sql("[dbo].[BIPUsers]")
|
||||
placeholder = self._get_placeholder()
|
||||
|
||||
# 根据数据库类型选择列名格式
|
||||
@@ -306,4 +311,4 @@ class BIPUsersDAO(BaseDAO):
|
||||
|
||||
with get_connection() as db:
|
||||
results = db.execute_query(sql, (username,))
|
||||
return results[0]['count'] > 0 if results else False
|
||||
return results[0]["count"] > 0 if results else False
|
||||
|
||||
Reference in New Issue
Block a user