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'])