49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
|
from typing import Optional, Union
|
|||
|
|
from chuangliangTool.db_base import Material1_db, Material2_db, Material3_db, Material4_db
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_material(material_param: Optional[Union[int, str]] = None) -> Optional[str]:
|
|||
|
|
"""
|
|||
|
|
获取素材
|
|||
|
|
参数:
|
|||
|
|
material_param: 可以是 material_id(int), material_name(str) 或 file_md5(str)
|
|||
|
|
返回:
|
|||
|
|
素材
|
|||
|
|
"""
|
|||
|
|
if material_param is None:
|
|||
|
|
raise ValueError("必须提供 material_id, material_name 或 file_md5 参数")
|
|||
|
|
|
|||
|
|
# 根据参数类型构建查询条件
|
|||
|
|
if isinstance(material_param, int):
|
|||
|
|
# 如果是整数，认为是material_id
|
|||
|
|
sql_template = f"SELECT * FROM chuangliang_ad_material.material WHERE material_id = '{material_param}'"
|
|||
|
|
else:
|
|||
|
|
# 如果是字符串，需要判断是material_name还是file_md5
|
|||
|
|
# 这里假设file_md5是32位的MD5值，material_name不是32位
|
|||
|
|
if len(material_param) == 32 and all(c in '0123456789abcdefABCDEF' for c in material_param):
|
|||
|
|
# 如果是32位且只包含十六进制字符，认为是file_md5
|
|||
|
|
sql_template = f"SELECT * FROM chuangliang_ad_material.material WHERE file_md5 = '{material_param}'"
|
|||
|
|
else:
|
|||
|
|
# 否则认为是material_name
|
|||
|
|
sql_template = f"SELECT * FROM chuangliang_ad_material.material WHERE material_name = '{material_param}'"
|
|||
|
|
|
|||
|
|
material_dbs = [Material1_db, Material2_db, Material3_db, Material4_db]
|
|||
|
|
db_names = ["Material1_db", "Material2_db", "Material3_db", "Material4_db"]
|
|||
|
|
|
|||
|
|
for db, db_name in zip(material_dbs, db_names):
|
|||
|
|
try:
|
|||
|
|
result = db.query_database_select_to_dataframe(sql_template)
|
|||
|
|
if result is not None and len(result) > 0:
|
|||
|
|
# 返回整个DataFrame，让调用者处理
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"查询{db_name}出错: {e}")
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
# 测试不同的参数类型
|
|||
|
|
material_data = get_material("豪品素材-城市生活情感街拍短视频 (86)")
|
|||
|
|
print("查询结果:", material_data)
|