diff --git a/BaiHu/Tools/addModel.py b/BaiHu/Tools/addModel.py index d8d71cf9..e6a754a2 100644 --- a/BaiHu/Tools/addModel.py +++ b/BaiHu/Tools/addModel.py @@ -1,37 +1,59 @@ # pip install psycopg2 +import requests +from Util.AesUtil import * +from Util.CommonUtil import set_http_proxy from Util.PgUtil import * -# 修改操作(插入、更新或删除) -sql = "INSERT INTO t_hy_model(model_id,model_name,memo,star,update_time,img_count,b_use,sort_id,model_type_id,style_id,is_cover) VALUES (%s,%s, %s,%s,now(),%s,1,1,%s,%s,%s);" - # 模型编号 -model_id = 24 - +model_id = 48 # 模型名称 -model_name = '草原风' - -# 提示信息 -memo = '1、请上传正面半身照。; 2、确保脸部无遮挡;3、图片要求不小于1.5MB(正常手机拍照即可)' - -# 星级 -star = 5 - -# 上传照片数量 -img_count = 1 - +model_name = '海的女儿' # 模板类型 # 1:SD TXT2IMG 2:SD IMG2IMG 3:COMFY_UI model_type_id = 2 - ''' 1 卡通风格 2 儿童摄影 3 风格写真 4 双人系列 ''' -style_id = 3 +style_id = 2 +# 是不是封面 is_cover = 1 -params = (model_id, model_name, memo, star, img_count, model_type_id, style_id, is_cover) -execute_modify(sql, params) +# 提示信息 +memo = '1、请上传正面半身照。; 2、确保脸部无遮挡;3、图片要求不小于1.5MB(正常手机拍照即可)' +# 星级 +star = 5 +# 上传照片数量 +img_count = 1 + +if __name__ == '__main__': + # 禁止代理 + set_http_proxy("") + # 配置文件 + config = ConfigUtil.getConfig() + # 处理机编号 + machine_id = config['system']['machine_id'] + # 获取签名 + en_data = getenData(machine_id) + # WEB服务器地址 + web_url = config['webServer']['web_url'] + # 请求的地址 + url = web_url + '/QingLong/HuiYa/addModel' + + # 要发送的数据,可以是字典形式 + data = { + "enData": en_data, + "model_name": model_name, + "memo": memo, + "star": star, + "img_count": img_count, + "model_type_id": model_type_id, + "style_id": style_id, + "is_cover": is_cover + } + # 发送POST请求 + response = requests.post(url, data=data) + print(response.text) diff --git a/BaiHu/Tools/addModelWithDataBase.py b/BaiHu/Tools/addModelWithDataBase.py new file mode 100644 index 00000000..d8d71cf9 --- /dev/null +++ b/BaiHu/Tools/addModelWithDataBase.py @@ -0,0 +1,37 @@ +# pip install psycopg2 + +from Util.PgUtil import * + +# 修改操作(插入、更新或删除) +sql = "INSERT INTO t_hy_model(model_id,model_name,memo,star,update_time,img_count,b_use,sort_id,model_type_id,style_id,is_cover) VALUES (%s,%s, %s,%s,now(),%s,1,1,%s,%s,%s);" + +# 模型编号 +model_id = 24 + +# 模型名称 +model_name = '草原风' + +# 提示信息 +memo = '1、请上传正面半身照。; 2、确保脸部无遮挡;3、图片要求不小于1.5MB(正常手机拍照即可)' + +# 星级 +star = 5 + +# 上传照片数量 +img_count = 1 + +# 模板类型 +# 1:SD TXT2IMG 2:SD IMG2IMG 3:COMFY_UI +model_type_id = 2 + +''' +1 卡通风格 +2 儿童摄影 +3 风格写真 +4 双人系列 +''' +style_id = 3 +is_cover = 1 + +params = (model_id, model_name, memo, star, img_count, model_type_id, style_id, is_cover) +execute_modify(sql, params) diff --git a/BaiHu/Tools/addModelWithInterface.py b/BaiHu/Tools/addModelWithInterface.py deleted file mode 100644 index e6a754a2..00000000 --- a/BaiHu/Tools/addModelWithInterface.py +++ /dev/null @@ -1,59 +0,0 @@ -# pip install psycopg2 -import requests - -from Util.AesUtil import * -from Util.CommonUtil import set_http_proxy -from Util.PgUtil import * - -# 模型编号 -model_id = 48 -# 模型名称 -model_name = '海的女儿' -# 模板类型 -# 1:SD TXT2IMG 2:SD IMG2IMG 3:COMFY_UI -model_type_id = 2 -''' -1 卡通风格 -2 儿童摄影 -3 风格写真 -4 双人系列 -''' -style_id = 2 -# 是不是封面 -is_cover = 1 - -# 提示信息 -memo = '1、请上传正面半身照。; 2、确保脸部无遮挡;3、图片要求不小于1.5MB(正常手机拍照即可)' -# 星级 -star = 5 -# 上传照片数量 -img_count = 1 - -if __name__ == '__main__': - # 禁止代理 - set_http_proxy("") - # 配置文件 - config = ConfigUtil.getConfig() - # 处理机编号 - machine_id = config['system']['machine_id'] - # 获取签名 - en_data = getenData(machine_id) - # WEB服务器地址 - web_url = config['webServer']['web_url'] - # 请求的地址 - url = web_url + '/QingLong/HuiYa/addModel' - - # 要发送的数据,可以是字典形式 - data = { - "enData": en_data, - "model_name": model_name, - "memo": memo, - "star": star, - "img_count": img_count, - "model_type_id": model_type_id, - "style_id": style_id, - "is_cover": is_cover - } - # 发送POST请求 - response = requests.post(url, data=data) - print(response.text)