Pokefusion. - Printable Version

Python Code
Code:
import urllib

import os

import re

import time

class Pokefusion:

    def __init__(self):

        self.__outputDir = 'PokeFusion'

        self.__mainURL = 'http://alexonsager.net/pokemon/index.php'

        self.__fusionURL = 'http://alexonsager.net/pokemon/images/fused/%d.%d.png'

    def GetMainPage(self):

        webPagePtr = urllib.urlopen(self.__mainURL)

        webPageText = webPagePtr.read()

        webPagePtr.close()

        return webPageText

    def GetPokeDict(self):

        webPageText = self.GetMainPage()

        pokeList = re.findall(r'([a-zA-Z\(\)]*?) \(\#(\d+?)\)', webPageText)

        returnDictionary = {}

        for eachPokemon in pokeList:

            returnDictionary[eachPokemon[1]] = eachPokemon[0]

        return returnDictionary

    def GetPokeFiles(self):

        pokeDict = self.GetPokeDict()

        numberPokemon = len(pokeDict)

        for pokemonNumber in range(1, numberPokemon + 1):

            for subPokemonNumber in range(1, numberPokemon + 1):

                outputLocation = os.path.join(self.__outputDir, pokeDict[str(pokemonNumber)])

                fileName = pokeDict[str(subPokemonNumber)] + '.png'

                fullLocation = os.path.join(outputLocation, fileName)

                print(fullLocation)                    

                if os.access(fullLocation, os.F_OK):

                    continue

                else:                

                    fusedImage = self.GetPokeImage(pokemonNumber, subPokemonNumber)

                    self.CreateDirectory(outputLocation)

                    self.OutputFile(fullLocation, fusedImage)

    def OutputFile(self, fullLocation, fusedImage):

        pokeImage = open(fullLocation, 'wb')

        pokeImage.write(fusedImage)

        pokeImage.close()

    def CreateDirectory(self, directory):

        try:

            os.makedirs(directory)

        except:

            pass

    def GetPokeImage(self, pokeNumber, subPokeNumber):

        pokeURL = self.__fusionURL % (pokeNumber, subPokeNumber)

        pokePtr = urllib.urlopen(pokeURL)

        pokeImage = pokePtr.read()

        pokePtr.close()

        return pokeImage

newPokemon = Pokefusion()

newPokemon.GetPokeFiles()