def main(self):
        """Get the objects from the parser (or cache), 
        search through them, and print the results """

        args = self.parser.parse_args()
        # Little magic/abomination to avoid a if-else
        # funct == search_all or search_any
        funct = getattr(self, "search_" + args.search_funct) 
        cache_file_path = self.get_cache_file_path()

        # If cache is missing or user asked to force an update
        # then scrape the wiki
        if not os.path.isfile( cache_file_path ) or args.update:
            self.data_values = WikiTableScraper().scrapeTables()
            pickle.dump( self.data_values, open( cache_file_path, "wb" ) )
        # else load the minecraft objects from cache
        else:
            self.data_values = pickle.load( open( cache_file_path, "rb" ) )

        # call the search function (will return list of ids)
        for k in funct(args.words):
            print k.ljust(4), # Don't print a newline
            # Don't print the name if user requested only id's
            if not args.only_id:
                print self.data_values[k],
            print # print the newline now
예제 #2
0
#! /usr/bin/env python2

from TableScraper import WikiTableScraper

scraper = WikiTableScraper()
ids = scraper.scrapeTables()

for k,v in sorted(ids.iteritems()):
    print k.ljust(4), v

class Controller(object):
    """This class fetches the minecraft objects from WikiTableScraper, 
    and searches for item names that match search words. 
    Uses a cache through the pickle module to speed up subsequent searches.
    """
    def __init__(self):
        """Setup the argument parser"""
        # Dict of minecraft object in form of "dict[id] = name"
        self.data_values = dict()
        self.parser = self.setup_parser()

    def main(self):
        """Get the objects from the parser (or cache), 
        search through them, and print the results """

        args = self.parser.parse_args()
        # Little magic/abomination to avoid a if-else
        # funct == search_all or search_any
        funct = getattr(self, "search_" + args.search_funct) 
        cache_file_path = self.get_cache_file_path()

        # If cache is missing or user asked to force an update
        # then scrape the wiki
        if not os.path.isfile( cache_file_path ) or args.update:
            self.data_values = WikiTableScraper().scrapeTables()
            pickle.dump( self.data_values, open( cache_file_path, "wb" ) )
        # else load the minecraft objects from cache
        else:
            self.data_values = pickle.load( open( cache_file_path, "rb" ) )

        # call the search function (will return list of ids)
        for k in funct(args.words):
            print k.ljust(4), # Don't print a newline
            # Don't print the name if user requested only id's
            if not args.only_id:
                print self.data_values[k],
            print # print the newline now

    def get_cache_file_path(self):
        """Returns the path of the cache file.
        Side-Effect: creates ~/.cache or ~/.cache/CACHE_DIR"""
        home_path = os.path.expanduser("~")
        # path to the programs cache directory
        full_cache_dir =  os.path.join(home_path, ".cache", CACHE_DIR)

        if not os.path.exists( full_cache_dir ):
            os.makedirs( full_cache_dir )
        
        return os.path.join( full_cache_dir, FILE_NAME )

    def search_all(self, word_list):
        """Returns list of keys of minecraft objects whose names contain
        ALL the search words """
        return [k for k,v in self.data_values.iteritems() 
                if all(w.lower() in v.lower() for w in word_list)]

    def search_any(self, word_list):
        """Returns list of keys of minecraft objects whose names contain
        ANY the search words """
        # Same as search_all except uses the built-in any()
        return [k for k,v in self.data_values.iteritems() 
                if any(w.lower() in v.lower() for w in word_list)]

    def setup_parser(self):
        """Returns an argument parser already initialized"""
        parser = argparse.ArgumentParser(description=DESCRIPTION)
        parser.add_argument('words', metavar='W', nargs='+', help=POSITIONAL_HELP)
        parser.add_argument('-a','--any', dest="search_funct", action="store_const", 
                            const='any', default='all', help=SEARCH_HELP)
        parser.add_argument('-o','--only-id', action='store_true', help=ID_HELP)
        parser.add_argument('-u', '--update', action='store_true', help=UPDATE_HELP)
        return parser