修改了服务器的查询
This commit is contained in:
parent
f117aa90c6
commit
34d43fab29
File diff suppressed because it is too large
Load Diff
@ -137,19 +137,18 @@ class PromptService:
|
||||
|
||||
def get_style_content(self, styleName: str) -> str:
|
||||
"""获取风格提示词内容"""
|
||||
# 尝试从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
with self.db_pool.get_connection() as conn:
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute(
|
||||
"SELECT description FROM contentStyle WHERE styleName = %s",
|
||||
(styleName,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
if result:
|
||||
logger.info(f"从数据库获取风格提示词: {styleName}")
|
||||
return result['description']
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute(
|
||||
"SELECT description FROM contentStyle WHERE styleName = %s",
|
||||
(styleName,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
if result:
|
||||
logger.info(f"从数据库获取风格提示词: {styleName}")
|
||||
return result['description']
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取风格提示词失败: {e}")
|
||||
|
||||
@ -177,19 +176,16 @@ class PromptService:
|
||||
# 优先从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute(
|
||||
"SELECT description FROM targetAudience WHERE audienceName = %s",
|
||||
(audience_name,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
if result:
|
||||
logger.info(f"从数据库获取受众提示词: {audience_name}")
|
||||
return result["description"]
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute(
|
||||
"SELECT description FROM targetAudience WHERE audienceName = %s",
|
||||
(audience_name,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
if result:
|
||||
logger.info(f"从数据库获取受众提示词: {audience_name}")
|
||||
return result["description"]
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取受众提示词失败: {e}")
|
||||
|
||||
@ -233,22 +229,18 @@ class PromptService:
|
||||
Returns:
|
||||
景区信息内容
|
||||
"""
|
||||
# 优先从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute(
|
||||
"SELECT description FROM scenicSpot WHERE name = %s",
|
||||
(spot_name,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
if result:
|
||||
logger.info(f"从数据库获取景区信息: {spot_name}")
|
||||
return result["description"]
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute(
|
||||
"SELECT description FROM scenicSpot WHERE name = %s",
|
||||
(spot_name,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
if result:
|
||||
logger.info(f"从数据库获取景区信息: {spot_name}")
|
||||
return result["description"]
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取景区信息失败: {e}")
|
||||
|
||||
@ -280,22 +272,18 @@ class PromptService:
|
||||
Returns:
|
||||
产品信息内容
|
||||
"""
|
||||
# 优先从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute(
|
||||
"SELECT detailedDescription FROM product WHERE productName = %s",
|
||||
(product_name,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
if result:
|
||||
logger.info(f"从数据库获取产品信息: {product_name}")
|
||||
return result["detailedDescription"]
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute(
|
||||
"SELECT detailedDescription FROM product WHERE productName = %s",
|
||||
(product_name,)
|
||||
)
|
||||
result = cursor.fetchone()
|
||||
if result:
|
||||
logger.info(f"从数据库获取产品信息: {product_name}")
|
||||
return result["detailedDescription"]
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取产品信息失败: {e}")
|
||||
|
||||
@ -400,16 +388,13 @@ class PromptService:
|
||||
# 优先从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute("SELECT styleName as name, description FROM contentStyle")
|
||||
results = cursor.fetchall()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
if results:
|
||||
logger.info(f"从数据库获取所有风格: {len(results)}个")
|
||||
return results
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute("SELECT styleName as name, description FROM contentStyle")
|
||||
results = cursor.fetchall()
|
||||
if results:
|
||||
logger.info(f"从数据库获取所有风格: {len(results)}个")
|
||||
return results
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取所有风格失败: {e}")
|
||||
|
||||
@ -447,16 +432,13 @@ class PromptService:
|
||||
# 优先从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute("SELECT audienceName as name, description FROM targetAudience")
|
||||
results = cursor.fetchall()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
if results:
|
||||
logger.info(f"从数据库获取所有受众: {len(results)}个")
|
||||
return results
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute("SELECT audienceName as name, description FROM targetAudience")
|
||||
results = cursor.fetchall()
|
||||
if results:
|
||||
logger.info(f"从数据库获取所有受众: {len(results)}个")
|
||||
return results
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取所有受众失败: {e}")
|
||||
|
||||
@ -494,19 +476,16 @@ class PromptService:
|
||||
# 优先从数据库获取
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute("SELECT name as name, description FROM scenicSpot")
|
||||
results = cursor.fetchall()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
|
||||
if results:
|
||||
logger.info(f"从数据库获取所有景区: {len(results)}个")
|
||||
return [{
|
||||
"name": item["name"],
|
||||
"description": item["description"][:100] + "..." if len(item["description"]) > 100 else item["description"]
|
||||
} for item in results]
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor(dictionary=True) as cursor:
|
||||
cursor.execute("SELECT name as name, description FROM scenicSpot")
|
||||
results = cursor.fetchall()
|
||||
if results:
|
||||
logger.info(f"从数据库获取所有景区: {len(results)}个")
|
||||
return [{
|
||||
"name": item["name"],
|
||||
"description": item["description"][:100] + "..." if len(item["description"]) > 100 else item["description"]
|
||||
} for item in results]
|
||||
except Exception as e:
|
||||
logger.error(f"从数据库获取所有景区失败: {e}")
|
||||
|
||||
@ -543,34 +522,32 @@ class PromptService:
|
||||
# 优先保存到数据库
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor()
|
||||
|
||||
# 检查是否存在
|
||||
cursor.execute(
|
||||
"SELECT COUNT(*) FROM contentStyle WHERE styleName = %s",
|
||||
(name,)
|
||||
)
|
||||
count = cursor.fetchone()[0]
|
||||
|
||||
if count > 0:
|
||||
# 更新
|
||||
cursor.execute(
|
||||
"UPDATE contentStyle SET description = %s WHERE styleName = %s",
|
||||
(description, name)
|
||||
)
|
||||
else:
|
||||
# 插入
|
||||
cursor.execute(
|
||||
"INSERT INTO contentStyle (styleName, description) VALUES (%s, %s)",
|
||||
(name, description)
|
||||
)
|
||||
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
logger.info(f"风格保存到数据库成功: {name}")
|
||||
return True
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor() as cursor:
|
||||
|
||||
# 检查是否存在
|
||||
cursor.execute(
|
||||
"SELECT COUNT(*) FROM contentStyle WHERE styleName = %s",
|
||||
(name,)
|
||||
)
|
||||
count = cursor.fetchone()[0]
|
||||
|
||||
if count > 0:
|
||||
# 更新
|
||||
cursor.execute(
|
||||
"UPDATE contentStyle SET description = %s WHERE styleName = %s",
|
||||
(description, name)
|
||||
)
|
||||
else:
|
||||
# 插入
|
||||
cursor.execute(
|
||||
"INSERT INTO contentStyle (styleName, description) VALUES (%s, %s)",
|
||||
(name, description)
|
||||
)
|
||||
|
||||
conn.commit()
|
||||
logger.info(f"风格保存到数据库成功: {name}")
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"风格保存到数据库失败: {e}")
|
||||
|
||||
@ -609,34 +586,32 @@ class PromptService:
|
||||
# 优先保存到数据库
|
||||
if self.db_pool:
|
||||
try:
|
||||
conn = self.db_pool.get_connection()
|
||||
cursor = conn.cursor()
|
||||
|
||||
# 检查是否存在
|
||||
cursor.execute(
|
||||
"SELECT COUNT(*) FROM targetAudience WHERE audienceName = %s",
|
||||
(name,)
|
||||
)
|
||||
count = cursor.fetchone()[0]
|
||||
|
||||
if count > 0:
|
||||
# 更新
|
||||
cursor.execute(
|
||||
"UPDATE targetAudience SET description = %s WHERE audienceName = %s",
|
||||
(description, name)
|
||||
)
|
||||
else:
|
||||
# 插入
|
||||
cursor.execute(
|
||||
"INSERT INTO targetAudience (audienceName, description) VALUES (%s, %s)",
|
||||
(name, description)
|
||||
)
|
||||
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
logger.info(f"受众保存到数据库成功: {name}")
|
||||
return True
|
||||
with self.db_pool.get_connection() as conn:
|
||||
with conn.cursor() as cursor:
|
||||
|
||||
# 检查是否存在
|
||||
cursor.execute(
|
||||
"SELECT COUNT(*) FROM targetAudience WHERE audienceName = %s",
|
||||
(name,)
|
||||
)
|
||||
count = cursor.fetchone()[0]
|
||||
|
||||
if count > 0:
|
||||
# 更新
|
||||
cursor.execute(
|
||||
"UPDATE targetAudience SET description = %s WHERE audienceName = %s",
|
||||
(description, name)
|
||||
)
|
||||
else:
|
||||
# 插入
|
||||
cursor.execute(
|
||||
"INSERT INTO targetAudience (audienceName, description) VALUES (%s, %s)",
|
||||
(name, description)
|
||||
)
|
||||
|
||||
conn.commit()
|
||||
logger.info(f"受众保存到数据库成功: {name}")
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"受众保存到数据库失败: {e}")
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user