コード例 #1
0
ファイル: stalk.py プロジェクト: dalimatt/Instastalk
def main(wf):
    
    # Check for an update
    if wf.update_available:
        wf.add_item('New version available',
                'Action this item to install the update',
                autocomplete='workflow:update',
                icon=ICON_INFO)
   
    # Check settings for access_token
    settings = wf.settings
    if 'primary_access_token' not in settings.keys():
        wf.add_item( title='Run the configuration to connect an Instagram account',
                     subtitle='Type: "stalk.configuration"',
                     valid=False,
                     autocomplete='.configuration'
                   )
    # Process the query, find the encoded command if one exists from autocomplete 
    #  of previously run script
    else:
        data = None
        command = None
        alt_command = False
        # Get query passed from Alfred
        if len(wf.args[0]):
            # Check for an 'alt' activated command
            if wf.args[0][0] == unichr(ALT):  # ALT command
                alt_command = True
                if len(wf.args[0]) == 1:
                    # The delete button was used to eliminate all but the ALT character
                    query = False
                else:
                    log.debug(u'Found ALT command')
                    query = wf.args[0][1:]
            else:
                query = wf.args[0]
                log.debug(u'Query: <{0}>, len(query) = {1}'.format(query, len(query)))
            # Process query, the first character may be a non-printing unicode charcter 
            #  that indicates the command the script should run
            if query:
                firstchar = query[0]
                unival = ord(firstchar)
                if unival in AlfredItems.non_printing_univals():
                    log.debug('Found special unicode! value={0}'.format(unival))
                    # We have a special character key from autocomplete of previous script
                    # Find the cached information from previously run script containing data
                    #  for each actionable alfred item
                    info = wf.cached_data('special_info', max_age=FROM_CACHE)
                    if info and len(query) == 1:
                        # The query passed from Alfred should contain a special, non-printing, 
                        #  unicode character whose ordinal is the key for the cached info dict
                        command, data = info[unival]                       
                        # Cache value to exclude it from next time script is run. Alfred will 
                        #  not rerun the script if the autocomplete value is the same as before
                        wf.cache_data('unival',unival)
                        log.debug(u'COMMAND={0}'.format(COMMANDS[command]))
                    else:
                        # More than one character in query indicates a search
                        command = SEARCH
                        query = query[1:]
                    # Initialize generators for special info
                    previous_unival = unival
                    wf.alfred_items = AlfredItems(wf, previous_unival)
                    if data:
                        # If user scrolls through alfred history the previous command and data 
                        #  may be recovered if reissued here. The previous unicode value is 
                        #  excluded from new data that is issued so there is no conflict of 
                        #  items using the same unicode value
                        old_data = {previous_unival: (command, data) }
                        wf.alfred_items.special_info.send(old_data)
                else:
                    # No special unicode character found
                    wf.alfred_items = AlfredItems(wf)
                    command = SEARCH
            else:
                # There is no query, only the ALT character
                command = FAVORITES
                wf.alfred_items = AlfredItems(wf)
        else:
            # An empty query means the script is being run for the first time
            command = FAVORITES
            # No previous unival in query
            wf.alfred_items = AlfredItems(wf)
        

            
                          
        log.debug('Final COMMAND:{0}'.format(COMMANDS[command]))    
        # Execute command
        if command is LOAD_USER:
            load_user(data['user_id'], data['username'])
        elif command is FAVORITES:
            load_favorites()
        elif command is ADD_FAV:
            add_favorite(data['user_id'], data['username'])
        elif command is REMOVE_FAV:
            remove_favorite(data['user_id'], data['username'])
            load_favorites()
        elif command is SEARCH:
            search_users(query)
        elif command is CHECK_FOLLOWS:
            check_user_follows(data['user_id'], data['username'])
        elif command is MAKE_FAVORITE:
            add_or_remove_favorite(data['user_id'], data['username'])
        elif command is GET_LIKES:
            if alt_command:
                cached_likes(data['user_id'], data['username'])
            else:
                user_id = data['user_id']
                process_alias = user_id + '.get_likes'
                args = ['/usr/bin/python', wf.workflowfile('get_recent_likes.py'),
                        user_id, wf.likesdir, wf.mediadir]
                if background.is_running(process_alias):
                    log.debug('Process `{alias}` is already running'.format(alias=process_alias))
                else:
                    background.run_in_background(process_alias, args)
                load_user(data['user_id'], data['username'])
        elif command is CACHED_LIKES:
            cached_likes(data['user_id'], data['username'])
        elif command is CHECK_FOLLOWED_BY:
            check_user_followed_by(data['user_id'], data['username'])
        elif command is RECENT_MEDIA:
            recent_media(data['user_id'], data['username'])
        else:
            # Bad command
            log.debug(u'Received unknown command=<{0}>'.format(str(command)))
        # Write the item data to a file so the item the user selected determines how the script is run next
        wf.alfred_items.special_info.close()
    # Send XML formatted result to Alred
    wf.send_feedback()
コード例 #2
0
ファイル: configure.py プロジェクト: dalimatt/Instastalk
def main(wf):
    """Configure options and access tokens for the Instastalk workflow"""
    
    
    # Get query passed from alfred
    query = wf.args[0]
    # An empty query means the script is being run for the first time
    if query == '':
        wf.alfred_items = AlfredItems(wf, name='configure_special_info')
        # Check if the settings file exists
        settings = wf.settings
        if not settings:
            configure_initial()
        else:
            configure_main()
    
    # Otherwise find the previously selected item from query
    else:
        firstchar = query[0]
        # Default values for command and data
        command = None
        data = None
        unival = ord(firstchar)
        if unival in AlfredItems.non_printing_univals():
            # Found a special character key from autocomplete of previous script
            info = wf.cached_data('configure_special_info', max_age=FROM_CACHE)
            # Get the selected command and associated data from previously run script
            #  'info' contains a list of all options to select from previous script
            #  'unival' is the ordinal of the unicode character passed from autocompleting
            #     the actionied item from previous script
            command, data = info[unival]
            # # Cache value to exclude it from next time script is run. Alfred will not rerun the script if the autocomplete value is the same as before
            # wf.cache_data('unival',unival)
            log.debug('Received special unicode value, command: {command}'.format(command=COMMANDS[command]))
            
            # Initialize generators for special info
            # First find previously chosen unival to exclude it from new range
            previous_unival = unival
            wf.alfred_items = AlfredItems(wf, previous_unival, name='configure_special_info')
            # Send previous info to alfred_items in case user accesses the previous information in alfred by scrolling
            wf.alfred_items.special_info.send({previous_unival: (command, data)})
        else:
            wf.alfred_items = AlfredItems(wf, name='configure_special_info')

        # Execute navigtion function appropriate to the command
        if command == CHANGE_PRIMARY:
            change_primary()
        elif command == MAKE_PRIMARY:
            new_primary_token = data
            exchange_primary_and_secondary_tokens(new_primary_token)
            configure_main()
        elif command == NEW_ACCOUNT:
            associate_instagram_account('secondary')
        elif command == REMOVE_ACCOUNT:
            remove_account()
        elif command == NEW_PRIMARY_FROM_SECONDARY:
            new_primary_from_secondary()
        elif command == REMOVE_PRIMARY:
            remove_primary()
        elif command == REMOVE_SECONDARY:
            secondary_token = data
            remove_secondary(secondary_token)
        elif command == NEW_PRIMARY:
            secondary_token = data
            new_primary(secondary_token)
    # Write the item data to a file so the item the user selected determines how the script is run next
    wf.alfred_items.special_info.close()
    # Send XML formatted result to Alred
    wf.send_feedback()