示例#1
0
from parse import Parse
from features import Features
from classification import Classification

if __name__ == '__main__':
    if len(sys.argv) > 1:
        print 'Parsing...',
        sys.stdout.flush()
        p = Parse(sys.argv[1])
        p.compute_fqdn()
        print 'DONE'

        print 'Computing features (Can take some time because of whois queries)...',
        sys.stdout.flush()
        features = Features(p)
        features.compute()
        print 'DONE'

        print 'Classification...',
        sys.stdout.flush()
        classification = Classification(features, p)
        classification.compute()
        print 'DONE'

        print 'Launching webserver...',
        sys.stdout.flush()
        flask_app = Flask('caphaw-dns-classifier')
        print 'DONE'

        @flask_app.route('/')
        def index():