BlenderKit initial commit.

BlenderKit add-on is the official addon of the BlenderKit service for Blender 3d. (www.blenderkit.com)
It enables users to upload, search, download, and rate different assets for blender.
It works together with BlenderKit server.

1 files changed, 207 insertions, 0 deletions

diff --git a/blenderkit/ratings.py b/blenderkit/ratings.py
new file mode 100644
index 00000000..6928b87a
--- /dev/null
+++ b/blenderkit/ratings.py
@@ -0,0 +1,207 @@
+# ##### BEGIN GPL LICENSE BLOCK #####
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU General Public License
+#  as published by the Free Software Foundation; either version 2
+#  of the License, or (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software Foundation,
+#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+#
+# ##### END GPL LICENSE BLOCK #####
+
+if "bpy" in locals():
+    import imp
+
+    imp.reload(paths)
+
+else:
+    from blenderkit import paths, utils
+
+import bpy
+import requests, threading
+
+from bpy.props import (
+    IntProperty,
+    FloatProperty,
+    StringProperty,
+    EnumProperty,
+    BoolProperty,
+    PointerProperty,
+)
+from bpy.types import (
+    Operator,
+    Panel,
+)
+
+
+def pretty_print_POST(req):
+    """
+    At this point it is completely built and ready
+    to be fired; it is "prepared".
+
+    However pay attention at the formatting used in
+    this function because it is programmed to be pretty
+    printed and may differ from the actual request.
+    """
+    print('{}\n{}\n{}\n\n{}'.format(
+        '-----------START-----------',
+        req.method + ' ' + req.url,
+        '\n'.join('{}: {}'.format(k, v) for k, v in req.headers.items()),
+        req.body,
+    ))
+
+
+def uplaod_rating_thread(url, ratings, headers):
+    for rating_name, score in ratings:
+        if (score != -1 and score != 0):
+            rating_url = url + rating_name + '/'
+            data = {
+                "score": score,  # todo this kind of mixing is too much. Should have 2 bkit structures, upload, use
+            }
+
+            try:
+                r = requests.put(rating_url, data=data, verify=True, headers=headers)
+
+            except requests.exceptions.RequestException as e:
+                print('ratings upload failed: %s' % str(e))
+
+
+def uplaod_review_thread(url, reviews, headers):
+    r = requests.put(url, data=reviews, verify=True, headers=headers)
+
+    # except requests.exceptions.RequestException as e:
+    #     print('reviews upload failed: %s' % str(e))
+
+
+def upload_rating(asset):
+    user_preferences = bpy.context.preferences.addons['blenderkit'].preferences
+    api_key = user_preferences.api_key
+    headers = {"accept": "application/json", "Authorization": "Bearer %s" % api_key}
+
+    asset_data = asset['asset_data']
+
+    bkit_ratings = asset.bkit_ratings
+    print('rating asset', asset_data['name'], asset_data['asset_base_id'])
+    url = paths.get_bkit_url() + 'assets/' + asset['asset_data']['id'] + '/rating/'
+
+    ratings = [
+    ]
+
+    if bkit_ratings.rating_quality > 0.1:
+        ratings.append(('quality', bkit_ratings.rating_quality))
+    if bkit_ratings.rating_work_hours > 0.1:
+        ratings.append(('working_hours', round(bkit_ratings.rating_work_hours, 1)))
+
+    thread = threading.Thread(target=uplaod_rating_thread, args=(url, ratings, headers))
+    thread.start()
+
+    url = paths.get_bkit_url() + 'assets/' + asset['asset_data']['id'] + '/review'
+
+    reviews = {
+        'reviewText': bkit_ratings.rating_compliments,
+        'reviewTextProblems': bkit_ratings.rating_problems,
+    }
+    if not (bkit_ratings.rating_compliments == '' and bkit_ratings.rating_compliments == ''):
+        thread = threading.Thread(target=uplaod_review_thread, args=(url, reviews, headers))
+        thread.start()
+
+    # the info that the user rated an item is stored in the scene
+    s = bpy.context.scene
+    s['assets rated'] = s.get('assets rated', {})
+    if bkit_ratings.rating_quality > 0.1 and bkit_ratings.rating_work_hours > 0.1:
+        s['assets rated'][asset['asset_data']['asset_base_id']] = True
+
+
+class StarRatingOperator(bpy.types.Operator):
+    """Tooltip"""
+    bl_idname = "object.blenderkit_rating"
+    bl_label = "Rate the Asset"
+
+    property_name: StringProperty(
+        name="Rating Property",
+        description="Property that is rated",
+        default="",
+    )
+
+    rating: IntProperty(name="Rating", description="rating value", default=1, min=1, max=10)
+
+    def execute(self, context):
+        asset = utils.get_active_asset()
+        props = asset.bkit_ratings
+        props[self.property_name] = self.rating
+        return {'FINISHED'}
+
+
+asset_types = (
+    ('MODEL', 'Model', 'set of objects'),
+    ('SCENE', 'Scene', 'scene'),
+    ('MATERIAL', 'Material', 'any .blend Material'),
+    ('TEXTURE', 'Texture', 'a texture, or texture set'),
+    ('BRUSH', 'Brush', 'brush, can be any type of blender brush'),
+    ('ADDON', 'Addon', 'addnon'),
+)
+
+
+class UploadRatingOperator(bpy.types.Operator):
+    """Upload rating to the web db"""
+    bl_idname = "object.blenderkit_rating_upload"
+    bl_label = "Upload the Rating"
+
+    # type of upload - model, material, textures, e.t.c.
+    asset_type: EnumProperty(
+        name="Type",
+        items=asset_types,
+        description="Type of asset",
+        default="MODEL",
+    )
+
+    # @classmethod
+    # def poll(cls, context):
+    #    return bpy.context.active_object != None and bpy.context.active_object.get('asset_id') is not None
+    def draw(self, context):
+        layout = self.layout
+        layout.label(text='Rating sent to server. Thanks for rating!')
+
+    def execute(self, context):
+        return {'FINISHED'}
+
+    def invoke(self, context, event):
+        wm = context.window_manager
+        asset = utils.get_active_asset()
+        upload_rating(asset)
+        return wm.invoke_props_dialog(self)
+
+
+def draw_rating(layout, props, prop_name, name):
+    # layout.label(name)
+
+    row = layout.row(align=True)
+
+    for a in range(0, 10):
+        if eval('props.' + prop_name) < a + 1:
+            icon = 'SOLO_OFF'
+        else:
+            icon = 'SOLO_ON'
+
+        op = row.operator('object.blenderkit_rating', icon=icon, emboss=False, text='')
+        op.property_name = prop_name
+        op.rating = a + 1
+
+
+def register_ratings():
+    pass;
+    bpy.utils.register_class(StarRatingOperator)
+    bpy.utils.register_class(UploadRatingOperator)
+
+
+def unregister_ratings():
+    pass;
+    bpy.utils.unregister_class(StarRatingOperator)
+    bpy.utils.unregister_class(UploadRatingOperator)