def setUp(self): # create & init objects self.homePage_one = homeLink(HOMEPAGE_ONE) self.homePage_two = homeLink(HOMEPAGE_TWO) self.subPage = pageLink(SUBPAGE, SUBTAGS) self.stData = storeData()
class storeData: def __init__(self): self.webBase = {} def addPage(self, pageURL, pageTags): try: page = pageURL.split(pageURL.split(".com/")[1])[0] except Exception, e: page = pageURL #page = pageURL.split(pageURL.split(".com/")[1])[0] # if homeLink already in webBase, add new page if page in self.webBase: self.webBase[page].addPage(pageURL, pageTags) return True # create homeLink newLink = homeLink(page) self.webBase[page] = newLink # if the pageURL is not a homeLink, add pageLink if page != pageURL: self.webBase[page].addPage(pageURL, pageTags) return True
import sys from pageLink import pageLink from homeLink import homeLink from storeData import storeData from storeData import storeData stData = storeData() homeL = homeLink("http://www.google.com") pageL = pageLink("http://www.google.com/avengers", ['python', 'C++']) def main(): stData.addPage("http://www.google.com/avengers", ['jello', 'python']) print stData.getSearchData() homeL = stData.getLink("http://www.google.com/") print homeL.getPageLinks() pass if __name__ == '__main__': main()