From 9588ab3caa7b7bbf69f56ae9f5cbe041f8552e28 Mon Sep 17 00:00:00 2001
From: Kevin Chung <kchung@nyu.edu>
Date: Mon, 27 Apr 2020 02:36:59 -0400
Subject: [PATCH] Fix email issue and remove file extensions list

---
 populate.py | 137 +---------------------------------------------------
 1 file changed, 2 insertions(+), 135 deletions(-)

diff --git a/populate.py b/populate.py
index a40549c0..033d69fc 100644
--- a/populate.py
+++ b/populate.py
@@ -4,7 +4,6 @@
 import datetime
 import hashlib
 import random
-import sys
 import argparse
 
 from CTFd import create_app
@@ -53,136 +52,6 @@ categories = [
     "Cryptography",
     "Networking",
 ]
-extensions = [
-    ".doc",
-    ".log",
-    ".msg",
-    ".rtf",
-    ".txt",
-    ".wpd",
-    ".wps",
-    ".123",
-    ".csv",
-    ".dat",
-    ".db",
-    ".dll",
-    ".mdb",
-    ".pps",
-    ".ppt",
-    ".sql",
-    ".wks",
-    ".xls",
-    ".xml",
-    ".mng",
-    ".pct",
-    ".bmp",
-    ".gif",
-    ".jpe",
-    ".jpg",
-    ".png",
-    ".psd",
-    ".psp",
-    ".tif",
-    ".ai",
-    ".drw",
-    ".dxf",
-    ".eps",
-    ".ps",
-    ".svg",
-    ".3dm",
-    ".3dm",
-    ".ind",
-    ".pdf",
-    ".qxd",
-    ".qxp",
-    ".aac",
-    ".aif",
-    ".iff",
-    ".m3u",
-    ".mid",
-    ".mid",
-    ".mp3",
-    ".mpa",
-    ".ra",
-    ".ram",
-    ".wav",
-    ".wma",
-    ".3gp",
-    ".asf",
-    ".asx",
-    ".avi",
-    ".mov",
-    ".mp4",
-    ".mpg",
-    ".qt",
-    ".rm",
-    ".swf",
-    ".wmv",
-    ".asp",
-    ".css",
-    ".htm",
-    ".htm",
-    ".js",
-    ".jsp",
-    ".php",
-    ".xht",
-    ".fnt",
-    ".fon",
-    ".otf",
-    ".ttf",
-    ".8bi",
-    ".plu",
-    ".xll",
-    ".cab",
-    ".cpl",
-    ".cur",
-    ".dmp",
-    ".drv",
-    ".key",
-    ".lnk",
-    ".sys",
-    ".cfg",
-    ".ini",
-    ".reg",
-    ".app",
-    ".bat",
-    ".cgi",
-    ".com",
-    ".exe",
-    ".pif",
-    ".vb",
-    ".ws",
-    ".deb",
-    ".gz",
-    ".pkg",
-    ".rar",
-    ".sea",
-    ".sit",
-    ".sit",
-    ".zip",
-    ".bin",
-    ".hqx",
-    ".0 E",
-    ".mim",
-    ".uue",
-    ".cpp",
-    ".jav",
-    ".pl",
-    ".bak",
-    ".gho",
-    ".old",
-    ".ori",
-    ".tmp",
-    ".dmg",
-    ".iso",
-    ".toa",
-    ".vcd",
-    ".gam",
-    ".nes",
-    ".rom",
-    ".sav",
-    ".msi",
-]
 companies = ["Corp", "Inc.", "Squad", "Team"]
 icons = [
     None,
@@ -235,7 +104,7 @@ def gen_icon():
 
 
 def gen_file():
-    return gen_word() + random.choice(extensions)
+    return fake.file_name()
 
 
 def random_date(start, end):
@@ -315,9 +184,7 @@ if __name__ == "__main__":
             if name not in used:
                 used.append(name)
                 try:
-                    user = Users(
-                        name=name, email=name + gen_email(), password="password"
-                    )
+                    user = Users(name=name, email=gen_email(), password="password")
                     user.verified = True
                     if random_chance():
                         user.affiliation = gen_affiliation()