Exemplo n.º 1
0
def main(download, eclassify, classify, version, help):
    if (help):
        print(help_message)
        sys.exit(0)
    else:
        if (version):
            print('koolsla' +  ' ' + spampy_version)
        else:
            if (download):
                dataset_downloader.download_enron_dataset()
            elif (eclassify):
                email = click.prompt('Raw email', type=str)
                is_spam = spam_classifier.classify_email_with_enron(email)
                print(True if is_spam == 1 else False)
            elif (classify):
                email = click.prompt('Raw email', type=str)
                is_spam = spam_classifier.classify_email(email)
                print(True if is_spam == 1 else False)
Exemplo n.º 2
0
def main(download, eclassify, classify, version, help):
    if help:
        print(help_message)
        sys.exit(0)
    else:
        if version:
            print("koolsla" + " " + spampy_version)
        else:
            if download:
                dataset_downloader.download_enron_dataset()
            elif eclassify:
                email = click.prompt("Raw email", type=str)
                is_spam = spam_classifier.classify_email_with_enron(email)
                print(True if is_spam == 1 else False)
            elif classify:
                email = click.prompt("Raw email", type=str)
                is_spam = spam_classifier.classify_email(email)
                print(True if is_spam == 1 else False)
 def test_download_enron_dataset(self):
     system = platform.system()
     if system == "Linux" or system == "Darwin":
         dataset_downloader.download_enron_dataset()
         self.assertTrue(os.path.exists("spampy/datasets/enron"))
Exemplo n.º 4
0
 def test_download_enron_dataset(self):
     dataset_downloader.download_enron_dataset()
     self.assertTrue(os.path.exists('spampy/datasets/enron'))