Add command line interface for admin tasks

1 files changed, 34 insertions, 0 deletions

diff --git a/app/cli.py b/app/cli.py
new file mode 100644
index 0000000..e128c35
--- /dev/null
+++ b/app/cli.py
@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+
+import getpass
+
+from email_validator import validate_email
+
+from app import db
+from app.models import User
+
+
+def register(app):
+    @app.cli.group(help="Perform admin commands")
+    def admin():
+        pass
+
+    @admin.command()
+    def add():
+        username = input("Enter username: ")
+        email = input("Enter email address: ")
+        password = getpass.getpass()
+        assert password == getpass.getpass("Repeat password: ")
+
+        user = User.query.filter_by(username=username).first()
+        if user is not None:
+            raise ValueError("User already exists")
+
+        validate_email(email)
+
+        user = User(username=username, email=email, is_admin=True)
+        user.set_password(password)
+        db.session.add(user)
+        db.session.commit()
+
+        print("Admin user %r added successfully." % username)