def createStyletree(): #dic dic = Dic() dic.fromfile() sourceparser = SourceParser(dic) centre = CentreDicBuilder() paths = centre.buildPath() sourceparser.setPagenum(len(paths)) for path in paths: c = open(path).read() sourceparser.setSource(c) sourceparser.parse() sourceparser.styletree.cal() res = sourceparser.styletree.show() return (dic, sourceparser.styletree)
class CentreDicBuilder: def __init__(self): self.dic = Dic() self.feature_extrator = FeatureExtrator() self.config = Config() def buildPath(self): basepath = self.config.get('path', 'source') return [basepath + path for path in os.listdir(basepath)] def run(self): #log open('CentreDicBuilder.log', 'w').write(' ') fi = open('CentreDicBuilder.log', 'a') for path in self.buildPath(): c = open(path).read() self.feature_extrator.setSource(c) features = self.feature_extrator.getFeatures() for f in features: print f self.dic.add(f) fi.write(f + '\n') self.dic.done() self._tofile() fi.close() def _tofile(self, filename='features.dic'): self.dic.tofile(filename)
def __init__(self): self.dic = Dic() self.feature_extrator = FeatureExtrator() self.config = Config()