first commit

app/cli/commands.py

@@ -0,0 +1,81 @@
+import click
+from flask.cli import with_appcontext
+
+from ..extensions import db
+from ..models import User, seed_categories
+from ..services.reporting import send_due_reports
+
+
+@click.command('create-user')
+@click.option('--email', prompt=True)
+@click.option('--name', prompt=True)
+@click.option('--password', prompt=True, hide_input=True, confirmation_prompt=True)
+@click.option('--admin', is_flag=True, default=False)
+@with_appcontext
+def create_user(email, name, password, admin):
+    user = User(email=email.lower(), full_name=name, role='admin' if admin else 'user', must_change_password=False)
+    user.set_password(password)
+    db.session.add(user)
+    db.session.commit()
+    click.echo(f'Created user {email}')
+
+
+@click.command('reset-password')
+@click.option('--email', prompt=True)
+@click.option('--password', prompt=True, hide_input=True, confirmation_prompt=True)
+@with_appcontext
+def reset_password(email, password):
+    user = User.query.filter_by(email=email.lower()).first()
+    if not user:
+        raise click.ClickException('User not found')
+    user.set_password(password)
+    user.must_change_password = True
+    db.session.commit()
+    click.echo(f'Password reset for {email}')
+
+
+@click.command('make-admin')
+@click.option('--email', prompt=True)
+@with_appcontext
+def make_admin(email):
+    user = User.query.filter_by(email=email.lower()).first()
+    if not user:
+        raise click.ClickException('User not found')
+    user.role = 'admin'
+    db.session.commit()
+    click.echo(f'Granted admin to {email}')
+
+
+@click.command('deactivate-user')
+@click.option('--email', prompt=True)
+@with_appcontext
+def deactivate_user(email):
+    user = User.query.filter_by(email=email.lower()).first()
+    if not user:
+        raise click.ClickException('User not found')
+    user.is_active_user = False
+    db.session.commit()
+    click.echo(f'Deactivated {email}')
+
+
+@click.command('send-reports')
+@with_appcontext
+def send_reports_command():
+    count = send_due_reports()
+    click.echo(f'Sent {count} reports')
+
+
+@click.command('seed-categories')
+@with_appcontext
+def seed_categories_command():
+    seed_categories()
+    click.echo('Categories seeded')
+
+
+def register_commands(app):
+    app.cli.add_command(create_user)
+    app.cli.add_command(reset_password)
+    app.cli.add_command(make_admin)
+    app.cli.add_command(deactivate_user)
+    app.cli.add_command(send_reports_command)
+    app.cli.add_command(seed_categories_command)