operators.py

import bpy
import os
from bpy.types import Panel, UIList
from bpy.props import IntProperty, CollectionProperty #, StringProperty

from .func import *
###--- UI List items

# ui list item actions
class Uilist_actions(bpy.types.Operator):
    bl_idname = "sniptool.list_action"
    bl_label = "List Action"

    action = bpy.props.EnumProperty(
        items=(
            ('UP', "Up", ""),
            ('DOWN', "Down", ""),
            ('REMOVE', "Remove", ""),
            ('ADD', "Add", ""),
        )
    )

    def invoke(self, context, event):
        scn = context.scene
        idx = scn.sniptool_index

        try:
            item = scn.sniptool[idx]
        except IndexError:
            pass

        else:
            if self.action == 'DOWN' and idx < len(scn.sniptool) - 1:
                item_next = scn.sniptool[idx+1].name
                scn.sniptool_index += 1
                info = 'Item %d selected' % (scn.sniptool_index + 1)
                self.report({'INFO'}, info)

            elif self.action == 'UP' and idx >= 1:
                item_prev = scn.sniptool[idx-1].name
                scn.sniptool_index -= 1
                info = 'Item %d selected' % (scn.sniptool_index + 1)
                self.report({'INFO'}, info)

            elif self.action == 'REMOVE':
                info = 'Item %s removed from list' % (scn.sniptool[scn.sniptool_index].name)
                scn.sniptool_index -= 1
                self.report({'INFO'}, info)
                scn.sniptool.remove(idx)

        if self.action == 'ADD':
            ###---mypart
            snipname = clipit(context)
            if snipname:
                item = scn.sniptool.add()
                item.id = len(scn.sniptool)

                item.name = os.path.splitext(snipname)[0]

                scn.sniptool_index = (len(scn.sniptool)-1)
                info = '%s added to list' % (item.name)
                self.report({'INFO'}, info)
            else:
                self.report({'warning'}, 'nothing selected')

        return {"FINISHED"}




# -------------------------------------------------------------------
# draw
# -------------------------------------------------------------------

# sniptool list
class UL_items(UIList):

    def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
        #split = layout.split(0.3)
        ##add to draw index (useless)
        #split.label("%d" % (index))
        #split.prop(item, "name", text="", emboss=False, translate=False, icon='WORDWRAP_ON')
        ##delete icon to remove sheets icons (alsouseless)
        layout.prop(item, "name", text="", emboss=False, translate=False, icon='WORDWRAP_ON')

    def invoke(self, context, event):
        pass

# draw the panel
class UIListPanelExample(Panel):
    """Creates a Panel in the Object properties window"""
    bl_idname = 'OBJECT_PT_my_panel'
    bl_space_type = "TEXT_EDITOR"
    bl_region_type = "UI"
    bl_label = "Snippets List"

    bpy.types.Scene.new_snippets_name = bpy.props.StringProperty(description='name that snippets will take, name will be generated')

    def draw(self, context):
        layout = self.layout
        scn = bpy.context.scene

        rows = 2
        row = layout.row()
        row.operator("sniptool.reload_list", icon="FILE_REFRESH")
        row = layout.row()
        row.operator("sniptool.template_insert", icon="LIBRARY_DATA_DIRECT")#LIBRARY_DATA_DIRECT RIGHTARROW FORWARD
        row = layout.row()
        row.template_list("UL_items", "", scn, "sniptool", scn, "sniptool_index", rows=rows)

        col = row.column(align=True)
        # col.operator("sniptool.list_action", icon='ZOOMIN', text="").action = 'ADD'
        # col.operator("sniptool.list_action", icon='ZOOMOUT', text="").action = 'REMOVE'
        col.separator()
        col.operator("sniptool.list_action", icon='TRIA_UP', text="").action = 'UP'
        col.operator("sniptool.list_action", icon='TRIA_DOWN', text="").action = 'DOWN'

        row = layout.row()
        col = row.column(align=True)
        col.separator()
        col.prop(context.scene, 'new_snippets_name', text='snippets name')
        col.operator("sniptool.save_snippet", icon="SAVE_COPY")
        col.operator("sniptool.open_snippet_folder", icon="FILE_FOLDER")
        # col.separator()



class Uilist_saveSnippet(bpy.types.Operator):
    bl_idname = "sniptool.save_snippet"
    bl_label = "save snippet"
    bl_description = "save selection to a file named after this"

    def execute(self, context):
        scn = context.scene
        library = locateLibrary()
        if library:
            snipname = clipit(context)
            if snipname:
                item = scn.sniptool.add()
                item.id = len(scn.sniptool)

                item.name = os.path.splitext(snipname)[0]

                scn.sniptool_index = (len(scn.sniptool)-1)
                info = '%s added to list' % (item.name)
                self.report({'INFO'}, info)
            else:
                self.report({'warning'}, 'nothing selected')

            ###print all snipsauce :
            #for i in scn.sniptool:
            #    print (i.name, i.id)

        else:
            pathErrorMsg = locateLibrary(True) + ' not found or inaccessible'
            self.report({'ERROR'}, pathErrorMsg)
        return{'FINISHED'}


# insert button
class Uilist_insertTemplate(bpy.types.Operator):
    bl_idname = "sniptool.template_insert"
    bl_label = "insert List Item"
    bl_description = "insert Item in textBlock"

    def execute(self, context):
        scn = context.scene
        if locateLibrary():
            text = getattr(bpy.context.space_data, "text", None)
            if text:
                pass
                #print(text.name)
            else:
                pass
                #print('no text data')

            #context override for the ops.text.insert() function
            override = {'window': context.window,
                        'area'  : context.area,
                        'region': context.region,
                        'space': context.space_data,
                        'edit_text' : text
                        }
            snip = scn.sniptool[scn.sniptool_index].name

            Loaded_text = load_text(get_snippet(snip))
            #get character position in text
            charPos = text.current_character
            #print ('charPos', charPos)

            #indentedText = Loaded_text
            if Loaded_text:
                FormattedText = Loaded_text
                #indent text lines exept first (already at cursor position)
                if charPos > 0:
                    textLines = Loaded_text.split('\n')
                    if not len(textLines) == 1:
                        #print("indent subsequent lines")
                        indentedLines = []
                        indentedLines.append(textLines[0])
                        for line in textLines[1:]:
                            indentedLines.append(' '*charPos + line)

                        FormattedText = '\n'.join(indentedLines)

                # print(FormattedText)
                insert_template(override, FormattedText)

            else:
                print('Fail to load snippet !')
        else:
            pathErrorMsg = locateLibrary(True) + ' not found or inaccessible'
            self.report({'ERROR'}, pathErrorMsg)
        return{'FINISHED'}


# relaod button
class Uilist_reloadItems(bpy.types.Operator):
    bl_idname = "sniptool.reload_list"
    bl_label = "Reload List"
    bl_description = "Reload all items in the list"

    def execute(self, context):
        scn = context.scene
        lst = scn.sniptool
        current_index = scn.sniptool_index
        library = locateLibrary()
        if library:
            allsnip = reload_folder(library)

            if len(lst) > 0:#remove all item in list
                # reverse range to remove last item first
                for i in range(len(lst)-1,-1,-1):
                    scn.sniptool.remove(i)
                #self.report({'INFO'}, "All items removed")

            for snipname in allsnip:#populate list
                item = scn.sniptool.add()
                item.id = len(scn.sniptool)
                item.name = snipname
                scn.sniptool_index = (len(scn.sniptool)-1)
                #info = '%s added to list' % (item.name)

            # else:
            #     self.report({'INFO'}, "Nothing to add")
        else:
            pathErrorMsg = locateLibrary(True) + ' not found or inaccessible'
            self.report({'ERROR'}, pathErrorMsg)

        return{'FINISHED'}


class OpenSnippetsFolder(bpy.types.Operator):
    bl_idname = "sniptool.open_snippet_folder"
    bl_label = "open library folder"
    bl_description = "open snippets folder location"

    def execute(self, context):
        scn = context.scene
        library = locateLibrary()
        if library:
            #open folder
            openFolder(library)
        else:
            pathErrorMsg = locateLibrary(True) + ' not found or inaccessible'
            self.report({'ERROR'}, pathErrorMsg)
        return{'FINISHED'}

# Create sniptool property group
class sniptoolProp(bpy.types.PropertyGroup):
    '''name = StringProperty() '''
    id = IntProperty()