node_Tools.py

import bpy
import bmesh
import mathutils
from mathutils import Matrix
from util import *
from node_s import *
import webbrowser
import os
import urllib
import urllib.request
import traceback
from zipfile import ZipFile


def get_update_info():
    # needed to show current version
    sv_master_path = os.path.normpath(os.path.dirname(__file__))
    bl_addons_path = os.path.dirname(sv_master_path)
    sv_version = os.path.join(sv_master_path, 'version')
    print(sv_master_path)
    return sv_master_path, bl_addons_path, sv_version

sv_master_path, bl_addons_path, sv_version = get_update_info()

with open(sv_version) as sv_local_file:
    svversion_local = next(sv_local_file).strip()


class SverchokUpdateAll(bpy.types.Operator):

    """Sverchok update all"""
    bl_idname = "node.sverchok_update_all"
    bl_label = "Sverchok update all"
    bl_options = {'REGISTER', 'UNDO'}

    def execute(self, context):
        makeTreeUpdate2()
        speedUpdate()
        return {'FINISHED'}


class SverchokPurgeCache(bpy.types.Operator):

    """Sverchok purge cache"""
    bl_idname = "node.sverchok_purge_cache"
    bl_label = "Sverchok purge cache"
    bl_options = {'REGISTER', 'UNDO'}

    def execute(self, context):
        print(bpy.context.space_data.node_tree.name)
        return {'FINISHED'}


class SverchokHome(bpy.types.Operator):

    """Sverchok Home"""
    bl_idname = "node.sverchok_home"
    bl_label = "Sverchok go home"
    bl_options = {'REGISTER', 'UNDO'}

    def execute(self, context):
        page = 'http://nikitron.cc.ua/blend_scripts.html'
        if context.scene.use_webbrowser:
            try:
                webbrowser.open_new_tab(page)
            except:
                self.report({'WARNING'}, "Error in opening the page %s." % (page))
        return {'FINISHED'}


class SverchokUpdateAddon(bpy.types.Operator):

    """ Sverchok update addon on the fly. Press F8 to reload addon """
    bl_idname = "node.sverchok_update_addon"
    bl_label = "Sverchok update addon in linux"
    bl_options = {'REGISTER', 'UNDO'}

    #version = bpy.props.StringProperty(name='Your Blender version is', default='2.70')

    def execute(self, context):
        sv_master_path, bl_addons_path, sv_version = get_update_info()
        report = self.report

        # read online, cancel if problems.
        try:
            # https access may not be not available on all blender compiles
            url = 'https://raw.githubusercontent.com/nortikin/sverchok/master/version'
            version_url = urllib.request.urlopen(url).read().strip()

        except urllib.error.URLError:
            traceback.print_exc()
            report({'INFO'}, "Unable to contact github, or SSL not compiled.")
            return {'CANCELLED'}

        # with version online, check current version
        try:
            os.curdir = sv_master_path
            os.chdir(os.curdir)
            with open(sv_version) as sv_local_file:
                version_local = next(sv_local_file).strip()
        except:
            report({'INFO'}, "Failed to read local version")
            return {'CANCELLED'}

        # with both, we compare
        if version_local == version_url:
            report({'INFO'}, "You already have latest version of Sverchok!")
            return {'CANCELLED'}

        # reaches here if they are not the same:
        os.curdir = bl_addons_path
        os.chdir(os.curdir)
        try:
            url = 'https://github.com/nortikin/sverchok/archive/master.zip'
            master_dest = os.path.normpath(os.path.join(os.curdir, 'master.zip'))
            sv_archive = urllib.request.urlretrieve(url, master_dest)

            ZipFile(sv_archive[0]).extractall(path=os.curdir)
            os.remove(sv_archive[0])

        except:
            report({'ERROR'}, "cannot unzip archive somehow, removing zip")
            os.system('rm master.zip')
            report({'ERROR'}, "zip removed")
            return {'CANCELLED'}

        report({'INFO'}, "Unzipped, reload addons with F8 button")
        return {'FINISHED'}

    # def invoke(self, context, event):
        #wm = context.window_manager
        #wm.invoke_props_dialog(self, 250)
        # return {'RUNNING_MODAL'}


class SverchokToolsMenu(bpy.types.Panel):
    bl_idname = "Sverchok_tools_menu"
    bl_label = "Sverchok " + svversion_local
    bl_space_type = 'NODE_EDITOR'
    bl_region_type = 'UI'
    bl_category = 'Sverchok'
    use_pin = True

    @classmethod
    def poll(cls, context):
        try:
            return context.space_data.node_tree.bl_idname == 'SverchCustomTreeType'
        except:
            return False

    def draw(self, context):
        layout = self.layout
        # layout.scale_y=1.1
        layout.active = True
        col = layout.column()
        col.scale_y = 3.0

        col.operator(SverchokUpdateAll.bl_idname, text="UPDATE")

        box = layout.box()
        #box.label(text="Layout manager")
        little_width = 0.12
        col = box.column(align=True)
        row = col.row(align=True)
        row.label(text='Layout')
        col1 = row.column(align=True)
        col1.scale_x = little_width
        col1.label(icon='RESTRICT_VIEW_OFF', text=' ')
        # row.label(text='Bake')
        col2 = row.column(align=True)
        col2.scale_x = little_width
        col2.label(icon='ANIM', text=' ')
        col2.icon

        ng = bpy.data.node_groups

        for name, tree in ng.items():
            if tree.bl_idname == 'SverchCustomTreeType':

                row = col.row(align=True)

                if name == context.space_data.node_tree.name:
                    row.label(text=name, icon='LINK')
                else:
                    row.label(text=name)

                split = row.column(align=True)
                split.scale_x = little_width
                if tree.sv_show:
                    split.prop(tree, 'sv_show', icon='UNLOCKED', text=' ')
                else:
                    split.prop(tree, 'sv_show', icon='LOCKED', text=' ')
                split = row.column(align=True)
                split.scale_x = little_width
                if tree.sv_animate:
                    split.prop(tree, 'sv_animate', icon='UNLOCKED', text=' ')
                else:
                    split.prop(tree, 'sv_animate', icon='LOCKED', text=' ')

        layout.column().operator(SverchokUpdateAddon.bl_idname,
                                 text='Upgrade Sverchok addon')
        #       row.prop(tree, 'sv_bake',text=' ')

        #box = layout.box()
        #col = box.column(align=True)
        #col.label(text="Sverchok v_0.2.8")
        #col.label(text='layout: '+context.space_data.node_tree.name)
        #row = col.row(align=True)
        #row.operator('wm.url_open', text='Help!').url = 'http://wiki.blender.org/index.php/Extensions:2.6/Py/Scripts/Nodes/Sverchok'
        #row.operator('wm.url_open', text='Home!').url = 'http://nikitron.cc.ua/blend_scripts.html'
        #layout.operator(SverchokHome.bl_idname, text="WWW: Go home")
        #row = col.row(align=True)
        #row.operator('wm.url_open', text='FBack').url = 'http://www.blenderartists.org/forum/showthread.php?272679-Addon-WIP-Sverchok-parametric-tool-for-architects/'
        #row.operator('wm.url_open', text='Bugtr').url = 'https://docs.google.com/forms/d/1L2BIpDhjMgQEbVAc7pEq93432Qanu8UPbINhzJ5SryI/viewform'


class ToolsNode(Node, SverchCustomTreeNode):

    ''' Tools for different purposes '''
    bl_idname = 'ToolsNode'
    bl_label = 'Tools node'
    bl_icon = 'OUTLINER_OB_EMPTY'
    #bl_height_default = 110
    #bl_width_min = 20
    #color = (1,1,1)
    color_ = bpy.types.ColorRamp

    def init(self, context):
        pass

    def draw_buttons(self, context, layout):
        col = layout.column()
        col.scale_y = 15
        col.template_color_picker
        col.operator(SverchokUpdateAll.bl_idname, text="UPDATE")
        #box = layout.box()

        #col = box.column(align=True)
        #col.template_node_socket(color=(0.0, 0.9, 0.7, 1.0))
        #col.operator('wm.url_open', text='Help!').url = 'http://wiki.blender.org/index.php/Extensions:2.6/Py/Scripts/Nodes/Sverchok'
        #col.operator('wm.url_open', text='Home!').url = 'http://nikitron.cc.ua/blend_scripts.html'
        #layout.operator(SverchokHome.bl_idname, text="WWW: Go home")
        #col.operator('wm.url_open', text='FBack').url = 'http://www.blenderartists.org/forum/showthread.php?272679-Addon-WIP-Sverchok-parametric-tool-for-architects/'
        #col.operator('wm.url_open', text='Bugtr').url = 'https://docs.google.com/forms/d/1L2BIpDhjMgQEbVAc7pEq93432Qanu8UPbINhzJ5SryI/viewform'

        lennon = len(bpy.data.node_groups[self.id_data.name].nodes)
        group = self.id_data.name
        tex = str(lennon) + ' | ' + str(group)
        layout.label(text=tex)
        #layout.template_color_ramp(self, 'color_', expand=True)

    def update(self):
        self.use_custom_color = True
        self.color = (1.0, 0.0, 0.0)

    def update_socket(self, context):
        pass


def register():
    bpy.utils.register_class(SverchokUpdateAll)
    bpy.utils.register_class(SverchokUpdateAddon)
    bpy.utils.register_class(SverchokPurgeCache)
    bpy.utils.register_class(SverchokHome)
    bpy.utils.register_class(SverchokToolsMenu)
    bpy.utils.register_class(ToolsNode)


def unregister():
    bpy.utils.unregister_class(ToolsNode)
    bpy.utils.unregister_class(SverchokToolsMenu)
    bpy.utils.unregister_class(SverchokHome)
    bpy.utils.unregister_class(SverchokPurgeCache)
    bpy.utils.unregister_class(SverchokUpdateAddon)
    bpy.utils.unregister_class(SverchokUpdateAll)

if __name__ == "__main__":
    register()