shopdb-flask/shopdb/cli/__init__.py

"""Flask CLI commands."""

import click
from flask.cli import with_appcontext


@click.group('db-utils')
def db_cli():
    """Database utility commands."""
    pass


@db_cli.command('create-all')
@with_appcontext
def create_all():
    """Create all database tables."""
    from shopdb.extensions import db

    db.create_all()
    click.echo(click.style("All tables created.", fg='green'))


@db_cli.command('drop-all')
@click.confirmation_option(prompt='This will delete ALL data. Are you sure?')
@with_appcontext
def drop_all():
    """Drop all database tables."""
    from shopdb.extensions import db

    db.drop_all()
    click.echo(click.style("All tables dropped.", fg='yellow'))


@click.group('seed')
def seed_cli():
    """Database seeding commands."""
    pass


@seed_cli.command('reference-data')
@with_appcontext
def seed_reference_data():
    """Seed reference data (machine types, statuses, etc.)."""
    from shopdb.extensions import db
    from shopdb.core.models import MachineType, MachineStatus, OperatingSystem
    from shopdb.core.models.relationship import RelationshipType

    # Machine types
    machine_types = [
        {'machinetype': 'CNC Mill', 'category': 'Equipment', 'description': 'CNC Milling Machine'},
        {'machinetype': 'CNC Lathe', 'category': 'Equipment', 'description': 'CNC Lathe'},
        {'machinetype': 'CMM', 'category': 'Equipment', 'description': 'Coordinate Measuring Machine'},
        {'machinetype': 'EDM', 'category': 'Equipment', 'description': 'Electrical Discharge Machine'},
        {'machinetype': 'Grinder', 'category': 'Equipment', 'description': 'Grinding Machine'},
        {'machinetype': 'Inspection Station', 'category': 'Equipment', 'description': 'Inspection Station'},
        {'machinetype': 'Desktop PC', 'category': 'PC', 'description': 'Desktop Computer'},
        {'machinetype': 'Laptop', 'category': 'PC', 'description': 'Laptop Computer'},
        {'machinetype': 'Shopfloor PC', 'category': 'PC', 'description': 'Shopfloor Computer'},
        {'machinetype': 'Server', 'category': 'Network', 'description': 'Server'},
        {'machinetype': 'Switch', 'category': 'Network', 'description': 'Network Switch'},
        {'machinetype': 'Access Point', 'category': 'Network', 'description': 'Wireless Access Point'},
    ]

    for mt_data in machine_types:
        existing = MachineType.query.filter_by(machinetype=mt_data['machinetype']).first()
        if not existing:
            mt = MachineType(**mt_data)
            db.session.add(mt)

    # Machine statuses
    statuses = [
        {'status': 'In Use', 'description': 'Currently in use', 'color': '#28a745'},
        {'status': 'Spare', 'description': 'Available as spare', 'color': '#17a2b8'},
        {'status': 'Retired', 'description': 'No longer in use', 'color': '#6c757d'},
        {'status': 'In Repair', 'description': 'Currently being repaired', 'color': '#ffc107'},
        {'status': 'Pending', 'description': 'Pending installation', 'color': '#007bff'},
    ]

    for s_data in statuses:
        existing = MachineStatus.query.filter_by(status=s_data['status']).first()
        if not existing:
            s = MachineStatus(**s_data)
            db.session.add(s)

    # Operating systems
    os_list = [
        {'osname': 'Windows 10', 'osversion': '10.0'},
        {'osname': 'Windows 11', 'osversion': '11.0'},
        {'osname': 'Windows Server 2019', 'osversion': '2019'},
        {'osname': 'Windows Server 2022', 'osversion': '2022'},
        {'osname': 'Linux', 'osversion': 'Various'},
    ]

    for os_data in os_list:
        existing = OperatingSystem.query.filter_by(osname=os_data['osname']).first()
        if not existing:
            os_obj = OperatingSystem(**os_data)
            db.session.add(os_obj)

    # Connection types (how PC connects to equipment)
    connection_types = [
        {'relationshiptype': 'Serial Cable', 'description': 'RS-232 or similar serial connection'},
        {'relationshiptype': 'Direct Ethernet', 'description': 'Direct network cable (airgapped)'},
        {'relationshiptype': 'USB', 'description': 'USB connection'},
        {'relationshiptype': 'WiFi', 'description': 'Wireless network connection'},
        {'relationshiptype': 'Dualpath', 'description': 'Redundant/failover network path'},
    ]

    for ct_data in connection_types:
        existing = RelationshipType.query.filter_by(relationshiptype=ct_data['relationshiptype']).first()
        if not existing:
            ct = RelationshipType(**ct_data)
            db.session.add(ct)

    db.session.commit()
    click.echo(click.style("Reference data seeded.", fg='green'))


@seed_cli.command('test-user')
@with_appcontext
def seed_test_user():
    """Create a test admin user."""
    from shopdb.extensions import db
    from shopdb.core.models import User, Role
    from werkzeug.security import generate_password_hash

    # Create admin role if not exists
    admin_role = Role.query.filter_by(rolename='admin').first()
    if not admin_role:
        admin_role = Role(rolename='admin', description='Administrator')
        db.session.add(admin_role)

    # Create test user
    test_user = User.query.filter_by(username='admin').first()
    if not test_user:
        test_user = User(
            username='admin',
            email='admin@localhost',
            passwordhash=generate_password_hash('admin123'),
            isactive=True
        )
        test_user.roles.append(admin_role)
        db.session.add(test_user)
        db.session.commit()
        click.echo(click.style("Test user created: admin / admin123", fg='green'))
    else:
        click.echo(click.style("Test user already exists", fg='yellow'))