fix: create admin token

10 months ago · 49c1432220
parent 2f605647a4
commit 49c1432220
1 changed files with 10 additions and 2 deletions
--- a/scripts/create_admin_token.py
+++ b/scripts/create_admin_token.py
@ -1,5 +1,5 @@
+from jose import jws
 import tap
-from app.auth import create_admin_token


 class Arguments(tap.Tap):
@ -7,8 +7,16 @@ class Arguments(tap.Tap):
    secret: str


+def create_admin_token(election_ref: str, secret: str) -> str:
+    return jws.sign(
+        {"admin": True, "election": election_ref},
+        secret,
+        algorithm="HS256",
+    )
+
+
 def main(args: Arguments) -> None:
-    print(create_admin_token(args.ref))
+    print(create_admin_token(args.ref, args.secret))


 if __name__ == "__main__":