add --trolls

Author: jonbgamble

spamdb/modules/args.py

@@ -68,7 +68,7 @@ def parse_args() -> argparse.Namespace:
             with random image from lifat backgrounds. (default: 200)
         """,
         default=200,
-        choices=[100, 200, 400],
+        choices=[100, 110, 200, 400],
     )
     parser.add_argument(
         '--no-flair',
@@ -206,6 +206,11 @@ def parse_args() -> argparse.Namespace:
         action='store_true',
         help='list all spamdb users along with their ratings then exit',
     )
+    parser.add_argument(
+        '--trolls',
+        action='store_true',
+        help='all normal users are trolls',
+    )
     group = parser.add_mutually_exclusive_group()
     group.add_argument(
         '--dump-bson',

spamdb/modules/user.py

@@ -38,7 +38,7 @@ def update_user_colls() -> None:
     follow_factor = args.follow
 
     for uid in env.uids:
-        users.append(User(uid))
+        users.append(User(uid, ['troll'] if args.trolls else []))
         perfs, stats = users[-1].detach_perfs()
         userperfs.append(perf.UserPerfs(uid, perfs))
         history.append(History(userperfs[-1], users[-1].createdAt))
@@ -299,6 +299,8 @@ def _create_special_users():
     users.append(User('ai', [], [], False))
     if env.args.su is not None:
         users.append(User(env.args.su, [], ['ROLE_SUPER_ADMIN'], False))
+        if env.args.trolls:
+            return users
     else:
         users.append(User('superadmin', [], ['ROLE_SUPER_ADMIN'], False))
         users.append(User('admin', [], ['ROLE_ADMIN'], False))