def save_data(design_case):
image = Image.objects(img_url=design_case.cover_url).first()
if image:
print('此作品已存在素材库，添加失败，失败作品_id', str(design_case._id))
elif design_case.prize_label == '红星奖':
image = Image()
image.title = design_case.title # 标题
image.tags = design_case.tags # 标签
image.img_url = design_case.cover_url # 图片地址
image.designer = design_case.designer_name # 设计者姓名
# company = DesignCompany.objects(number=design_case.target_id).first() # 公司
if design_case.company_name:
image.company = design_case.company_name # 公司姓名
elif design_case.en_company_name:
image.company = design_case.en_company_name # 公司姓名
image.kind = 1 #类型
# image.brand_id = #品牌id
#d = prize_options(design_case.prize_label) # 获取奖项id和名称的字典映射
image.prize_id = 4 # 奖项ID
image.prize = design_case.prize_label # 奖项名称
image.prize_level = design_case.prize_level # 奖项级别
image.prize_time = design_case.award_time # 奖项时间
image.channel = 'opalus_design_case'
image.evt = 1
# 保存数据到素材库
ok = True
#ok = image.save()
if not ok:
print("保存成功: %s" % design_case.prize_label)
else:
print("保存失败: %s" % str(design_case._id))
print('作品以成功转入素材库,成功作品_id', str(design_case._id))
else:
print("作品奖项不匹配: %s" % str(design_case._id))
def insert_album_images(album_id, images):
result = []
for image in images:
data = bytes(image, "ascii")
entity = Image(data=data, album_id=album_id, name="No name")
db.session.add(entity)
db.session.commit()
result.append(entity.serialize)
return result
def produce_submit():
ok = False
form = SaveApi(request.values)
if not form.validate_on_submit():
return jsonify(code=3004, message=str(form.errors))
data = request.values.to_dict()
if not data:
return jsonify(code=3003, message='至少传入一个参数!')
try:
if not (data['title'] and data['channel']):
return jsonify(code=3003, message='标题和渠道为必填')
if not data['img_urls']:
return jsonify(code=3003, message='img_urls为空')
produce = Produce.objects(title=data['title'],
channel=data['channel']).first()
if not produce:
produce_data = data.copy()
produce_data.pop('img_urls')
produce = Produce(**produce_data)
produce.save()
else:
# if produce.editor_id != 0:
if produce.editor_id == 0:
produce_data = data.copy()
produce_data.pop('img_urls')
ok = produce.update(**produce_data)
if not ok:
return jsonify(code=3003, message='产品更新失败')
img_urls = data['img_urls'].split(',')
for i, img_url in enumerate(img_urls):
img_data = {}
img_data['title'] = data['title']
img_data['img_url'] = img_url
img_data['channel'] = data['channel']
img_data['url'] = data['url']
img_data['target_id'] = str(produce._id)
img = Image.objects(img_url=img_url).first()
if not img:
img = Image(**img_data)
ok = img.save()
else:
ok = img.update(target_id=str(produce._id))
if i == 0:
if not produce.cover_id:
ok = produce.update(cover_id=str(img._id))
if ok:
return jsonify(code=0, message='success!', data=produce)
else:
return jsonify(code=3010, message='操作失败！')
except (Exception) as e:
return jsonify(code=3011, message=str(e))
def upload_media(request):
latest_files = request.files.getlist("files")
event = request.form.get("event")
week = request.form.get("week")
if latest_files is not None:
for uploaded_file in latest_files:
original_filename, extension = os.path.splitext(
uploaded_file.filename)
MAX_SIZE = 1500, 1500
# Spara orginalet
filename = str(
uuid.uuid4()
) + extension ## Generera ett unikt filnamn så att det inte blir krockar
path = os.path.join("static", "media", filename)
img = Img.open(uploaded_file)
img.thumbnail(MAX_SIZE)
local_path = os.path.join(os.getcwd(), path)
img.save(local_path)
# Skapa en thumbnail
thumb = Img.open(local_path)
filename_thumb = str(uuid.uuid4()) + extension
path_thumb = os.path.join("static", "media", filename_thumb)
local_path_thumb = os.path.join(os.getcwd(), path_thumb)
thumb.thumbnail([400, 400])
thumb.save(local_path_thumb)
new_img = Image(filename=path,
event_id=event,
week=week,
thumbnail=path_thumb)
db.session.add(new_img)
latest_videos = request.form.getlist("videos")
if latest_videos is not None:
for video in latest_videos:
embed_link, thumbnail = handle_video(video)
new_vid = Video(video_link=embed_link,
event_id=event,
week=week,
thumbnail=thumbnail)
db.session.add(new_vid)
db.session.commit()
return True
def image_submit():
form = SaveApi(request.values)
if not form.validate_on_submit():
return jsonify(code=3004, message=str(form.errors))
data = request.values.to_dict()
if not data:
return jsonify(code=3003, message='至少传入一个参数!')
try:
item = Image(**data)
ok = item.save()
if ok:
return jsonify(code=0, message='success!', data=item)
else:
return jsonify(code=3010, message='操作失败！')
except (Exception) as e:
return jsonify(code=3011, message=str(e))
def test_image(client):
image = Image(filename='test', extension='test')
assert image.filename == 'test'
assert image.extension == 'test'
assert image.created == dt_helper(dt.utcnow())
def make_image(self, data, **kwargs):
return Image(**data)
def add_new(self, bidata, text=""):
new_image = Image(bidata, text)
db.session.add(new_image)
db.session.commit()
return new_image
