def append_asset(asset_data, **kwargs): # downloaders=[], location=None,
'''Link asset to the scene'''
file_names = paths.get_download_filenames(asset_data)
props = None
#####
# how to do particle drop:
# link the group we are interested in( there are more groups in File!!!! , have to get the correct one!)
#
scene = bpy.context.scene
user_preferences = bpy.context.preferences.addons['blenderkit'].preferences
if user_preferences.api_key == '':
user_preferences.asset_counter += 1
if asset_data['asset_type'] == 'scene':
scene = append_link.append_scene(file_names[0],
link=False,
fake_user=False)
props = scene.blenderkit
parent = scene
if asset_data['asset_type'] == 'model':
s = bpy.context.scene
downloaders = kwargs.get('downloaders')
s = bpy.context.scene
sprops = s.blenderkit_models
# TODO this is here because combinations of linking objects or appending groups are rather not-usefull
if sprops.append_method == 'LINK_COLLECTION':
sprops.append_link = 'LINK'
sprops.import_as = 'GROUP'
else:
sprops.append_link = 'APPEND'
sprops.import_as = 'INDIVIDUAL'
#copy for override
al = sprops.append_link
import_as = sprops.import_as
# set consistency for objects already in scene, otherwise this literally breaks blender :)
ain = asset_in_scene(asset_data)
#override based on history
if ain is not False:
if ain == 'LINKED':
al = 'LINK'
import_as = 'GROUP'
else:
al = 'APPEND'
import_as = 'INDIVIDUAL'
# first get conditions for append link
link = al == 'LINK'
# then append link
if downloaders:
for downloader in downloaders:
# this cares for adding particle systems directly to target mesh, but I had to block it now,
# because of the sluggishnes of it. Possibly re-enable when it's possible to do this faster?
if 0: # 'particle_plants' in asset_data['tags']:
append_link.append_particle_system(
file_names[-1],
target_object=kwargs['target_object'],
rotation=downloader['rotation'],
link=False,
name=asset_data['name'])
return
if sprops.import_as == 'GROUP':
parent, newobs = append_link.link_collection(
file_names[-1],
location=downloader['location'],
rotation=downloader['rotation'],
link=link,
name=asset_data['name'],
parent=kwargs.get('parent'))
else:
parent, newobs = append_link.append_objects(
file_names[-1],
location=downloader['location'],
rotation=downloader['rotation'],
link=link,
name=asset_data['name'],
parent=kwargs.get('parent'))
if parent.type == 'EMPTY' and link:
bmin = asset_data['bbox_min']
bmax = asset_data['bbox_max']
size_min = min(1.0, (bmax[0] - bmin[0] + bmax[1] -
bmin[1] + bmax[2] - bmin[2]) / 3)
parent.empty_display_size = size_min
elif kwargs.get('model_location') is not None:
if sprops.import_as == 'GROUP':
parent, newobs = append_link.link_collection(
file_names[-1],
location=kwargs['model_location'],
rotation=kwargs['model_rotation'],
link=link,
name=asset_data['name'],
parent=kwargs.get('parent'))
else:
parent, newobs = append_link.append_objects(
file_names[-1],
location=kwargs['model_location'],
rotation=kwargs['model_rotation'],
link=link,
parent=kwargs.get('parent'))
if parent.type == 'EMPTY' and link:
bmin = asset_data['bbox_min']
bmax = asset_data['bbox_max']
size_min = min(1.0, (bmax[0] - bmin[0] + bmax[1] - bmin[1] +
bmax[2] - bmin[2]) / 3)
parent.empty_display_size = size_min
if link:
group = parent.instance_collection
lib = group.library
lib['asset_data'] = asset_data
elif asset_data['asset_type'] == 'brush':
# TODO if already in scene, should avoid reappending.
inscene = False
for b in bpy.data.brushes:
if b.blenderkit.id == asset_data['id']:
inscene = True
brush = b
break
if not inscene:
brush = append_link.append_brush(file_names[-1],
link=False,
fake_user=False)
thumbnail_name = asset_data['thumbnail'].split(os.sep)[-1]
tempdir = paths.get_temp_dir('brush_search')
thumbpath = os.path.join(tempdir, thumbnail_name)
asset_thumbs_dir = paths.get_download_dirs('brush')[0]
asset_thumb_path = os.path.join(asset_thumbs_dir, thumbnail_name)
shutil.copy(thumbpath, asset_thumb_path)
brush.icon_filepath = asset_thumb_path
if bpy.context.view_layer.objects.active.mode == 'SCULPT':
bpy.context.tool_settings.sculpt.brush = brush
elif bpy.context.view_layer.objects.active.mode == 'TEXTURE_PAINT': # could be just else, but for future possible more types...
bpy.context.tool_settings.image_paint.brush = brush
# TODO set brush by by asset data(user can be downloading while switching modes.)
# bpy.context.tool_settings.image_paint.brush = brush
props = brush.blenderkit
parent = brush
elif asset_data['asset_type'] == 'material':
inscene = False
for m in bpy.data.materials:
if m.blenderkit.id == asset_data['id']:
inscene = True
material = m
break
if not inscene:
material = append_link.append_material(file_names[-1],
link=False,
fake_user=False)
target_object = bpy.data.objects[kwargs['target_object']]
if len(target_object.material_slots) == 0:
target_object.data.materials.append(material)
else:
target_object.material_slots[
kwargs['material_target_slot']].material = material
parent = material
scene['assets used'] = scene.get('assets used', {})
scene['assets used'][asset_data['asset_base_id']] = asset_data.copy()
scene['assets rated'] = scene.get('assets rated', {})
id = asset_data['asset_base_id']
scene['assets rated'][id] = scene['assets rated'].get(id, False)
parent[
'asset_data'] = asset_data # TODO remove this??? should write to blenderkit Props?
def download_blenderkit_material(asset_ref):
from blenderkit import paths, append_link, utils, version_checker, rerequests
import requests
def create_asset_data(rdata, asset_type):
for r in rdata['results']:
if r['assetType'] == asset_type and len(r['files']) > 0:
furl = None
tname = None
allthumbs = []
durl, tname = None, None
for f in r['files']:
if f['fileType'] == 'thumbnail':
tname = paths.extract_filename_from_url(
f['fileThumbnailLarge'])
small_tname = paths.extract_filename_from_url(
f['fileThumbnail'])
allthumbs.append(
tname) # TODO just first thumb is used now.
tdict = {}
for i, t in enumerate(allthumbs):
tdict['thumbnail_%i'] = t
if f['fileType'] == 'blend':
durl = f['downloadUrl'].split('?')[0]
# fname = paths.extract_filename_from_url(f['filePath'])
if durl and tname:
tooltip = blenderkit.search.generate_tooltip(r)
r['author']['id'] = str(r['author']['id'])
asset_data = {
'thumbnail': tname,
'thumbnail_small': small_tname,
# 'thumbnails':allthumbs,
'download_url': durl,
'id': r['id'],
'asset_base_id': r['assetBaseId'],
'name': r['name'],
'asset_type': r['assetType'],
'tooltip': tooltip,
'tags': r['tags'],
'can_download': r.get('canDownload', True),
'verification_status': r['verificationStatus'],
'author_id': r['author']['id'],
# 'author': r['author']['firstName'] + ' ' + r['author']['lastName']
# 'description': r['description'],
}
asset_data['downloaded'] = 0
# parse extra params needed for blender here
params = utils.params_to_dict(r['parameters'])
if asset_type == 'model':
if params.get('boundBoxMinX') != None:
bbox = {
'bbox_min': (float(params['boundBoxMinX']),
float(params['boundBoxMinY']),
float(params['boundBoxMinZ'])),
'bbox_max': (float(params['boundBoxMaxX']),
float(params['boundBoxMaxY']),
float(params['boundBoxMaxZ']))
}
else:
bbox = {
'bbox_min': (-.5, -.5, 0),
'bbox_max': (.5, .5, 1)
}
asset_data.update(bbox)
if asset_type == 'material':
asset_data['texture_size_meters'] = params.get(
'textureSizeMeters', 1.0)
asset_data.update(tdict)
r.update(asset_data)
# main
asset_base_id_str, asset_type_str = asset_ref.split()
asset_type = asset_type_str.split(':')[1]
scene_id = blenderkit.download.get_scene_id()
reqstr = '?query=%s+%s+order:_score' % (asset_base_id_str, asset_type_str)
reqstr += '&addon_version=%s' % version_checker.get_addon_version()
reqstr += '&scene_uuid=%s' % scene_id
url = paths.get_api_url() + 'search/' + reqstr
api_key = user_preferences = C.preferences.addons[
'blenderkit'].preferences.api_key
headers = utils.get_headers(api_key)
r = rerequests.get(url, headers=headers)
rdata = r.json()
create_asset_data(rdata, asset_type)
asset_data = rdata['results'][0]
has_url = blenderkit.download.get_download_url(asset_data, scene_id,
api_key)
file_names = paths.get_download_filenames(asset_data)
file_name = file_names[0]
if not os.path.exists(file_name):
with open(file_name, "wb") as f:
print("Downloading %s" % file_name)
response = requests.get(asset_data['url'], stream=True)
total_length = response.headers.get('Content-Length')
if total_length is None: # no content length header
f.write(response.content)
else:
dl = 0
for data in response.iter_content(chunk_size=4096):
dl += len(data)
f.write(data)
material = append_link.append_material(file_names[-1])
return material
def unhide_collection(cname):
collection = bpy.context.scene.collection.children[cname]
collection.hide_viewport = False
collection.hide_render = False
collection.hide_select = False
if __name__ == "__main__":
try:
bg_blender.progress('preparing thumbnail scene')
with open(BLENDERKIT_EXPORT_DATA, 'r') as s:
data = json.load(s)
# append_material(file_name, matname = None, link = False, fake_user = True)
mat = append_link.append_material(
file_name=BLENDERKIT_EXPORT_FILE_INPUT,
matname=data["material"],
link=True,
fake_user=False)
user_preferences = bpy.context.preferences.addons[
'blenderkit'].preferences
s = bpy.context.scene
colmapdict = {
'BALL': 'Ball',
'BALL_COMPLEX': 'Ball complex',
'FLUID': 'Fluid',
'CLOTH': 'Cloth',
'HAIR': 'Hair'
}
file_name=data['source_filepath'],
obnames=obnames,
rotation=(0, 0, 0))
g = bpy.data.collections.new(upload_data['name'])
for o in allobs:
g.objects.link(o)
bpy.context.scene.collection.children.link(g)
if export_data['type'] == 'SCENE':
sname = export_data['scene']
main_source = append_link.append_scene(
file_name=data['source_filepath'], scenename=sname)
bpy.data.scenes.remove(bpy.data.scenes['upload'])
main_source.name = sname
elif export_data['type'] == 'MATERIAL':
matname = export_data['material']
main_source = append_link.append_material(
file_name=data['source_filepath'], matname=matname)
elif export_data['type'] == 'BRUSH':
brushname = export_data['brush']
main_source = append_link.append_brush(
file_name=data['source_filepath'], brushname=brushname)
bpy.ops.file.pack_all()
# TODO fetch asset_id here
asset_id = main_source.blenderkit.asset_base_id
main_source.blenderkit.uploading = False
fpath = os.path.join(data['temp_dir'], asset_id + '.blend')
bpy.ops.wm.save_as_mainfile(filepath=fpath, compress=True, copy=False)
def unhide_collection(cname):
collection = bpy.context.scene.collection.children[cname]
collection.hide_viewport = False
collection.hide_render = False
collection.hide_select = False
if __name__ == "__main__":
try:
bg_blender.progress('preparing thumbnail scene')
with open(BLENDERKIT_EXPORT_DATA, 'r') as s:
data = json.load(s)
# append_material(file_name, matname = None, link = False, fake_user = True)
mat = append_link.append_material(file_name=BLENDERKIT_EXPORT_FILE_INPUT, matname=data["material"], link=True,
fake_user=False)
user_preferences = bpy.context.preferences.addons['blenderkit'].preferences
s = bpy.context.scene
colmapdict = {
'BALL': 'Ball',
'CUBE': 'Cube',
'FLUID': 'Fluid',
'CLOTH': 'Cloth',
'HAIR': 'Hair'
}
unhide_collection(colmapdict[data["thumbnail_type"]])
if data['thumbnail_background']:
def append_asset2(asset_data, **kwargs):
'''Link asset to the scene'''
file_names = paths.get_download_filenames(asset_data)
scene = bpy.context.scene
user_preferences = bpy.context.preferences.addons['blenderkit'].preferences
if user_preferences.api_key == '':
user_preferences.asset_counter += 1
if asset_data['asset_type'] == 'scene':
scene = append_link.append_scene(file_names[0],
link=False,
fake_user=False)
parent = scene
if asset_data['asset_type'] == 'model':
s = bpy.context.scene
downloaders = kwargs.get('downloaders')
s = bpy.context.scene
sprops = s.blenderkit_models
if sprops.append_method == 'LINK_COLLECTION':
sprops.append_link = 'LINK'
sprops.import_as = 'GROUP'
else:
sprops.append_link = 'APPEND'
sprops.import_as = 'INDIVIDUAL'
al = sprops.append_link
ain = asset_in_scene(asset_data)
if ain is not False:
if ain == 'LINKED':
al = 'LINK'
else:
al = 'APPEND'
link = al == 'LINK'
if downloaders:
for downloader in downloaders:
if link is True:
parent, newobs = append_link.link_collection(
file_names[-1],
location=downloader['location'],
rotation=downloader['rotation'],
link=link,
name=asset_data['name'],
parent=kwargs.get('parent'))
else:
parent, newobs = append_link.append_objects(
file_names[-1],
location=downloader['location'],
rotation=downloader['rotation'],
link=link,
name=asset_data['name'],
parent=kwargs.get('parent'))
if parent.type == 'EMPTY' and link:
bmin = asset_data['bbox_min']
bmax = asset_data['bbox_max']
size_min = min(1.0, (bmax[0] - bmin[0] + bmax[1] -
bmin[1] + bmax[2] - bmin[2]) / 3)
parent.empty_display_size = size_min
elif kwargs.get('model_location') is not None:
if link is True:
parent, newobs = append_link.link_collection(
file_names[-1],
location=kwargs['model_location'],
rotation=kwargs['model_rotation'],
link=link,
name=asset_data['name'],
parent=kwargs.get('parent'))
else:
parent, newobs = append_link.append_objects(
file_names[-1],
location=kwargs['model_location'],
rotation=kwargs['model_rotation'],
link=link,
parent=kwargs.get('parent'))
if parent.type == 'EMPTY' and link:
bmin = asset_data['bbox_min']
bmax = asset_data['bbox_max']
size_min = min(1.0, (bmax[0] - bmin[0] + bmax[1] - bmin[1] +
bmax[2] - bmin[2]) / 3)
parent.empty_display_size = size_min
if link:
group = parent.instance_collection
lib = group.library
lib['asset_data'] = asset_data
elif asset_data['asset_type'] == 'brush':
inscene = False
for b in bpy.data.brushes:
if b.blenderkit.id == asset_data['id']:
inscene = True
brush = b
break
if not inscene:
brush = append_link.append_brush(file_names[-1],
link=False,
fake_user=False)
thumbnail_name = asset_data['thumbnail'].split(os.sep)[-1]
tempdir = paths.get_temp_dir('brush_search')
thumbpath = os.path.join(tempdir, thumbnail_name)
asset_thumbs_dir = paths.get_download_dirs('brush')[0]
asset_thumb_path = os.path.join(asset_thumbs_dir, thumbnail_name)
shutil.copy(thumbpath, asset_thumb_path)
brush.icon_filepath = asset_thumb_path
if bpy.context.view_layer.objects.active.mode == 'SCULPT':
bpy.context.tool_settings.sculpt.brush = brush
elif bpy.context.view_layer.objects.active.mode == 'TEXTURE_PAINT':
bpy.context.tool_settings.image_paint.brush = brush
parent = brush
elif asset_data['asset_type'] == 'material':
inscene = False
for m in bpy.data.materials:
if m.blenderkit.id == asset_data['id']:
inscene = True
material = m
break
if not inscene:
material = append_link.append_material(file_names[-1],
link=False,
fake_user=False)
target_object = bpy.data.objects[kwargs['target_object']]
if len(target_object.material_slots) == 0:
target_object.data.materials.append(material)
else:
target_object.material_slots[
kwargs['material_target_slot']].material = material
parent = material
scene['assets used'] = scene.get('assets used', {})
scene['assets used'][asset_data['asset_base_id']] = asset_data.copy()
scene['assets rated'] = scene.get('assets rated', {})
id = asset_data['asset_base_id']
scene['assets rated'][id] = scene['assets rated'].get(id, False)
parent['asset_data'] = asset_data
if hasattr(parent.blenderkit, 'tags') and 'tags' in asset_data:
asset_data['tags'].remove('non-manifold')
parent.blenderkit.tags = ','.join(asset_data['tags'])
if hasattr(parent.blenderkit,
'description') and 'description' in asset_data:
if asset_data['description'] is not None:
parent.blenderkit.description = asset_data['description']
if hasattr(parent.blenderkit, 'custom_props') and 'metadata' in asset_data:
if 'product_info' in asset_data['metadata']:
product_info = asset_data['metadata'].pop('product_info')
clients = []
skus = []
for client_sku in product_info:
clients.append(client_sku['client'])
skus.append(client_sku['sku'])
if hasattr(parent.blenderkit, 'client') and hasattr(
parent.blenderkit, 'sku'):
parent.blenderkit.client = ','.join(clients)
parent.blenderkit.sku = ','.join(skus)
else:
parent.blenderkit.custom_props['client'] = ','.join(clients)
parent.blenderkit.custom_props['sku'] = ','.join(skus)
for key, value in asset_data['metadata'].items():
parent.blenderkit.custom_props[key] = value
bpy.ops.wm.undo_push_context(message='add %s to scene' %
asset_data['name'])
has_url = download.get_download_url(asset_data,
download.get_scene_id(),
user_preferences.api_key,
tcom=None,
resolution='blend')
if not has_url:
bg_blender.progress(
"couldn't download asset for thumnbail re-rendering")
exit()
# download first, or rather make sure if it's already downloaded
bg_blender.progress('downloading asset')
fpath = download.download_asset_file(asset_data)
data['filepath'] = fpath
mat = append_link.append_material(file_name=data['filepath'],
matname=data["asset_name"],
link=True,
fake_user=False)
s = bpy.context.scene
colmapdict = {
'BALL': 'Ball',
'BALL_COMPLEX': 'Ball complex',
'FLUID': 'Fluid',
'CLOTH': 'Cloth',
'HAIR': 'Hair'
}
unhide_collection(colmapdict[data["thumbnail_type"]])
if data['thumbnail_background']:
unhide_collection('Background')
bpy.data.materials["bg checker colorable"].node_tree.nodes['input_level'].outputs['Value'].default_value \
main_source, allobs = append_link.append_objects(file_name=data['source_filepath'],
obnames=obnames,
rotation=(0, 0, 0))
g = bpy.data.collections.new(upload_data['name'])
for o in allobs:
g.objects.link(o)
bpy.context.scene.collection.children.link(g)
if export_data['type'] == 'SCENE':
sname = export_data['scene']
main_source = append_link.append_scene(file_name=data['source_filepath'],
scenename=sname)
bpy.data.scenes.remove(bpy.data.scenes['upload'])
main_source.name = sname
elif export_data['type'] == 'MATERIAL':
matname = export_data['material']
main_source = append_link.append_material(file_name=data['source_filepath'], matname=matname)
elif export_data['type'] == 'BRUSH':
brushname = export_data['brush']
main_source = append_link.append_brush(file_name=data['source_filepath'], brushname=brushname)
bpy.ops.file.pack_all()
# TODO fetch asset_id here
asset_id = main_source.blenderkit.asset_base_id
main_source.blenderkit.uploading = False
fpath = os.path.join(data['temp_dir'], asset_id + '.blend')
bpy.ops.wm.save_as_mainfile(filepath=fpath, compress=True, copy=False)
os.remove(data['source_filepath'])
