wsgi.py

import click, sys
from models import db, User, Todo
from app import app
from sqlalchemy.exc import IntegrityError


@app.cli.command("init", help="Creates and initializes the database")
def initialize():
  db.drop_all()
  db.init_app(app)
  db.create_all()
  bob = User('bob', 'bob@mail.com', 'bobpass')
  bob.todos.append(Todo('make pancakes'))
  db.session.add(bob)
  db.session.commit()
  print(bob)
  print('database intialized')

@app.cli.command("get-user", help="Retrieves a User")
@click.argument('username', default='bob')
def get_user(username):
  bob = User.query.filter_by(username=username).first()
  if not bob:
    print(f'{username} not found!')
    return
  print(bob)

@app.cli.command("get-users", help="Retreives all users")
def get_users():
  users = User.query.all()
  print(users)

@app.cli.command("change-email", help="Changes a User email")
@click.argument('username', default='bob')
@click.argument('email', default='bob@mail.com')
def change_email(username, email):
  user = User.query.filter_by(username=username).first()
  if not user:
    print(f'{username} not found!')
    return
  user.email = email
  db.session.add(user)
  db.session.commit()
  print(user)

@app.cli.command("create-user", help="Creates a User")
@click.argument("username", default="EricShun")
@click.argument("email", default="ericshun@mail.com")
@click.argument("password", default="ericpass")
def create_user(username, email, password):
  newuser = User(username, email, password)
  try:
    db.session.add(newuser)
    db.session.commit()
  except IntegrityError as e:
    db.session.rollback()
    print(e.orig)
    print("Username or email already taken!")
  else:
    print(newuser)

@app.cli.command("delete-user", help="Deletes a User")
@click.argument("username", default="bob")
def delete_user(username):
  user = User.query.filter_by(username=username).first()
  if not user:
    print("{username} not found!")
    return
  db.session.delete(user)
  db.session.commit()
  print(f'{username} deleted')

@app.cli.command("get-todos")
@click.argument("username", default="bob")
def get_user_todos(username):
  user = User.query.filter_by(username=username).first()
  if not user:
    print(f"{username} not found!")
    return
  print(user.todos)

@app.cli.command("add-todo", help="Add a todo to a User's todos")
@click.argument("username", default="bob")
@click.argument("text", default="bake bread")
def add_task(username, text):
  user = User.query.filter_by(username=username).first()
  if not user:
    print(f"{username} not found!")
    return
  new_todo = Todo(text)
  user.todos.append(new_todo)
  db.session.add(user)
  db.session.commit()

@click.argument("username", default="bob")
@click.argument("todo_id", default=1)
@app.cli.command("toggle-todo")
def toggle_todo_command(username, todo_id):
  user = User.query.filter_by(username=username).first()
  if not user:
    print(f'{username} not found!')
    return
  todo = Todo.query.filter_by(id=todo_id, user_id=user.id).first()
  if not todo:
    print(f'{username} has no todo id {todo_id}')
  todo.toggle()
  print(f"{todo.text} is {'done' if todo.done else 'not done'}!")

@click.argument('username', default='bob')
@click.argument('todo_id', default=6)
@click.argument('category', default='chores')
@app.cli.command('add-category', help="Adds a category to a todo")
def add_todo_category_command(username, todo_id, category):
  user = User.query.filter_by(username=username).first()
  if not user:
    print(f'{username} not found!')
    return

  res = user.add_todo_category(todo_id, category)
  if not res:
    print(f'{username} has no todo id {todo_id}')
    return

  print('Category added!')