addon_updater_ops.py

import bpy
from .addon_updater import Updater as updater
from bpy.app.handlers import persistent
import os

addonname = "crowdmaster"  # more consistent than using updater.addon.lower()

# -----------------------------------------------------------------------------
# Example operators utilizing Updater
# -----------------------------------------------------------------------------

# simple popup for prompting checking for update & allowing to install if available
class addon_updater_install_popup(bpy.types.Operator):
    """Check and install update if available"""
    bl_label = "Update {x} addon".format(x=updater.addon)
    bl_idname = addonname+".updater_install_popup"
    bl_description = "Popup menu to check and display current updates available"

    def invoke(self, context, event):
        return context.window_manager.invoke_props_dialog(self)

    def draw(self, context):
        layout = self.layout
        if updater.update_ready == True:
            layout.label("Update ready! Press OK to install v"+str(updater.update_version))
            # could offer to remove popups here, but window will not close or redraw
            # so may be confusing to the user/look like a bug
            # row = layout.row()
            # row.label("Prevent future popups:")
            # row.operator(addon_updater_ignore.bl_idname,text="Ignore update")
        elif updater.update_ready == False:
            layout.label("No updates available")
            layout.label("Press okay to dismiss dialog")
            # add option to force install
        else:
            # case: updater.update_ready = None
            # we have not yet checked for the update
            layout.label("Check for update now?")

        # potentially in future, could have UI for 'check to select old version'
        # to revert back to.

    def execute(self,context):

        if updater.update_ready == True:
            res = updater.run_update(force=False, callback=post_update_callback)
            # should return 0, if not something happened
            if updater.verbose:
                if res==0: print("Updater returned successful")
                else: print("Updater returned "+str(res)+", error occured")

        elif updater.update_ready == None:
            (update_ready, version, link) = updater.check_for_update(now=True)
            
            # re-launch this dialog
            atr = addon_updater_install_popup.bl_idname.split(".")
            getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
            #bpy.ops.retopoflow.updater_install_popup('INVOKE_DEFAULT')

        else:
            print("Doing nothing, not ready for update")
        return {'FINISHED'}


# User preference check-now operator
class addon_updater_check_now(bpy.types.Operator):
    bl_label = "Check now for "+updater.addon+" update"
    bl_idname = addonname+".updater_check_now"
    bl_description = "Check now for an update to the {x} addon".format(
                                                        x=updater.addon)

    def execute(self,context):

        if updater.async_checking == True:
            # Check already happened
            # Used here to just avoid constant applying settings below
            return 

        # apply the UI settings
        settings = context.user_preferences.addons[__package__].preferences
        updater.set_check_interval(enable=settings.auto_check_update,
                    months=settings.updater_intrval_months,
                    days=settings.updater_intrval_days,
                    hours=settings.updater_intrval_hours,
                    minutes=settings.updater_intrval_minutes
                    ) # optional, if auto_check_update 
        
        # input is an optional callback function
        # this function should take a bool input, if true: update ready
        # if false, no update ready
        updater.check_for_update_now()

        return {'FINISHED'}

class addon_updater_update_now(bpy.types.Operator):
    bl_label = "Update "+updater.addon+" addon now"
    bl_idname = addonname+".updater_update_now"
    bl_description = "Update to the latest verison of the {x} addon".format(
                                                        x=updater.addon)


    def execute(self,context):

        if updater.update_ready == True:
            # if it fails, offer to open the website instead
            try:
                #updater.run_update(force=False)
                res = updater.run_update(force=False, callback=post_update_callback)
                # should return 0, if not something happened
                if updater.verbose:
                    if res==0: print("Updater returned successful")
                    else: print("Updater returned "+str(res)+", error occured")
            except:
                atr = addon_updater_install_manually.bl_idname.split(".")
                getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
        elif updater.update_ready == None:
            (update_ready, version, link) = updater.check_for_update(now=True)
            # re-launch this dialog
            atr = addon_updater_install_popup.bl_idname.split(".")
            getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
            
            #bpy.ops.retopoflow.updater_install_popup('INVOKE_DEFAULT')
        elif updater.update_ready == False:
            self.report({'INFO'}, "Nothing to update")
        else:
            self.report({'ERROR'}, "Encountered problem while trying to update")

        return {'FINISHED'}


class addon_updater_update_target(bpy.types.Operator):
    bl_label = updater.addon+" addon version target"
    bl_idname = addonname+".updater_update_target"
    bl_description = "Install a targeted version of the {x} addon".format(
                                                        x=updater.addon)

    def target_version(self, context):
        ret = []
        i=0
        #print(len(updater.tags))
        for tag in updater.tags:
            #print(tag)
            ret.append( (tag,tag,"Select to install version "+tag) )
            i+=1
            print(tag)
        return ret

    target = bpy.props.EnumProperty(
        name="Target version",
        description="Select the version to install",
        items=target_version
        )

    @classmethod
    def poll(cls, context):
        return updater.update_ready != None

    def invoke(self, context, event):
        return context.window_manager.invoke_props_dialog(self)

    def draw(self, context):
        layout = self.layout
        split = layout.split(percentage=0.66)
        subcol = split.column()
        subcol.label("Select install version")
        subcol = split.column()
        subcol.prop(self, "target", text="")


    def execute(self,context):

        res = updater.run_update(force=False,revert_tag=self.target, callback=post_update_callback)
        # should return 0, if not something happened
        if updater.verbose:
            if res==0: print("Updater returned successful")
            else: print("Updater returned "+str(res)+", error occured")
        # try:
        # 	updater.run_update(force=False,revert_tag=self.target)
        # except:
        # 	self.report({'ERROR'}, "Problem installing target version")

        return {'FINISHED'}


class addon_updater_install_manually(bpy.types.Operator):
    """As a fallback, direct the user to download the addon manually"""
    bl_label = "Install update manually"
    bl_idname = addonname+".updater_install_manually"
    bl_description = "Proceed to manually install update"

    # not an "okay" to execut, but just oeprators
    def invoke(self, context, event):
        return context.window_manager.invoke_popup(self)

    def draw(self, context):
        layout = self.layout
        # use a "failed flag"? it show this label if the case failed.
        if False:
            layout.label("There was an issue trying to auto-install")
        else:
            layout.label("Install the addon manually")
            layout.label("Press the download button below and install")
            layout.label("the zip file like a normal addon.")

        # if check hasn't happened, ie accidentally called this menu
        # allow to check here

        row = layout.row()

        if updater.update_link != None:
            row.operator("wm.url_open",text="Direct download").url=updater.update_link
        else:
            row.operator("wm.url_open",text="(failed to retreive)")
            row.enabled = False

            if updater.website != None:
                row = layout.row()
                row.label("Grab update from account")

                row.operator("wm.url_open",text="Open website").url=updater.website
            else:
                row = layout.row()

                row.label("See source website to download the update")

    def execute(self,context):

        return {'FINISHED'}


class addon_updater_updated_successful(bpy.types.Operator):
    """Addon in place, popup telling user it completed"""
    bl_label = "Success"
    bl_idname = addonname+".updater_update_successful"
    bl_description = "Update installation was successful"
    bl_options = {'REGISTER', 'UNDO'}

    # not an "okay" to execut, but just oeprators
    def invoke(self, context, event):
        return context.window_manager.invoke_props_popup(self, event)

    def draw(self, context):
        layout = self.layout
        # use a "failed flag"? it show this label if the case failed.
        saved = updater.json
        if updater.auto_reload_post_update == False:
            # tell user to restart blender
            if "just_restored" in saved and saved["just_restored"] == True:
                layout.label("Addon restored")
                layout.label("Restart blender to reload.")
                updater.json_reset_restore()
            else:
                layout.label("Addon succcessfully installed")
                layout.label("Restart blender to reload.")

        else:
            # reload addon, but still recommend they restart blender
            if "just_restored" in saved and saved["just_restored"] == True:
                layout.label("Addon restored")
                layout.label("Consider restarting blender to fully reload.")
                updater.json_reset_restore()
            else:
                layout.label("Addon succcessfully installed.")
                layout.label("Consider restarting blender to fully reload.")
    
    def execut(self, context):
        return {'FINISHED'}


class addon_updater_restore_backup(bpy.types.Operator):
    """Restore addon from backup"""
    bl_label = "Restore backup"
    bl_idname = addonname+".updater_restore_backup"
    bl_description = "Restore addon from backup"

    # not an "okay" to execut, but just oeprators
    @classmethod
    def poll(cls, context):
        try:
            return os.path.isdir(os.path.join(updater.stage_path,"backup"))
        except:
            return False
    
    def execute(self, context):
        updater.restore_backup()
        return {'FINISHED'}


class addon_updater_ignore(bpy.types.Operator):
    """Restore addon from backup"""
    bl_label = "Ignore update"
    bl_idname = addonname+".updater_ignore"
    bl_description = "Ignore update to prevent future popups"

    # not an "okay" to execut, but just oeprators
    @classmethod
    def poll(cls, context):
        if updater.update_ready == True:
            return True
        else:
            return False
    
    def execute(self, context):
        updater.ignore_update()
        return {'FINISHED'}

# -----------------------------------------------------------------------------
# Handler related, to create popups
# -----------------------------------------------------------------------------


# global vars used to prevent duplciate popup handlers
ran_autocheck_install_popup = False
ran_update_sucess_popup = False

# global var for preventing successive calls 
ran_background_check = False

@persistent
def updater_run_success_popup_handler(scene):
    global ran_update_sucess_popup
    ran_update_sucess_popup = True
    try:
        bpy.app.handlers.scene_update_post.remove(updater_run_success_popup_handler)
    except:
        pass

    atr = addon_updater_updated_successful.bl_idname.split(".")
    getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')


@persistent
def updater_run_install_popup_handler(scene):
    global ran_autocheck_install_popup
    ran_autocheck_install_popup = True
    try:
        bpy.app.handlers.scene_update_post.remove(updater_run_install_popup_handler)
    except:
        pass

    if "ignore" in updater.json and updater.json["ignore"] == True:
        return # don't do popup if ignore pressed
    atr = addon_updater_install_popup.bl_idname.split(".")
    getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
    

# passed into the updater, background thread updater
def background_update_callback(update_ready):
    global ran_autocheck_install_popup

    if update_ready != True:
        return


    
    if updater_run_install_popup_handler not in \
                bpy.app.handlers.scene_update_post and \
                ran_autocheck_install_popup==False:
        bpy.app.handlers.scene_update_post.append(updater_run_install_popup_handler)
        
        ran_autocheck_install_popup = True

# a callback for once the updater has completed
# Only makes sense to use this if "auto_reload_post_update" == False,
# ie don't auto-restart the addon
def post_update_callback():
    # this is the same code as in conditional at the end of the register function
    # ie if "auto_reload_post_update" == True, comment out this code
    if updater.verbose: print("Running post update callback")
    #bpy.app.handlers.scene_update_post.append(updater_run_success_popup_handler)

    atr = addon_updater_updated_successful.bl_idname.split(".")
    getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
    global ran_update_sucess_popup
    ran_update_sucess_popup = True
    return



# function for asynchronous background check, which *could* be called on register
def check_for_update_background(context):

    
    global ran_background_check
    if ran_background_check == True:
        # Global var ensures check only happens once
        return
    elif updater.update_ready != None or updater.async_checking == True:
        # Check already happened
        # Used here to just avoid constant applying settings below
        return 

    # apply the UI settings
    settings = context.user_preferences.addons[__package__].preferences
    updater.set_check_interval(enable=settings.auto_check_update,
                months=settings.updater_intrval_months,
                days=settings.updater_intrval_days,
                hours=settings.updater_intrval_hours,
                minutes=settings.updater_intrval_minutes
                ) # optional, if auto_check_update 
    
    # input is an optional callback function
    # this function should take a bool input, if true: update ready
    # if false, no update ready
    if updater.verbose: print("Running background check for update")
    updater.check_for_update_async(background_update_callback)
    ran_background_check = True


# a function that can be placed in front of other operators to launch when pressed
def check_for_update_nonthreaded(self, context):

    # only check if it's ready, ie after the time interval specified
    # should be the async wrapper call here

    settings = context.user_preferences.addons[__package__].preferences
    updater.set_check_interval(enable=settings.auto_check_update,
                months=settings.updater_intrval_months,
                days=settings.updater_intrval_days,
                hours=settings.updater_intrval_hours,
                minutes=settings.updater_intrval_minutes
                ) # optional, if auto_check_update 

    (update_ready, version, link) = updater.check_for_update(now=False)
    if update_ready == True:
        atr = addon_updater_install_popup.bl_idname.split(".")
        getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
        # or manually update the name of the operator bl_label
        #bpy.ops.{the updater.addon+".updater_install_popup" text}('INVOKE_DEFAULT')
    else:
        if updater.verbose: print("No update ready")
        self.report({'INFO'}, "No update ready")

# -----------------------------------------------------------------------------
# Example includable UI integrations
# -----------------------------------------------------------------------------


# UI to place e.g. at the end of a UI panel where to notify update available
def update_notice_box_ui(self, context):
    if updater.update_ready != True: return
    saved_state = updater.json
    if updater.auto_reload_post_update == False:
        if "just_updated" in saved_state and saved_state["just_updated"] == True:
            layout = self.layout
            layout.label("Restart blender", icon="ERROR")
            layout.label("to complete update")
            return

    settings = context.user_preferences.addons[__package__].preferences
    layout = self.layout
    box = layout.box()
    col = box.column(align=True)
    col.label("Update ready!",icon="ERROR")
    col.operator("wm.url_open", text="Open website").url = updater.website
    #col.operator("wm.url_open",text="Direct download").url=updater.update_link
    # atr = addon_updater_install_manually.bl_idname.split(".")
    # 			getattr(getattr(bpy.ops, atr[0]),atr[1])('INVOKE_DEFAULT')
    col.operator(addon_updater_install_manually.bl_idname, "Install manually")
    col.operator(addon_updater_update_now.bl_idname,
                    "Update now", icon="LOOP_FORWARDS") # could also do popup instead
    if "ignore" in saved_state and saved_state["ignore"] == True:
        col.label("Popups ignored")
    else:
        col.operator(addon_updater_ignore.bl_idname,icon="X")



# create a function that can be run inside of a user preferences panel for prefs UI
# place inside UI draw using: addon_updater_ops.updaterSettingsUI(self, context)
# or by: addon_updater_ops.updaterSettingsUI(context)
def update_settings_ui(self, context):
    settings = context.user_preferences.addons[__package__].preferences

    layout = self.layout
    box = layout.box()

    # auto-update settings
    box.label("Updater Settings")
    row = box.row()

    # special case to tell user to restart blender, if set that way
    if updater.auto_reload_post_update == False:
        saved_state = updater.json
        if "just_updated" in saved_state and saved_state["just_updated"] == True:
            row.label("Restart blender to complete update", icon="ERROR")
            return

    split = row.split(percentage=0.3)
    subcol = split.column()
    subcol.prop(settings, "auto_check_update")
    subcol = split.column()

    if settings.auto_check_update==False: subcol.enabled = False
    subrow = subcol.row()
    subrow.label("Interval between checks")
    subrow = subcol.row(align=True)
    checkcol = subrow.column(align=True)
    checkcol.prop(settings,"updater_intrval_months")
    checkcol = subrow.column(align=True)
    checkcol.prop(settings,"updater_intrval_days")
    checkcol = subrow.column(align=True)
    checkcol.prop(settings,"updater_intrval_hours")
    checkcol = subrow.column(align=True)
    checkcol.prop(settings,"updater_intrval_minutes")


    # checking / managing updates
    row = box.row()
    col = row.column()
    movemosue = False
    if updater.update_ready == None and updater.async_checking == False:
        col.scale_y = 2
        col.operator(addon_updater_check_now.bl_idname)
    elif updater.update_ready == None: # async is running
        col.scale_y = 2
        col.enabled = False
        col.operator(addon_updater_check_now.bl_idname, "Checking for update....")
        movemosue = True # tell user to move mouse, trigger re-draw on background check
    elif updater.update_ready == True and updater.update_version != updater.current_version:
        col.scale_y = 2
        col.operator(addon_updater_update_now.bl_idname,
                    "Update now to "+str(updater.update_version))
    else:
        subcol = col.row(align=True)
        subcol.scale_y = 1
        split = subcol.split(align=True)
        split.enabled = False
        split.scale_y = 2
        split.operator(addon_updater_check_now.bl_idname,
                        "Addon is up to date")
        split = subcol.split(align=True)
        split.scale_y = 2
        split.operator(addon_updater_check_now.bl_idname,
                        text = "", icon="FILE_REFRESH")


    col = row.column(align=True)
    col.operator(addon_updater_update_target.bl_idname,
                "Reinstall / install old verison")
    lastdate = "none found"
    backuppath = os.path.join(updater.stage_path,"backup")
    if "backup_date" in updater.json and os.path.isdir(backuppath):
        if updater.json["backup_date"] == "":
            lastdate = "Date not found"
        else:
            lastdate = updater.json["backup_date"]
    backuptext = "Restore addon backup ({x})".format(x=lastdate)
    col.operator(addon_updater_restore_backup.bl_idname, backuptext)

    row = box.row()
    lastcheck = updater.json["last_check"]
    if movemosue == True:
        row.label("Move mouse if button doesn't update")
    elif lastcheck != "" and lastcheck != None:
        lastcheck = lastcheck[0: lastcheck.index(".") ]
        row.label("Last update check: " + lastcheck)
    else:
        row.label("Last update check: None")




# -----------------------------------------------------------------------------
# Register, should be run in the register module itself
# -----------------------------------------------------------------------------

# registering the operators in this module
def register(bl_info):

    print("Running updater reg")

    updater.user = "johnroper100"
    updater.repo = "CrowdMaster"
    updater.addon =  "CrowdMaster" # optional, default gets from __package__ name
    updater.website = "https://github.com/johnroper100/CrowdMaster" # optional
    updater.use_releases = True # ie use tags instead of releases, default True
    updater.current_version = bl_info["version"]

    # Below: ie  make users restart blender to load
    # instead of auto-reload which can cause issues
    updater.auto_reload_post_update = False # False is the default value

    updater.verbose = False
    updater.backup_current = True # True by default
    updater.fake_install = False # Set to true to test callback/reloading

    
    # best practice to ensure failing doesn't create issue with register,
    # always enclose in try/except in production
    
    # try:
    #	updater.check_for_update_async()
    # except:
    # 	print("Failed to check for update")


    bpy.utils.register_class(addon_updater_install_popup)
    bpy.utils.register_class(addon_updater_check_now)
    bpy.utils.register_class(addon_updater_update_now)
    bpy.utils.register_class(addon_updater_update_target)
    bpy.utils.register_class(addon_updater_install_manually)
    bpy.utils.register_class(addon_updater_updated_successful)
    bpy.utils.register_class(addon_updater_restore_backup)
    bpy.utils.register_class(addon_updater_ignore)
    

    # special situation: we JUST updated the addon, show a popup
    # to tell the user it worked
    # shoudl be enclosed in try/catch in case other issues arise
    saved_state = updater.json
    global ran_update_sucess_popup
    if saved_state != None and "just_updated" in saved_state and saved_state["just_updated"] == True:
        updater.json_reset_postupdate() # so this only runs once
        if updater.auto_reload_post_update == False: return # no handlers in this case
        if updater_run_success_popup_handler not in bpy.app.handlers.scene_update_post and ran_update_sucess_popup==False:   
            bpy.app.handlers.scene_update_post.append(updater_run_success_popup_handler)
            ran_update_sucess_popup = True



def unregister():
    bpy.utils.unregister_class(addon_updater_install_popup)
    bpy.utils.unregister_class(addon_updater_check_now)
    bpy.utils.unregister_class(addon_updater_update_now)
    bpy.utils.unregister_class(addon_updater_update_target)
    bpy.utils.unregister_class(addon_updater_install_manually)
    bpy.utils.unregister_class(addon_updater_updated_successful)
    bpy.utils.unregister_class(addon_updater_restore_backup)
    bpy.utils.unregister_class(addon_updater_ignore)

    global ran_autocheck_install_popup
    global ran_update_sucess_popup

    ran_autocheck_install_popup = False
    ran_update_sucess_popup = False

    global ran_background_check
    ran_background_check = False